Fixed goroutine exit

twitchdev · Apr 21, 2023 · e748f82 · e748f82
1 parent 96a6b3f
commit e748f82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/events/websocket/mock_server/server.go b/internal/events/websocket/mock_server/server.go
@@ -175,7 +175,7 @@ func (ws *WebSocketServer) WsPageHandler(w http.ResponseWriter, r *http.Request)
 			case <-client.keepAliveLoopChan:
 				client.keepAliveTimer.Stop()
 				client.keepAliveLoopChan = nil
-				break
+				return
 
 			case <-client.keepAliveTimer.C: // Send KeepAlive message
 				keepAliveMsg, _ := json.Marshal(
@@ -207,7 +207,7 @@ func (ws *WebSocketServer) WsPageHandler(w http.ResponseWriter, r *http.Request)
 			case <-client.pingLoopChan:
 				client.pingTimer.Stop()
 				client.pingLoopChan = nil
-				break
+				return
 
 			case <-client.pingTimer.C: // Send ping
 				err := client.SendMessage(websocket.PingMessage, []byte{})