reconnect while close abnormally

MrXiaoM · Jan 7, 2024 · 10537f2 · 10537f2
1 parent 3a893c5
commit 10537f2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/onebot/src/main/kotlin/cn/evolvefield/onebot/client/connection/WSClient.kt b/onebot/src/main/kotlin/cn/evolvefield/onebot/client/connection/WSClient.kt
@@ -13,6 +13,7 @@ import kotlinx.coroutines.runBlocking
 import kotlinx.coroutines.sync.Mutex
 import kotlinx.coroutines.sync.withLock
 import org.java_websocket.client.WebSocketClient
+import org.java_websocket.framing.CloseFrame
 import org.java_websocket.handshake.ServerHandshake
 import org.slf4j.Logger
 import java.net.URI
@@ -98,6 +99,9 @@ class WSClient(
             if (mutex.isLocked) mutex.unlock()
             if (ActionSendUtils.mutex.isLocked) ActionSendUtils.mutex.unlock()
         }
+        if (retryCount < 1 && code != CloseFrame.NORMAL) { // TODO: 测试确认异常关闭码
+            reconnectBlocking()
+        }
     }
 
     override fun onError(ex: Exception) {