From ed749fb606ac7738bf46d84608667e9f382271a9 Mon Sep 17 00:00:00 2001
From: WrBug <iwrbug@gmail.com>
Date: Sat, 14 Feb 2026 05:47:05 +0800
Subject: [PATCH] =?UTF-8?q?feat(crypto-tail):=20=E5=81=9C=E7=94=A8?=
 =?UTF-8?q?=E7=AD=96=E7=95=A5=E6=97=B6=E5=85=B3=E9=97=AD=20WebSocket?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 无启用策略时关闭订单簿 WS 并取消重连（closedForNoStrategies 标志）
- 启动时若无启用策略则不建立连接
- 再次启用策略时 refreshAndSubscribe 会重新 connect 并订阅

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 .../CryptoTailOrderbookWsService.kt           | 36 +++++++++++++++++--
 1 file changed, 33 insertions(+), 3 deletions(-)

diff --git a/backend/src/main/kotlin/com/wrbug/polymarketbot/service/cryptotail/CryptoTailOrderbookWsService.kt b/backend/src/main/kotlin/com/wrbug/polymarketbot/service/cryptotail/CryptoTailOrderbookWsService.kt
index 5827244..7a3d576 100644
--- a/backend/src/main/kotlin/com/wrbug/polymarketbot/service/cryptotail/CryptoTailOrderbookWsService.kt
+++ b/backend/src/main/kotlin/com/wrbug/polymarketbot/service/cryptotail/CryptoTailOrderbookWsService.kt
@@ -26,6 +26,7 @@ import org.springframework.context.event.EventListener
 import org.springframework.stereotype.Service
 import jakarta.annotation.PostConstruct
 import java.math.BigDecimal
+import java.util.concurrent.atomic.AtomicBoolean
 import java.util.concurrent.atomic.AtomicReference
 
 /**
@@ -55,6 +56,9 @@ class CryptoTailOrderbookWsService(
     /** 重连延迟（毫秒） */
     private val reconnectDelayMs = 10_000L
 
+    /** 因无启用策略而主动关闭 WS 时置为 true，onClosing 中不触发重连 */
+    private val closedForNoStrategies = AtomicBoolean(false)
+
     data class WsBookEntry(
         val strategy: CryptoTailStrategy,
         val periodStartUnix: Long,
@@ -65,7 +69,7 @@ class CryptoTailOrderbookWsService(
 
     @PostConstruct
     fun init() {
-        connect()
+        if (strategyRepository.findAllByEnabledTrue().isNotEmpty()) connect()
     }
 
     private fun connect() {
@@ -84,7 +88,7 @@ class CryptoTailOrderbookWsService(
 
                 override fun onClosing(webSocket: WebSocket, code: Int, reason: String) {
                     this@CryptoTailOrderbookWsService.webSocket = null
-                    scheduleReconnect()
+                    if (!closedForNoStrategies.getAndSet(false)) scheduleReconnect()
                 }
 
                 override fun onFailure(webSocket: WebSocket, t: Throwable, response: okhttp3.Response?) {
@@ -192,7 +196,14 @@ class CryptoTailOrderbookWsService(
         periodEndCountdownJob = null
         val (tokenIds, newMap) = buildSubscriptionMap()
         tokenToEntries.set(newMap)
-        if (tokenIds.isEmpty()) return
+        if (tokenIds.isEmpty()) {
+            closeWebSocketForNoStrategies()
+            return
+        }
+        if (webSocket == null) {
+            connect()
+            return
+        }
         val marketSlugs = newMap.values.asSequence().flatten()
             .distinctBy { "${it.strategy.marketSlugPrefix}-${it.periodStartUnix}" }
             .map { "${it.strategy.marketSlugPrefix}-${it.periodStartUnix}" }
@@ -208,6 +219,25 @@ class CryptoTailOrderbookWsService(
         scheduleRefreshAtPeriodEnd(newMap)
     }
 
+    /**
+     * 无启用策略或无需订阅时关闭 WebSocket，并取消重连；停用策略后刷新订阅会走到此处。
+     */
+    private fun closeWebSocketForNoStrategies() {
+        reconnectJob?.cancel()
+        reconnectJob = null
+        val ws = webSocket
+        if (ws != null) {
+            closedForNoStrategies.set(true)
+            webSocket = null
+            try {
+                ws.close(1000, "no_enabled_strategies")
+            } catch (e: Exception) {
+                logger.debug("关闭尾盘策略 WebSocket 时异常: ${e.message}")
+            }
+            logger.info("尾盘策略订单簿 WebSocket 已关闭（无启用策略）")
+        }
+    }
+
     /**
      * 订阅成功后设置倒计时：在当前周期结束时自动刷新订阅，无需等消息触发。
      */