feat: add health

fastrodev · Nov 30, 2024 · a3bd912 · a3bd912
1 parent eee5d69
commit a3bd912
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/modules/hook/socket.ts b/modules/hook/socket.ts
@@ -25,11 +25,12 @@ const useWebSocket = (url: string, room: string, user: string) => {
             ...data,
           }),
         );
-        return countRef.current++;
+        countRef.current++;
+        console.log("ws ping:", room);
+      } else {
+        console.log("ws close:", room);
       }
-
-      console.log(`WebSocket connection closed: ${room}`);
-    }, 300);
+    }, 500);
   }
 
   useEffect(() => {
@@ -52,12 +53,14 @@ const useWebSocket = (url: string, room: string, user: string) => {
     };
 
     socketRef.current.onclose = () => {
+      console.error("WebSocket close");
       setIsConnected(false);
     };
 
     socketRef.current.onerror = (error) => {
       console.error("WebSocket error:", error);
       // Close the socket on error
+      setIsConnected(false);
       socketRef.current?.close();
     };