41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
|
class TransactionContext:
|
|||
|
|
"""
|
|||
|
|
事务上下文管理器类（异步版本）
|
|||
|
|
支持异步事务管理，配合异步Db类使用
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(self, db_instance):
|
|||
|
|
"""
|
|||
|
|
初始化事务上下文管理器
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
db_instance: 异步Db实例
|
|||
|
|
"""
|
|||
|
|
self.db = db_instance
|
|||
|
|
self.session = None
|
|||
|
|
|
|||
|
|
async def __aenter__(self):
|
|||
|
|
"""
|
|||
|
|
进入异步上下文时创建会话
|
|||
|
|
"""
|
|||
|
|
self.session = await self.db.get_session()
|
|||
|
|
return self.session
|
|||
|
|
|
|||
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
|||
|
|
"""
|
|||
|
|
退出异步上下文时提交或回滚事务
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
if exc_type is None:
|
|||
|
|
# 如果没有异常，提交事务
|
|||
|
|
await self.session.commit()
|
|||
|
|
else:
|
|||
|
|
# 如果有异常，回滚事务
|
|||
|
|
await self.session.rollback()
|
|||
|
|
finally:
|
|||
|
|
# 无论如何都关闭会话
|
|||
|
|
if self.session:
|
|||
|
|
await self.session.close()
|
|||
|
|
|
|||
|
|
# 返回False表示不抑制异常
|
|||
|
|
return False
|