Improve graceful shutdown

This commit is contained in:
Frédéric Guillot 2018-04-17 21:50:52 -07:00
parent 0429bbb19d
commit 657e96e133
1 changed files with 3 additions and 1 deletions

View File

@ -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")