切换虎牙宝典活动任务到WSS会话

2026-09-01 12:23:43 +08:00
parent 17e72d40a5
commit 2fca1210bb
3 changed files with 26 additions and 20 deletions
@@ -64,6 +64,20 @@ class HuyaBatchRunnerCore:
        self._wss_sessions.append(session)
        return session

+    def _activity_client(self, uid: int, cookie: str):
+        """活动任务优先使用最新活动 WSS，建连失败才回退 HTTP。"""
+        from core.huya.http_client import HuyaHttpClient
+
+        try:
+            session = self._open_wss_session(uid, cookie, "activity")
+            self._push_log("info", f"[{uid}] 活动链路使用 WSS 会话")
+            return session
+        except Exception as exc:  # noqa: BLE001 - fallback is part of transport policy
+            self._push_log(
+                "warning", f"[{uid}] 活动 WSS 不可用，回退 HTTP: {type(exc).__name__}"
+            )
+            return HuyaHttpClient(logger=lambda msg: self._push_log("info", f"[{uid}] {msg}"))
+
    def _close_wss_sessions(self):
        for session in reversed(self._wss_sessions):
            try: