瀏覽代碼

fix: websocket disconnect when window focus

GyDi 1 年之前
父節點
當前提交
6f5acee1c3
共有 2 個文件被更改,包括 9 次插入2 次删除
  1. 7 0
      src/hooks/use-visibility.ts
  2. 2 2
      src/hooks/use-websocket.ts

+ 7 - 0
src/hooks/use-visibility.ts

@@ -8,10 +8,17 @@ export const useVisibility = () => {
       setVisible(document.visibilityState === "visible");
     };
 
+    const handleFocus = () => setVisible(true);
+    const handleClick = () => setVisible(true);
+
     handleVisibilityChange();
+    document.addEventListener("focus", handleFocus);
+    document.addEventListener("pointerdown", handleClick);
     document.addEventListener("visibilitychange", handleVisibilityChange);
 
     return () => {
+      document.removeEventListener("focus", handleFocus);
+      document.removeEventListener("pointerdown", handleClick);
       document.removeEventListener("visibilitychange", handleVisibilityChange);
     };
   }, []);

+ 2 - 2
src/hooks/use-websocket.ts

@@ -2,12 +2,12 @@ import { useRef } from "react";
 
 export type WsMsgFn = (event: MessageEvent<any>) => void;
 
-interface Options {
+export interface WsOptions {
   errorCount?: number; // default is 5
   retryInterval?: number; // default is 2500
 }
 
-export const useWebsocket = (onMessage: WsMsgFn, options?: Options) => {
+export const useWebsocket = (onMessage: WsMsgFn, options?: WsOptions) => {
   const wsRef = useRef<WebSocket | null>(null);
   const timerRef = useRef<any>(null);