Improve graceful shutdown

2018-04-17 21:50:52 -07:00
parent 0429bbb19d
commit 657e96e133
1 changed files with 3 additions and 1 deletions
@@ -54,7 +54,9 @@ func Run(cfg *config.Config, store *storage.Storage) {

 	<-stop
 	logger.Info("Shutting down the server...")
-	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
 	server.Shutdown(ctx)
 	store.Close()
 	logger.Info("Server gracefully stopped")