auto-sync: 2026-04-12 10:19:30

2026-04-12 10:19:30 +08:00
parent 6ad2b82649
commit 5bce2b111d
2 changed files with 81 additions and 0 deletions
@@ -170,5 +170,6 @@



+


@@ -0,0 +1,80 @@
+"""
+自动化回测服务 - 主入口
+启动 FastAPI 服务，接受回测任务提交，执行回测，返回结果
+遵循 vnpy 原生设计，只做外层封装
+"""
+import uvicorn
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+from .config import settings
+from .api import router
+from .task_queue import task_queue
+
+
+app = FastAPI(
+    title="sanguo 自动化回测服务",
+    description="基于 vnpy 原生 BacktestingEngine 的自动化回测API服务",
+    version="1.0.0"
+)
+
+# CORS 配置
+if settings.cors_allow_all:
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=["*"],
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+# 注册API路由
+app.include_router(router, tags=["backtest"])
+
+
+# 健康检查已经在API路由中定义
+from .models import HealthCheckResponse
+from .task_queue import task_queue
+
+
+@app.get("/api/backtest/health", summary="服务健康检查", response_model=ApiResponse[HealthCheckResponse])
+def health():
+    """服务健康检查，返回当前任务统计"""
+    from .models import ApiResponse, HealthCheckResponse
+    return ApiResponse(
+        code=0,
+        msg="ok",
+        data=HealthCheckResponse(
+            pending_count=len(task_queue.pending_tasks),
+            running_count=len(task_queue.running_tasks),
+            completed_count=len(task_queue.completed_tasks),
+            failed_count=len(task_queue.failed_tasks),
+            max_workers=settings.max_workers
+        )
+    )
+
+
+def main():
+    """启动服务"""
+    # 启动工作进程池
+    task_queue.start_worker_pool()
+    
+    print(f"自动化回测服务启动中...")
+    print(f"监听地址: http://{settings.host}:{settings.port}")
+    print(f"最大并发回测数: {settings.max_workers}")
+    print(f"API 文档: http://{settings.host}:{settings.port}/docs")
+    
+    try:
+        uvicorn.run(
+            "main:app",
+            host=settings.host,
+            port=settings.port,
+            reload=settings.debug,
+        )
+    finally:
+        # 关闭进程池
+        task_queue.close_worker_pool()
+
+
+if __name__ == "__main__":
+    main()