Improved with a cancelable context

insighio · Apr 5, 2024 · 1dbc52e · 1dbc52e
1 parent 3ae1d1b
commit 1dbc52e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/mqtt/websocket/websocket.go b/pkg/mqtt/websocket/websocket.go
@@ -64,8 +64,9 @@ func (p Proxy) handle() http.Handler {
 			return
 		}
 
-		ctx := context.Background() // Create a new context
+		ctx, cancel := context.WithCancel(context.Background())
 		go func() {
+			defer cancel() // Cancel the context when the goroutine finishes
 			p.pass(ctx, cconn)
 			if err := ctx.Err(); err != nil {
 				p.logger.Info("Context error at the end of goroutine", slog.Any("error", err))