Update pkg/app.go

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
QuokkaStake · Jun 28, 2024 · 0aaff48 · 0aaff48
1 parent 9f5189f
commit 0aaff48
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/app.go b/pkg/app.go
@@ -152,7 +152,9 @@ func (a *App) Start() {
 
 func (a *App) Stop() {
 	a.Logger.Info().Str("addr", a.Config.ListenAddress).Msg("Shutting down server...")
-	_ = a.Server.Shutdown(context.Background())
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	_ = a.Server.Shutdown(ctx)
 }
 
 func (a *App) Handler(w http.ResponseWriter, r *http.Request) {