ensure stream write timeout gets cancelled

superseriousbusiness · Feb 12, 2024 · 99df0d5 · 99df0d5
1 parent 46ef2e5
commit 99df0d5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/internal/api/client/streaming/stream.go b/internal/api/client/streaming/stream.go
@@ -354,11 +354,14 @@ func (m *Module) writeToWSConn(
 
 	for {
 		// Wrap stream context with timeout to send a ping.
-		ctx, _ := context.WithTimeout(stream.Context(), ping)
+		ctx, cncl := context.WithTimeout(stream.Context(), ping)
 
 		// Block on receipt of next message.
 		msg, ok, open := stream.Recv(ctx)
 
+		// Stop timer.
+		cncl()
+
 		if !open {
 			// Stream was
 			// closed.