fix client & server experimental fix

internal/hub.go

@@ -11,7 +11,7 @@ import (
 const (
 	writeWait  = 10 * time.Second
 	pongWait   = 60 * time.Second
-	pingPeriod = 55 * time.Second
+	pingPeriod = 25 * time.Second
 )
 
 type Client struct {
@@ -95,7 +95,7 @@ func WritePump(c *Client, h *Hub) {
 			}
 		case <-pingTicker.C:
 			c.Conn.SetWriteDeadline(time.Now().Add(writeWait))
-			if err := c.Conn.WriteMessage(websocket.PingMessage, nil); err != nil {
+			if err := c.Conn.WriteMessage(websocket.PingMessage, []byte{}); err != nil {
 				return
 			}
 		}

v1/client/client.go

@@ -124,6 +124,7 @@ func (wsClient *SafeWebsocketClient) connect() error {
 	}
 
 	conn.SetPingHandler(func(pingData string) error {
+		return wsClient.mu.WriteHandler(func() error {
 			if err := conn.WriteMessage(websocket.PongMessage, []byte(pingData)); err != nil {
 				if err == websocket.ErrCloseSent {
 					return nil
@@ -135,6 +136,7 @@ func (wsClient *SafeWebsocketClient) connect() error {
 			}
 			return nil
 		})
+	})
 
 	wsClient.mu.WriteHandler(func() error {
 		ctx, cancel := context.WithCancel(context.Background())
@@ -142,6 +144,7 @@ func (wsClient *SafeWebsocketClient) connect() error {
 		wsClient.cancel = cancel
 		wsClient.conn = conn
 		wsClient.isConnected = true
+
 		return nil
 	})
 
@@ -179,22 +182,26 @@ func (wsClient *SafeWebsocketClient) startReceiveHandler() {
 			log.Println("Reconnect handler stopped")
 			return
 		default:
-			if err := wsClient.mu.ReadHandler(func() error {
+			// if err := wsClient.mu.ReadHandler(func() error {
 			conn := wsClient.conn
 
 			if conn == nil {
-					return fmt.Errorf("no active connection, waiting for reconnect")
+				wsClient.triggerReconnect()
+				return
+				// return fmt.Errorf("no active connection, waiting for reconnect")
 			}
 			_, message, err := conn.ReadMessage()
 			if err != nil {
-					return err
-				}
-				wsClient.dataChannel <- message
-				return nil
-			}); err != nil {
 				wsClient.triggerReconnect()
 				return
+				// return err
 			}
+			wsClient.dataChannel <- message
+			// return nil
+			// }); err != nil {
+			// 	wsClient.triggerReconnect()
+			// 	return
+			// }
 		}
 	}
 }