format

app/database.py

@@ -1,10 +1,15 @@
-from typing import AsyncGenerator
+from collections.abc import AsyncGenerator
+from http.client import HTTPException
 
 from fastapi.encoders import jsonable_encoder
+from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
 from sqlalchemy.orm import sessionmaker
 
 from app import config
+from app.utils import get_logger
+
+logger = get_logger(__name__)
 
 global_settings = config.get_settings()
 url = global_settings.asyncpg_url
@@ -18,10 +23,28 @@ engine = create_async_engine(
 
 # expire_on_commit=False will prevent attributes from being expired
 # after commit.
-async_session_factory = sessionmaker(engine, expire_on_commit=False, class_=AsyncSession)
+AsyncSessionFactory = sessionmaker(engine, autoflush=False, expire_on_commit=False, class_=AsyncSession)
 
 
 # Dependency
 async def get_db() -> AsyncGenerator:
-    async with async_session_factory() as session:
+    async with AsyncSessionFactory() as session:
+        logger.debug(f"ASYNC Pool: {engine.pool.status()}")
         yield session
+
+
+async def get_async_db() -> AsyncGenerator:
+    try:
+        session: AsyncSession = AsyncSessionFactory()
+        logger.debug(f"ASYNC Pool: {engine.pool.status()}")
+        yield session
+    except SQLAlchemyError as sql_ex:
+        await session.rollback()
+        raise sql_ex
+    except HTTPException as http_ex:
+        await session.rollback()
+        raise http_ex
+    else:
+        await session.commit()
+    finally:
+        await session.close()