bd_ai_fastapi/app/__init__.py

import os
import logging
from contextlib import asynccontextmanager
from fastapi import FastAPI
from tortoise import Tortoise
from redis.asyncio import Redis
from fastapi.staticfiles import StaticFiles
from aiomysql import create_pool, OperationalError as AiomysqlOperationalError

from app.core.exceptions import SettingNotFound
from app.core.init_app import (
    init_data,
    make_middlewares,
    register_exceptions,
    register_routers,
)

try:
    from app.settings.config import settings
except ImportError:
    raise SettingNotFound("Can not import settings")

# 配置日志
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


@asynccontextmanager
async def lifespan(app: FastAPI):
    # 初始化资源
    app.state.mysql_pool = None
    app.state.redis_client = None

    try:
        # 初始化 MySQL 连接池
        app.state.mysql_pool = await create_pool(
            host=settings.FLOW_MYSQL_HOST,
            port=settings.FLOW_MYSQL_PORT,
            user=settings.FLOW_MYSQL_USER,
            password=settings.FLOW_MYSQL_PASSWORD,
            db=settings.FLOW_MYSQL_DB,
            # 核心并发参数
            minsize=15,  # 初始保持20个连接
            maxsize=50,  # 最大连接数
            # 连接管理优化
            pool_recycle=100,  # 180秒回收连接
            connect_timeout=15,  # 连接超时时间
            # 其他优化
            charset='utf8mb4',
            echo=False,
        )

        # 验证连接有效性
        async with app.state.mysql_pool.acquire() as conn:
            await conn.ping()
        logger.info("✅ MySQL 连接池初始化成功")

    except AiomysqlOperationalError as e:
        logger.error(f"❌ MySQL 连接池初始化失败: 数据库操作错误 - {str(e)}")
        raise
    except Exception as e:
        logger.error(f"❌ MySQL 连接池初始化失败: 未知错误 - {str(e)}")
        raise

    try:
        # 初始化 Redis 连接
        app.state.redis_client = Redis(
            host=settings.REDIS_HOST,
            port=settings.REDIS_PORT,
            db=settings.REDIS_DB,
            decode_responses=True,
            password=settings.REDIS_PASSWORD,
            socket_connect_timeout=5,
            socket_keepalive=True,
            retry_on_timeout=True,
            # 连接池参数
            max_connections=1024,  # Redis连接池最大连接数
            health_check_interval=30,
            retry_on_error=[ConnectionError, TimeoutError],  # 重试错误类型
            single_connection_client=False,  # 不使用单连接客户端
            auto_close_connection_pool=True  # 自动关闭连接池
        )

        await app.state.redis_client.ping()
        logger.info("✅ Redis 连接成功")

    except Exception as e:
        logger.error(f"❌ Redis 连接失败: {str(e)}")

    try:
        await init_data()
    except Exception as e:
        logger.error(f"❌ 初始化数据失败: {str(e)}")
        raise

    yield

    # 资源清理
    if app.state.redis_client:
        try:
            await app.state.redis_client.close()
            logger.info("🛑 Redis 连接已关闭")
        except Exception as e:
            logger.warning(f"⚠️ Redis 关闭失败: {str(e)}")

    if app.state.mysql_pool:
        try:
            app.state.mysql_pool.close()
            await app.state.mysql_pool.wait_closed()
            logger.info("🛑 MySQL 连接池已关闭")
        except Exception as e:
            logger.warning(f"⚠️ MySQL 连接池关闭失败: {str(e)}")

    try:
        await Tortoise.close_connections()
        logger.info("🛑 Tortoise 连接已关闭")
    except Exception as e:
        logger.warning(f"⚠️ Tortoise 连接关闭失败: {str(e)}")


def create_app() -> FastAPI:
    app = FastAPI(
        title=settings.APP_TITLE,
        description=settings.APP_DESCRIPTION,
        version=settings.VERSION,
        openapi_url="/openapi.json",
        middleware=make_middlewares(),
        lifespan=lifespan,
    )

    static_dir = os.path.join(settings.BASE_DIR, 'web', 'public', 'resource')
    if os.path.exists(static_dir):
        app.mount("/resource", StaticFiles(directory=static_dir), name="resource")
    else:
        logger.warning(f"⚠️ 静态文件目录不存在: {static_dir}")

    register_exceptions(app)
    register_routers(app, prefix="/api")
    return app


app = create_app()