复制 local 代码到 serve

2026-07-04 16:59:25 +08:00
parent cee04c1b46
commit 0474e5fb46
302 changed files with 78290 additions and 0 deletions
@@ -0,0 +1,147 @@
+import { useEffect, useRef, useCallback } from 'react'
+import { useQueryClient } from '@tanstack/react-query'
+import { SSE_INVALIDATE_PREFIXES, QK } from './queryKeys'
+import { getQueryConfig } from './useQueryConfig'
+import { toast } from '@/components/Toast'
+import { pushAlertToasts } from '@/components/AlertToast'
+import { feedReviewEvent } from './reviewStore'
+import type { StrategyAlertEvent } from './api'
+
+/**
+ * 全局 SSE hook: 监听后端行情更新推送 + 策略监控通知。
+ *
+ * - 行情更新 (quotes_updated): 根据 sseRefreshPages 配置过滤 invalidation
+ * - 策略监控通知 (strategy_alert): 通过 onAlert 回调弹 toast
+ *
+ * 应在顶层 Layout 中调用一次。
+ */
+export function useQuoteStream(
+  enabled: boolean,
+  sseRefreshPages: Record<string, boolean> | undefined,
+  onAlert?: (alerts: StrategyAlertEvent[]) => void,
+) {
+  const qc = useQueryClient()
+  const esRef = useRef<EventSource | null>(null)
+  const retryRef = useRef<ReturnType<typeof setTimeout>>()
+  const pagesRef = useRef(sseRefreshPages)
+  pagesRef.current = sseRefreshPages
+
+  const handleAlerts = useCallback((alerts: StrategyAlertEvent[]) => {
+    // depth 系统接管通知: 单独处理, 不走 strategy 回调
+    const depthAlerts = alerts.filter(a => a.source === 'depth')
+    const strategyAlerts = alerts.filter(a => a.source !== 'depth')
+
+    // depth 通知直接 toast(防刷屏: 后端已在状态切换时才推)
+    for (const a of depthAlerts.slice(0, 1)) {
+      toast(a.message, 'success')
+    }
+
+    // 监控告警: 用专用 AlertToast (整批只响一声, 每条都弹, 受 maxVisible 上限保护)
+    if (strategyAlerts.length > 0) {
+      // 有 onAlert 回调时走回调, 否则弹 AlertToast
+      if (onAlert) {
+        onAlert(strategyAlerts)
+      }
+      // 批量弹通知 (去掉了 slice(0,2) 截断, 让每只新命中都弹 toast; 声音整批只响一次)
+      pushAlertToasts(strategyAlerts as any)
+    }
+  }, [onAlert])
+
+  const enabledRef = useRef(enabled)
+  enabledRef.current = enabled
+
+  useEffect(() => {
+    // SSE 始终连接 — 监控告警不依赖实时行情开关
+    // (quotes_updated 行情刷新受 enabled 控制, strategy_alert 始终处理)
+
+    const connect = () => {
+      const es = new EventSource('/api/intraday/stream')
+      esRef.current = es
+
+      // sse-starlette ping 心跳走 SSE comment，不会到达这里
+
+      es.addEventListener('quotes_updated', () => {
+        // 实时行情未开启时不处理行情刷新
+        if (!enabledRef.current) return
+        // 根据用户配置过滤 invalidation
+        const pages = pagesRef.current
+        if (pages) {
+          // 只 invalidate 开启的页面对应的 prefix
+          const activePrefixes = SSE_INVALIDATE_PREFIXES.filter((p) => {
+            // 'quote-status' 始终刷新 (全局状态)
+            if (p === 'quote-status') return true
+            return pages[p] !== false
+          })
+          qc.invalidateQueries({
+            predicate: (query) =>
+              activePrefixes.some(
+                (prefix) => String(query.queryKey[0]).startsWith(prefix),
+              ),
+          })
+        } else {
+          // 无配置时全部刷新 (向后兼容)
+          qc.invalidateQueries({
+            predicate: (query) =>
+              SSE_INVALIDATE_PREFIXES.some(
+                (prefix) => String(query.queryKey[0]).startsWith(prefix),
+              ),
+          })
+        }
+      })
+
+      es.addEventListener('depth_updated', () => {
+        // 五档修正完成: 刷新连板梯队 + 看板封单数据。
+        // 不受实时行情开关限制 — 修正轮询独立于行情轮询, 用户开了修正就想看实时封单。
+        qc.invalidateQueries({ queryKey: ['limit-ladder'] })
+        qc.invalidateQueries({ queryKey: ['overview-market'] })
+      })
+
+      es.addEventListener('strategy_alert', (e: MessageEvent) => {
+        try {
+          const data = JSON.parse(e.data)
+            const alerts: StrategyAlertEvent[] = data.alerts || []
+            if (alerts.length > 0) {
+              handleAlerts(alerts)
+              // 实时刷新触发记录列表 + 监控中心徽标
+              qc.invalidateQueries({ queryKey: ['alerts'] })
+              qc.invalidateQueries({ queryKey: ['alerts-total'] })
+            }
+        } catch {
+          // 忽略解析错误
+        }
+      })
+
+      // 定时复盘流式进度: 后端到点生成时把 meta/delta/done 推来, 喂进 reviewStore
+      // 开着复盘页可看到「边生成边显示」, 切走再回来也能看到生成中/已生成
+      es.addEventListener('review_progress', (e: MessageEvent) => {
+        try {
+          const evt = JSON.parse(e.data)
+          feedReviewEvent(evt)
+          // done(后端已归档) → 刷新历史列表, 让新报告出现并可查看
+          if (evt.type === 'done') {
+            qc.invalidateQueries({ queryKey: QK.reviewReports })
+          }
+        } catch {
+          // 忽略解析错误
+        }
+      })
+
+      es.onerror = () => {
+        es.close()
+        esRef.current = null
+        const delay = getQueryConfig().sse.reconnectDelay
+        retryRef.current = setTimeout(connect, delay)
+      }
+    }
+
+    connect()
+
+    return () => {
+      clearTimeout(retryRef.current)
+      if (esRef.current) {
+        esRef.current.close()
+        esRef.current = null
+      }
+    }
+  }, [qc, handleAlerts])
+}