From 6b12b04cab7af5022ffa38cad6d89ab5821eb046 Mon Sep 17 00:00:00 2001 From: Sakurasan <26715255+Sakurasan@users.noreply.github.com> Date: Mon, 10 Aug 2026 20:20:55 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=8C=82=E8=BD=BD=E6=A0=B8=E5=BF=83?= =?UTF-8?q?=E8=82=A1=E8=B7=AF=E7=94=B1=E5=B9=B6=E5=90=AF=E5=8A=A8=E6=AF=8F?= =?UTF-8?q?=E6=97=A5=E9=87=87=E9=9B=86=E4=BB=BB=E5=8A=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude --- backend/main.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/backend/main.py b/backend/main.py index e2598a3..b6c9f7c 100644 --- a/backend/main.py +++ b/backend/main.py @@ -1,3 +1,4 @@ +import asyncio import os from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware @@ -6,7 +7,8 @@ from contextlib import asynccontextmanager from dotenv import load_dotenv from database import init_db -from routes import stock, collections, shares, sectors, themes +from routes import stock, collections, shares, sectors, themes, core_stocks +from services.daily_collector import collector_loop load_dotenv() @@ -14,7 +16,11 @@ load_dotenv() @asynccontextmanager async def lifespan(app: FastAPI): init_db() - yield + collector_task = asyncio.create_task(collector_loop()) + try: + yield + finally: + collector_task.cancel() app = FastAPI(title="AUV API", version="1.0.0", lifespan=lifespan) @@ -32,6 +38,7 @@ app.include_router(collections.router, prefix="/api/collections") app.include_router(shares.router, prefix="/api/share") app.include_router(sectors.router, prefix="/api/sectors") app.include_router(themes.router, prefix="/api/themes") +app.include_router(core_stocks.router, prefix="/api/core-stocks") # 生产模式：后端同时托管前端静态文件 # catch-all 路由在 API 路由之后注册，所以 API 优先级更高