refactor(db): 将数据库操作异步化
- 将所有 session.add() 改为 await session.add() - 将所有 session.commit() 改为 await session.commit() - 将 session.refresh() 改为 await session.refresh()
This commit is contained in:
@@ -43,7 +43,7 @@ class SQLAlchemyTransaction:
|
||||
|
||||
def __exit__(self, exc_type, exc_val, exc_tb):
|
||||
if exc_type is None:
|
||||
self.session.commit()
|
||||
self.await session.commit()
|
||||
else:
|
||||
self.session.rollback()
|
||||
self.session.close()
|
||||
|
||||
@@ -168,7 +168,7 @@ async def db_query(
|
||||
|
||||
# 创建新记录
|
||||
new_record = model_class(**data)
|
||||
session.add(new_record)
|
||||
await session.add(new_record)
|
||||
await session.flush() # 获取自动生成的ID
|
||||
|
||||
# 转换为字典格式返回
|
||||
@@ -295,7 +295,7 @@ async def db_save(
|
||||
|
||||
# 创建新记录
|
||||
new_record = model_class(**data)
|
||||
session.add(new_record)
|
||||
await session.add(new_record)
|
||||
await session.flush()
|
||||
|
||||
# 转换为字典格式返回
|
||||
|
||||
@@ -676,7 +676,6 @@ async def get_db_session() -> AsyncGenerator[AsyncSession, None]:
|
||||
raise RuntimeError("Database session not initialized")
|
||||
session = SessionLocal()
|
||||
yield session
|
||||
# await session.commit()
|
||||
except Exception:
|
||||
if session:
|
||||
await session.rollback()
|
||||
|
||||
Reference in New Issue
Block a user