인프런 커뮤니티 질문&답변

강의에서는 비동기적으로 engine을 만들때, 아래 처럼

engine = create_async_engine(
    DATABASE_URL,
    echo=True,
    future=True, # SQLAlchemy 2.0
    # encoding="utf-8"
)

했는데.... poolclass=QueuePool를 사용하여, DB 연결을 풀링하여 재사용하는 것이 도움이 될 듯한데... 강의에서는 언급이 없으셔서, AI에게 물어봤더니...

"FastAPI에서 create_async_engine과 함께 QueuePool을 사용하는 것은 비동기 데이터베이스 연결을 효율적으로 관리하기 위함입니다. QueuePool은 데이터베이스 연결을 풀링하여 재사용함으로써, 매번 새로운 연결을 생성하고 종료하는 오버헤드를 줄여줍니다. 특히 FastAPI와 같이 비동기 요청을 처리하는 환경에서는 여러 연결을 동시에 관리해야 하므로, 풀링은 성능 향상에 큰 도움이 됩니다."

이렇게 답을 주고...

from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy.pool import QueuePool

DATABASE_URL = "your_database_url"  # 예: postgresql+asyncpg://user:password@host:port/database

engine = create_async_engine(
    DATABASE_URL,
    poolclass=QueuePool,
    pool_size=10,  # 풀에 유지할 연결 수
    max_overflow=20, # 풀 크기가 초과될 때 허용할 추가 연결 수
    pool_recycle=3600, # 초 단위로 연결을 재활용할 시간 (초과 시 재연결)
    echo=True  # SQL 실행 로그 출력 (디버깅용)
)

# 이후 엔진을 사용하여 데이터베이스 작업 수행

예시코드로 이렇게 주던데...

실제로, 비동기적으로 engine을 만들때에 AI의 답변처럼 도움이 되나요?

저희 강의의 코드는 아래와 같은 흐름을 타고 있던데...

from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
from sqlalchemy.orm import sessionmaker, declarative_base
from typing import AsyncGenerator

from sqlalchemy.orm import configure_mappers

configure_mappers()

DATABASE_URL = "sqlite+aiosqlite:///./sql_app.db?charset=utf8"

engine = create_async_engine(
    DATABASE_URL,
    echo=True,
    future=True, # SQLAlchemy 2.0
    # encoding="utf-8"
)

from sqlalchemy.ext.asyncio import async_sessionmaker

AsyncSessionLocal = async_sessionmaker(
    engine,
    class_=AsyncSession, # add
    expire_on_commit=False,
    autocommit=False,
    autoflush=False
)


Base = declarative_base()

async def get_db() -> AsyncGenerator[AsyncSession, None]:
    session: AsyncSession = AsyncSessionLocal()
    print(f"async def get_db(): Session created: {session}")
    try:
        yield session
    except Exception as e:
        print(f"Session rollback triggered due to exception: {e}")
        await session.rollback()
        raise
    finally:
        print(f"async def get_db(): Closing session: {session}")
        await session.close()