ETest-Vue-FastAPI/ruoyi-fastapi-backend/config/get_db.py

from config.database import async_engine, AsyncSessionLocal, Base
from config.migration import run_database_migrations
from utils.log_util import logger


async def get_db():
    """
    每一个请求处理完毕后会关闭当前连接，不同的请求使用不同的连接

    :return:
    """
    async with AsyncSessionLocal() as current_db:
        yield current_db


async def get_db_session():
    """
    获取数据库会话（用于非请求上下文）

    :return:
    """
    async with AsyncSessionLocal() as current_db:
        yield current_db


async def init_create_table():
    """
    应用启动时初始化数据库连接

    :return:
    """
    logger.info('初始化数据库连接...')
    async with async_engine.begin() as conn:
        await conn.run_sync(Base.metadata.create_all)
    logger.info('数据库连接成功')
    
    # 执行数据库迁移
    logger.info('开始执行数据库迁移...')
    await run_database_migrations()
    logger.info('数据库迁移完成')