From 9853ff01573b9495da848e09a62cd11add19e4bf Mon Sep 17 00:00:00 2001 From: Jacky Zhao Date: Sun, 9 Jul 2023 15:05:12 -0700 Subject: move disconnection to cleanup --- backend/api/api.go | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'backend/api/api.go') 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...") } -- cgit v1.2.3