Update Middleware

2024-11-17 23:42:09 +03:00
parent 6b2914bff4
commit 7e2f56f20b
8 changed files with 149 additions and 97 deletions
@@ -14,9 +14,10 @@ class DatabaseMiddleware(BaseMiddleware):
        event: TelegramObject,
        data: Dict[str, Any],
    ) -> Any:
-        session = await asyncpg.connect(DATABASE_URL)
-        data["session"] = session
-        try:
-            return await handler(event, data)
-        finally:
-            await session.close()
+        async with await asyncpg.create_pool(DATABASE_URL) as pool:
+            async with pool.acquire() as session:
+                data["session"] = session
+                try:
+                    return await handler(event, data)
+                finally:
+                    await pool.release(session)