Simon
0ea229d766
- 数据表：job / experiment（spec/result JSON 存档、code_version），Alembic 迁移 53113c80257f
- Job：queued→running→(success|failed) 状态机，BackgroundTasks 本地执行 + 失败兜底标记；结果与 Experiment 关联
- Experiment：每次研究成功自动归档（含 git commit 与收益摘要），支持一键复跑（同 spec 重建 Job）
- API：POST /api/jobs、GET /api/jobs/{id}（内嵌结果）、SSE /api/jobs/{id}/events、/api/experiments 列表/详情/rerun
- 前端：新增「实验」页（列表 / 详情 / 复跑 + Job 轮询）；导航更新
- 端到端验证：真实 20 股 job 提交→后台执行→success→EXP 归档（-12.41%）；executor 成功/失败路径单测
- 测试 73 passed（新增 5 项 Job/Experiment）/ ruff clean / 前端 tsc + build 通过
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""API 依赖注入：Repository / 研究服务的装配点（composition root 的一部分）。
|
|
|
|
路由层统一使用 Annotated 注入（FastAPI 推荐写法，配合 ruff B008 无冲突）。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Annotated
|
|
|
|
from fastapi import Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.domain.repositories.jobs import ExperimentRepository, JobRepository
|
|
from app.domain.repositories.market import (
|
|
DailyBarRepository,
|
|
StockRepository,
|
|
)
|
|
from app.infrastructure.persistence.sqlalchemy.repositories.market_impl import (
|
|
SqlAlchemyDailyBarRepository,
|
|
SqlAlchemyStockRepository,
|
|
)
|
|
from app.infrastructure.persistence.sqlalchemy.session import get_session
|
|
from app.quant.engine import LocalEngine, QuantEngine
|
|
from app.quant.service import ResearchService
|
|
|
|
DbSession = Annotated[Session, Depends(get_session)]
|
|
|
|
|
|
def _stock_repo_factory(session: DbSession) -> StockRepository:
|
|
return SqlAlchemyStockRepository(session)
|
|
|
|
|
|
def _daily_repo_factory(session: DbSession) -> DailyBarRepository:
|
|
return SqlAlchemyDailyBarRepository(session)
|
|
|
|
|
|
def _engine_factory() -> QuantEngine:
|
|
return LocalEngine()
|
|
|
|
|
|
def _service_factory(
|
|
stock_repo: Annotated[StockRepository, Depends(_stock_repo_factory)],
|
|
daily_repo: Annotated[DailyBarRepository, Depends(_daily_repo_factory)],
|
|
engine: Annotated[QuantEngine, Depends(_engine_factory)],
|
|
) -> ResearchService:
|
|
return ResearchService(stock_repo, daily_repo, engine)
|
|
|
|
|
|
StockRepoDep = Annotated[StockRepository, Depends(_stock_repo_factory)]
|
|
DailyRepoDep = Annotated[DailyBarRepository, Depends(_daily_repo_factory)]
|
|
EngineDep = Annotated[QuantEngine, Depends(_engine_factory)]
|
|
ResearchServiceDep = Annotated[ResearchService, Depends(_service_factory)]
|
|
|
|
|
|
def _job_repo_factory(session: DbSession):
|
|
from app.infrastructure.persistence.sqlalchemy.repositories.jobs_impl import (
|
|
SqlAlchemyJobRepository,
|
|
)
|
|
|
|
return SqlAlchemyJobRepository(session)
|
|
|
|
|
|
def _experiment_repo_factory(session: DbSession):
|
|
from app.infrastructure.persistence.sqlalchemy.repositories.jobs_impl import (
|
|
SqlAlchemyExperimentRepository,
|
|
)
|
|
|
|
return SqlAlchemyExperimentRepository(session)
|
|
|
|
|
|
JobRepoDep = Annotated[JobRepository, Depends(_job_repo_factory)]
|
|
ExperimentRepoDep = Annotated[ExperimentRepository, Depends(_experiment_repo_factory)]
|