Don't remove connection from WaitGroup on keepAlive (#230)

claudiodangelis · Mar 12, 2022 · 64227a4 · 64227a4
1 parent 01a36c4
commit 64227a4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)