From 9a47a7271a1b7fa6270fde99b999ed983ff560ab Mon Sep 17 00:00:00 2001
From: HengJing Wang <56909676+a982246809@users.noreply.github.com>
Date: Tue, 20 Feb 2024 17:33:50 +0800
Subject: [PATCH] fix(useWebSocket): reset `wsRef` on close, fix #3706 (#3707)

Co-authored-by: Doctor Wu <44631608+Doctor-wu@users.noreply.github.com>
---
 packages/core/useWebSocket/index.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/core/useWebSocket/index.ts b/packages/core/useWebSocket/index.ts
index 85dbf2bc77f..72e97a7e817 100644
--- a/packages/core/useWebSocket/index.ts
+++ b/packages/core/useWebSocket/index.ts
@@ -192,6 +192,7 @@ export function useWebSocket<Data = any>(
     resetHeartbeat()
     heartbeatPause?.()
     wsRef.value.close(code, reason)
+    wsRef.value = undefined
   }
 
   const send = (data: string | ArrayBuffer | Blob, useBuffer = true) => {
@@ -222,7 +223,6 @@ export function useWebSocket<Data = any>(
 
     ws.onclose = (ev) => {
       status.value = 'CLOSED'
-      wsRef.value = undefined
       onDisconnected?.(ws, ev)
 
       if (!explicitlyClosed && options.autoReconnect) {