init

2026-03-04 01:24:09 +08:00
parent 1ca83248a2
commit 9357a46902
2 changed files with 14 additions and 1 deletions
@@ -67,6 +67,14 @@ class DatabaseManager:
            self.is_connected = False
            logger.info("数据库连接已断开")
    
+    async def session(self):
+        """获取数据库会话的异步上下文管理器"""
+        if not self.is_connected:
+            await self.connect()
+        
+        async with self.async_session() as session:
+            yield session
+    
    async def get_session(self) -> AsyncSession:
        """获取数据库会话"""
        if not self.is_connected: