easy-tdx/web-ui/src/stores/backtest.ts

// 回测状态管理（Pinia）。
// 持有：策略列表、当前 OHLCV、回测结果、运行状态、错误信息。

import { defineStore } from 'pinia'
import { computed, ref } from 'vue'

import { fetchStrategies, formatError, runBacktest, submitPortfolioTask, fetchTask } from '../api'
import type {
  BacktestRequest,
  BacktestResult,
  Bar,
  PortfolioBacktestRequest,
  PortfolioResult,
  StrategySchema,
} from '../types'

export const useBacktestStore = defineStore('backtest', () => {
  // ── 策略 ─────────────────────────────────────────────────────────────────
  const strategies = ref<StrategySchema[]>([])
  const strategiesLoaded = ref(false)

  async function loadStrategies() {
    if (strategiesLoaded.value) return
    const resp = await fetchStrategies()
    strategies.value = resp.strategies
    strategiesLoaded.value = true
  }

  // ── OHLCV 行情（前端始终持有，回测与 K 线共用） ───────────────────────────
  const ohlcv = ref<Bar[]>([])
  const barsSource = ref<string>('') // 来源描述，如 "SZ:000001 DAY×250"

  function setOhlcv(bars: Bar[], source: string) {
    ohlcv.value = bars
    barsSource.value = source
  }

  const hasBars = computed(() => ohlcv.value.length >= 2)

  // ── 回测结果 ──────────────────────────────────────────────────────────────
  const result = ref<BacktestResult | null>(null)
  const running = ref(false)
  const error = ref<string>('')

  /** 运行同步回测（内联 OHLCV）。 */
  async function run(req: Omit<BacktestRequest, 'ohlcv'>) {
    if (!hasBars.value) {
      error.value = '请先取行情数据或粘贴 OHLCV'
      return
    }
    running.value = true
    error.value = ''
    try {
      const fullReq: BacktestRequest = { ...req, ohlcv: ohlcv.value }
      result.value = await runBacktest(fullReq)
    } catch (e) {
      error.value = formatError(e)
      result.value = null
    } finally {
      running.value = false
    }
  }

  function clearResult() {
    result.value = null
    error.value = ''
  }

  // ── 组合回测（Phase 3） ───────────────────────────────────────────────────
  const portfolioResult = ref<PortfolioResult | null>(null)
  const portfolioRunning = ref(false)

  /** 提交组合回测后台任务并轮询直到完成。 */
  async function runPortfolio(req: PortfolioBacktestRequest) {
    portfolioRunning.value = true
    error.value = ''
    portfolioResult.value = null
    try {
      const { task_id } = await submitPortfolioTask(req)
      // 轮询
      const start = Date.now()
      // eslint-disable-next-line no-constant-condition
      while (true) {
        const state = await fetchTask(task_id)
        if (state.status === 'done' && state.result) {
          portfolioResult.value = state.result as PortfolioResult
          break
        }
        if (state.status === 'failed') {
          throw new Error(state.error || '组合回测失败')
        }
        if (Date.now() - start > 120_000) throw new Error('组合回测超时（120s）')
        await new Promise((r) => setTimeout(r, 300))
      }
    } catch (e) {
      error.value = formatError(e)
      portfolioResult.value = null
    } finally {
      portfolioRunning.value = false
    }
  }

  function clearPortfolio() {
    portfolioResult.value = null
    error.value = ''
  }

  return {
    // state
    strategies,
    strategiesLoaded,
    ohlcv,
    barsSource,
    result,
    running,
    error,
    portfolioResult,
    portfolioRunning,
    // getters
    hasBars,
    // actions
    loadStrategies,
    setOhlcv,
    run,
    clearResult,
    runPortfolio,
    clearPortfolio,
  }
})