close channels

kurtosis-tech · Dec 18, 2023 · 0f9b454 · 0f9b454
1 parent 5483950
commit 0f9b454
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/engine/server/engine/streaming/websocket_pump.go b/engine/server/engine/streaming/websocket_pump.go
@@ -52,6 +52,7 @@ func NewWebsocketPump[T interface{}](ctx echo.Context, cors cors.Cors) (*Websock
 	return &WebsocketPump[T]{
 		websocket:  conn,
 		inputChan:  make(chan *T),
+		infoChan:   make(chan *api_type.ResponseInfo),
 		ctx:        ctxWithCancel,
 		cancelFunc: cancelFunc,
 	}, nil
@@ -62,6 +63,8 @@ func (pump WebsocketPump[T]) StartPumping() {
 	defer func() {
 		ticker.Stop()
 		pump.websocket.Close()
+		close(pump.inputChan)
+		close(pump.infoChan)
 	}()
 
 	logrus.WithFields(logrus.Fields{