fix(useWebSocket): don't hide pong setTimeout reference (#2206)

Co-authored-by: Anthony Fu <anthonyfu117@hotmail.com>
vueuse · Nov 9, 2022 · 65f4341 · 65f4341
1 parent 934a893
commit 65f4341
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/core/useWebSocket/index.ts b/packages/core/useWebSocket/index.ts
@@ -190,6 +190,7 @@ export function useWebSocket<Data = any>(
 
   const resetHeartbeat = () => {
     clearTimeout(pongTimeoutWait)
+    pongTimeoutWait = undefined
   }
 
   const send = (data: string | ArrayBuffer | Blob, useBuffer = true) => {
@@ -269,6 +270,8 @@ export function useWebSocket<Data = any>(
     const { pause, resume } = useIntervalFn(
       () => {
         send(message, false)
+        if (pongTimeoutWait != null)
+          return
         pongTimeoutWait = setTimeout(() => {
           // auto-reconnect will be trigger with ws.onclose()
           close()