mirror of
https://github.com/grillazz/fastapi-sqlalchemy-asyncpg.git
synced 2025-08-26 16:40:40 +03:00
drop experimental async scoped db conn
This commit is contained in:
parent
16fb7b5406
commit
2e8fbb5ab8
@ -1,11 +1,7 @@
|
||||
from asyncio import current_task
|
||||
|
||||
from typing import AsyncGenerator
|
||||
|
||||
from fastapi import HTTPException
|
||||
from fastapi.encoders import jsonable_encoder
|
||||
from sqlalchemy.exc import SQLAlchemyError
|
||||
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine, async_scoped_session
|
||||
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from app import config
|
||||
@ -23,21 +19,9 @@ 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)
|
||||
AsyncScopedSession = async_scoped_session(async_session_factory, scopefunc=current_task)
|
||||
|
||||
|
||||
# Dependency
|
||||
async def get_db() -> AsyncGenerator:
|
||||
async with async_session_factory() as session:
|
||||
try:
|
||||
yield session
|
||||
await session.commit()
|
||||
except SQLAlchemyError as sql_ex:
|
||||
await session.rollback()
|
||||
raise sql_ex
|
||||
except HTTPException as http_ex:
|
||||
await session.rollback()
|
||||
raise http_ex
|
||||
finally:
|
||||
await session.close()
|
||||
|
||||
yield session
|
||||
|
Loading…
x
Reference in New Issue
Block a user