fix: websocket disconnect when window focus

This commit is contained in:
GyDi
2023-08-05 17:21:15 +08:00
parent 2ce944034d
commit cb1955c217
2 changed files with 9 additions and 2 deletions

View File

@@ -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);