From bd438e7b620fdbb916ff0be67fb6fdc3ca674a45 Mon Sep 17 00:00:00 2001
From: kron <kai_hong@126.com>
Date: Sun, 22 Jun 2025 10:57:42 +0800
Subject: [PATCH] =?UTF-8?q?bug=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/websocket.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/websocket.js b/src/websocket.js
index ed4b420..d7d2afc 100644
--- a/src/websocket.js
+++ b/src/websocket.js
@@ -40,7 +40,7 @@ function createWebSocket(token, onUpdate, onMessage) {
     console.log("WebSocket 已关闭", result);
     onUpdate(false);
     stopHeartbeat();
-    reconnect(onMessage);
+    reconnect(onUpdate, onMessage);
   });
 
   // 启动心跳
@@ -50,7 +50,7 @@ function createWebSocket(token, onUpdate, onMessage) {
 /**
  * 重连机制
  */
-function reconnect(onMessage) {
+function reconnect(onUpdate, onMessage) {
   if (reconnectCount >= RECONNECT_CONFIG.MAX_COUNT) return;
 
   reconnectTimer && clearTimeout(reconnectTimer);
@@ -64,7 +64,7 @@ function reconnect(onMessage) {
   );
   reconnectTimer = setTimeout(() => {
     console.log("reconnecting...");
-    createWebSocket(token, onMessage);
+    createWebSocket(token, onUpdate, onMessage);
     reconnectCount++;
   }, delay);
 }