diff options
Diffstat (limited to 'backend/api')
| -rw-r--r-- | backend/api/api.go | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/backend/api/api.go b/backend/api/api.go index 9dd68a9..183c090 100644 --- a/backend/api/api.go +++ b/backend/api/api.go @@ -1,6 +1,7 @@ package api import ( + "context" "net/http" "os" "os/signal" @@ -10,9 +11,15 @@ import ( mux "github.com/gorilla/mux" log "github.com/sirupsen/logrus" + "github.com/jackyzha0/ctrl-v/db" ) func cleanup() { + ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) + defer cancel() + if err := db.Client.Disconnect(ctx); err != nil { + panic(err) + } log.Print("Shutting down server...") } |