Add logic to make sure to close all http connection once the server i…

…s closed
ella-to · May 2, 2024 · d9809f3 · d9809f3
1 parent bb56f70
commit d9809f3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/server/server.go b/server/server.go
@@ -241,6 +241,8 @@ func (h *Handler) consumerHandler(w http.ResponseWriter, r *http.Request) {
 		if isAutoAck {
 			if lastEventId != "" {
 				select {
+				case <-h.closeSignal:
+					return
 				case <-ctx.Done():
 					return
 				case events, ok := <-batch:
@@ -271,6 +273,8 @@ func (h *Handler) consumerHandler(w http.ResponseWriter, r *http.Request) {
 				}
 			} else {
 				select {
+				case <-h.closeSignal:
+					return
 				case <-ctx.Done():
 					return
 				case events, ok := <-batch:
@@ -292,6 +296,8 @@ func (h *Handler) consumerHandler(w http.ResponseWriter, r *http.Request) {
 			}
 		} else {
 			select {
+			case <-h.closeSignal:
+				return
 			case <-ctx.Done():
 				return
 			case events, ok := <-batch: