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 typing import AsyncGenerator
|
||||||
|
|
||||||
from fastapi import HTTPException
|
|
||||||
from fastapi.encoders import jsonable_encoder
|
from fastapi.encoders import jsonable_encoder
|
||||||
from sqlalchemy.exc import SQLAlchemyError
|
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine, async_scoped_session
|
|
||||||
from sqlalchemy.orm import sessionmaker
|
from sqlalchemy.orm import sessionmaker
|
||||||
|
|
||||||
from app import config
|
from app import config
|
||||||
@ -23,21 +19,9 @@ engine = create_async_engine(
|
|||||||
# expire_on_commit=False will prevent attributes from being expired
|
# expire_on_commit=False will prevent attributes from being expired
|
||||||
# after commit.
|
# after commit.
|
||||||
async_session_factory = sessionmaker(engine, expire_on_commit=False, class_=AsyncSession)
|
async_session_factory = sessionmaker(engine, expire_on_commit=False, class_=AsyncSession)
|
||||||
AsyncScopedSession = async_scoped_session(async_session_factory, scopefunc=current_task)
|
|
||||||
|
|
||||||
|
|
||||||
# Dependency
|
# Dependency
|
||||||
async def get_db() -> AsyncGenerator:
|
async def get_db() -> AsyncGenerator:
|
||||||
async with async_session_factory() as session:
|
async with async_session_factory() as session:
|
||||||
try:
|
yield session
|
||||||
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()
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user