分模候选方案

This commit is contained in:
2026-04-23 23:57:34 +08:00
parent c1f6b6e827
commit f9acdb767b
5 changed files with 94 additions and 35 deletions
+23 -12
View File
@@ -1,16 +1,27 @@
from fastapi import APIRouter
import importlib
from api.v1.health_router import router as health_router
from api.v1.upload_router import router as upload_router
from api.v1.task_router import router as task_router
from api.v1.history_router import router as history_router
from api.v1.debug_router import router as debug_router
from api.v1.advanced_router import router as advanced_router
from utils.logger import get_logger
logger = get_logger(__name__)
router = APIRouter()
router.include_router(health_router)
router.include_router(upload_router)
router.include_router(task_router)
router.include_router(history_router)
router.include_router(debug_router)
router.include_router(advanced_router)
def _safe_include(module_path: str, label: str):
try:
module = importlib.import_module(module_path)
router_obj = getattr(module, "router", None)
if router_obj is None:
raise ValueError("未找到 router 对象")
router.include_router(router_obj)
logger.info(f"{label} 路由加载成功")
except Exception as exc:
logger.warning(f"{label} 路由加载失败,已跳过: {exc}")
_safe_include("api.v1.health_router", "健康检查")
_safe_include("api.v1.upload_router", "上传")
_safe_include("api.v1.task_router", "任务")
_safe_include("api.v1.history_router", "历史")
_safe_include("api.v1.debug_router", "调试")
_safe_include("api.v1.advanced_router", "高级")
-2
View File
@@ -15,7 +15,6 @@ from sqlalchemy.ext.asyncio import AsyncSession
from services.auth_service import get_current_active_user
from services.processing_service import processing_service
from models.database import User
from api.routes import tasks as legacy_tasks
logger = get_logger(__name__)
@@ -66,7 +65,6 @@ async def upload_stp(
upload_time=str(datetime.now())
)
await redis_task_manager.set_task(task_id, task_info)
legacy_tasks[task_id] = task_info
# 后台处理(使用独立数据库会话,避免请求会话关闭问题)
background_tasks.add_task(