Function lifespan

main.py:37–101

(app: FastAPI)

@asynccontextmanager
async def lifespan(app: FastAPI):
    # 初始化 Token 数据库
    from app.services.request_log_dao import init_request_log_dao
    from app.services.token_automation import (
        run_directory_import,
        start_token_automation_scheduler,
        stop_token_automation_scheduler,
    )
    from app.services.token_dao import init_token_database

    await init_token_database()
    init_request_log_dao()

    if (
        settings.TOKEN_AUTO_IMPORT_ENABLED
        and settings.TOKEN_AUTO_IMPORT_SOURCE_DIR.strip()
    ):
        try:
            await run_directory_import(
                settings.TOKEN_AUTO_IMPORT_SOURCE_DIR,
                provider="zai",
            )
            logger.info("✅ 启动阶段已完成一次目录自动导入")
        except Exception as exc:
            logger.warning(f"⚠️ 启动阶段目录自动导入失败: {exc}")

    # 从数据库初始化认证 token 池
    from app.utils.token_pool import initialize_token_pool_from_db

    token_pool = await initialize_token_pool_from_db(
        provider="zai",
        failure_threshold=settings.TOKEN_FAILURE_THRESHOLD,
        recovery_timeout=settings.TOKEN_RECOVERY_TIMEOUT,
    )

    if not token_pool and not settings.ANONYMOUS_MODE:
        logger.warning(
            "⚠️ 未找到可用 Token 且未启用匿名模式，服务可能无法正常工作"
        )

    if settings.ANONYMOUS_MODE:
        from app.utils.guest_session_pool import initialize_guest_session_pool

        guest_pool = await initialize_guest_session_pool(
            pool_size=settings.GUEST_POOL_SIZE,
        )
        guest_status = guest_pool.get_pool_status()
        logger.info(
            "🫥 匿名会话池已就绪: "
            f"{guest_status.get('valid_sessions', 0)} 个可用会话"
        )

    await warmup_upstream_client()
    await start_token_automation_scheduler()

    yield

    logger.info("🔄 应用正在关闭...")