fix: 修复readyState判断问题

alibaba · Dec 10, 2024 · 73d9ff7 · 73d9ff7
1 parent c7bb04c
commit 73d9ff7
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/packages/hooks/src/useWebSocket/index.ts b/packages/hooks/src/useWebSocket/index.ts
@@ -43,18 +43,19 @@ export default function useWebSocket(socketUrl: string, options: Options = {}):
     protocols,
   } = options;
 
+  const [latestMessage, setLatestMessage] = useState<WebSocketEventMap['message']>();
+  const [readyState, setReadyState] = useState<ReadyState>(ReadyState.Closed);
+
   const onOpenRef = useLatest(onOpen);
   const onCloseRef = useLatest(onClose);
   const onMessageRef = useLatest(onMessage);
   const onErrorRef = useLatest(onError);
+  const readyStateRef = useLatest(readyState);
 
   const reconnectTimesRef = useRef(0);
   const reconnectTimerRef = useRef<ReturnType<typeof setTimeout>>();
   const websocketRef = useRef<WebSocket>();
 
-  const [latestMessage, setLatestMessage] = useState<WebSocketEventMap['message']>();
-  const [readyState, setReadyState] = useState<ReadyState>(ReadyState.Closed);
-
   const reconnect = () => {
     if (
       reconnectTimesRef.current < reconnectLimit &&
@@ -123,7 +124,7 @@ export default function useWebSocket(socketUrl: string, options: Options = {}):
   };
 
   const sendMessage: WebSocket['send'] = (message) => {
-    if (readyState === ReadyState.Open) {
+    if (readyStateRef.current === ReadyState.Open) {
       websocketRef.current?.send(message);
     } else {
       throw new Error('WebSocket disconnected');