Update websocket_server.go

stream/websocket_server.go

@@ -49,13 +49,16 @@ func (ws *WebsocketServer) handleClient(c *websocket.Conn, w http.ResponseWriter
 		var cmd command
 		err := wsjson.Read(r.Context(), c, &cmd)
 		if err != nil {
-			wsjson.Write(r.Context(), c, map[string]any{
+			err := wsjson.Write(r.Context(), c, map[string]any{
 				"error": map[string]any{
 					"code":    -32600,
 					"message": "invalid message",
 					"data":    err.Error(),
 				},
 			})
+			if err != nil {
+				break
+			}
 			continue
 		}
 		switch cmd.Method {