diff options
| author | Jacky Zhao <[email protected]> | 2023-07-09 15:05:12 -0700 |
|---|---|---|
| committer | Jacky Zhao <[email protected]> | 2023-07-09 15:05:12 -0700 |
| commit | 9853ff01573b9495da848e09a62cd11add19e4bf (patch) | |
| tree | cbb663bcea9c731b5ffd02c71488682a1dab083e /backend/api | |
| parent | add tls config (diff) | |
| download | ctrl-v-9853ff01573b9495da848e09a62cd11add19e4bf.tar.xz ctrl-v-9853ff01573b9495da848e09a62cd11add19e4bf.zip | |
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...") } |