Kris
ff36b9f334
新增消息日志、发件箱、会话等领域模型，新增多类型业务异常定义，添加工厂协议端口与数据库工作单元实现，同时补充飞书、钩子、Web渠道的请求验证器，以及数据转换类和异常翻译工具
18 lines
414 B
Python
18 lines
414 B
Python
from __future__ import annotations
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
|
|
class SqlAlchemyUnitOfWork:
|
|
def __init__(self, session: AsyncSession):
|
|
self._session = session
|
|
|
|
async def commit(self) -> None:
|
|
await self._session.commit()
|
|
|
|
async def rollback(self) -> None:
|
|
await self._session.rollback()
|
|
|
|
def session(self) -> AsyncSession:
|
|
return self._session
|