- 删除 ValidationQueue 双轨持久化队列,替换为纯内存 AsyncWorkerPool - 引入统一后台任务框架 JobExecutor(Job/CrawlJob/ValidateAllJob) - 新增 PluginRunner 统一插件执行(超时、重试、健康检查、统计) - 重构 SchedulerService 职责收敛为仅定时触发 ValidateAllJob - 使用 AsyncExitStack 重构 lifespan,安全管理长生命周期资源 - 路由层瘦身 50%+,业务异常上抛由全局中间件统一处理 - 实现设置全热更新(WorkerPool 并发、Validator 超时即时生效) - 前端 Store 强制写后重新拉取,消除乐观更新数据不同步 - 删除 queue.py / task_repo.py / task_service.py - 新增 execution 单元测试,全部 85 个测试通过
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""FastAPI 应用工厂"""
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from app.api.lifespan import lifespan
|
|
from app.api.routes import api_router
|
|
from app.api.errors import proxy_pool_exception_handler, pydantic_validation_handler, general_exception_handler
|
|
from app.core.exceptions import ProxyPoolException
|
|
from pydantic import ValidationError
|
|
from app.core.config import settings as app_settings
|
|
|
|
# 导入并注册所有插件(显式注册模式)
|
|
import app.plugins
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
app = FastAPI(
|
|
title="代理池API",
|
|
version="2.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
# CORS
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=app_settings.cors_origins,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# 异常处理
|
|
app.add_exception_handler(ProxyPoolException, proxy_pool_exception_handler)
|
|
app.add_exception_handler(ValidationError, pydantic_validation_handler)
|
|
app.add_exception_handler(Exception, general_exception_handler)
|
|
|
|
# 路由
|
|
app.include_router(api_router)
|
|
|
|
@app.get("/")
|
|
async def root():
|
|
return {"message": "欢迎使用代理池API", "status": "running", "data": None}
|
|
|
|
@app.get("/health")
|
|
async def health_check():
|
|
from datetime import datetime
|
|
scheduler = app.state.scheduler
|
|
return {
|
|
"status": "healthy",
|
|
"timestamp": datetime.now().isoformat(),
|
|
"database": "connected",
|
|
"scheduler": "running" if scheduler.running else "stopped",
|
|
"version": "2.0.0",
|
|
}
|
|
|
|
return app
|