31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from collections.abc import AsyncGenerator
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
|
|
from sqlalchemy.pool import NullPool
|
|
|
|
from ..core.config import settings
|
|
|
|
# 根据不同数据库驱动调整连接池参数，确保在多数据库环境下表现稳定
|
|
engine_kwargs = {"echo": settings.debug}
|
|
if settings.is_sqlite_backend:
|
|
# SQLite 场景下禁用连接池并放宽线程检查，避免多协程读写冲突
|
|
engine_kwargs.update(
|
|
pool_pre_ping=False,
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=NullPool,
|
|
)
|
|
else:
|
|
# MySQL 场景保持健康检查与连接复用，适用于生产环境的长连接需求
|
|
engine_kwargs.update(pool_pre_ping=True, pool_recycle=3600)
|
|
|
|
engine = create_async_engine(settings.sqlalchemy_database_uri, **engine_kwargs)
|
|
|
|
# 统一的 Session 工厂，禁用 expire_on_commit 方便返回模型对象
|
|
AsyncSessionLocal = async_sessionmaker(bind=engine, expire_on_commit=False)
|
|
|
|
|
|
async def get_session() -> AsyncGenerator[AsyncSession, None]:
|
|
"""FastAPI 依赖项：提供一个作用域内共享的数据库会话。"""
|
|
async with AsyncSessionLocal() as session:
|
|
yield session
|