fix issue

This commit is contained in:
EduBrain Dev
2026-04-14 21:04:00 +08:00
parent 1080bc3882
commit dbb3cf622f

View File

@@ -50,9 +50,15 @@ async_session_factory = async_sessionmaker(
async def get_db() -> AsyncGenerator[AsyncSession, None]: async def get_db() -> AsyncGenerator[AsyncSession, None]:
""" """
FastAPI 依赖项:获取异步数据库会话。 FastAPI 依赖项:获取异步数据库会话。
请求正常结束后自动 commit异常时自动 rollback。
""" """
async with async_session_factory() as session: async with async_session_factory() as session:
yield session try:
yield session
await session.commit()
except Exception:
await session.rollback()
raise
async def close_db() -> None: async def close_db() -> None: