From 64227a45a6c523d361cdb5a248e60c587471571d Mon Sep 17 00:00:00 2001 From: Claudio d'Angelis Date: Sat, 12 Mar 2022 18:29:38 +0100 Subject: [PATCH] Don't remove connection from WaitGroup on keepAlive (#230) --- server/server.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/server/server.go b/server/server.go index 3ff1e2e..bff3a98 100644 --- a/server/server.go +++ b/server/server.go @@ -214,7 +214,9 @@ func New(cfg *config.Config) (*Server, error) { } } // Remove connection from the waitgroup when done - defer waitgroup.Done() + if !cfg.KeepAlive { + defer waitgroup.Done() + } w.Header().Set("Content-Disposition", "attachment; filename="+ app.payload.Filename) http.ServeFile(w, r, app.payload.Path)