wrap db session with context manager

the_app/database.py

@@ -1,5 +1,6 @@
 from typing import AsyncGenerator
 
+from fastapi import HTTPException
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
 from sqlalchemy.orm import sessionmaker
@@ -22,13 +23,15 @@ async_session = sessionmaker(engine, expire_on_commit=False, class_=AsyncSession
 
 # Dependency
 async def get_db() -> AsyncGenerator:
-    session = async_session()
+    async with async_session() as session:
-    try:
+        try:
-        yield session
+            yield session
-        await session.commit()
+            await session.commit()
-    except SQLAlchemyError as ex:
+        except SQLAlchemyError as sql_ex:
-        await session.rollback()
+            await session.rollback()
-        raise ex
+            raise sql_ex
-    finally:
+        except HTTPException as http_ex:
-        await session.close()
+            session.rollback()
-
+            raise http_ex
+        finally:
+            await session.close()