106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
from typing import AsyncGenerator
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
|
|
from sqlalchemy.orm import DeclarativeBase
|
|
|
|
from config import db_settings
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
# =============================================================================
|
|
# 메인 엔진 (FastAPI 요청용)
|
|
# =============================================================================
|
|
engine = create_async_engine(
|
|
url=db_settings.MYSQL_URL,
|
|
echo=False,
|
|
pool_size=20, # 기본 풀 크기: 20
|
|
max_overflow=20, # 추가 연결: 20 (총 최대 40)
|
|
pool_timeout=30, # 풀에서 연결 대기 시간 (초)
|
|
pool_recycle=3600, # 1시간마다 연결 재생성
|
|
pool_pre_ping=True, # 연결 유효성 검사
|
|
pool_reset_on_return="rollback", # 반환 시 롤백으로 초기화
|
|
connect_args={
|
|
"connect_timeout": 10, # DB 연결 타임아웃
|
|
"charset": "utf8mb4",
|
|
},
|
|
)
|
|
|
|
# 메인 세션 팩토리 (FastAPI DI용)
|
|
AsyncSessionLocal = async_sessionmaker(
|
|
bind=engine,
|
|
class_=AsyncSession,
|
|
expire_on_commit=False,
|
|
autoflush=False, # 명시적 flush 권장
|
|
)
|
|
|
|
|
|
# =============================================================================
|
|
# 백그라운드 태스크 전용 엔진 (메인 풀과 분리)
|
|
# =============================================================================
|
|
background_engine = create_async_engine(
|
|
url=db_settings.MYSQL_URL,
|
|
echo=False,
|
|
pool_size=10, # 백그라운드용 풀 크기: 10
|
|
max_overflow=10, # 추가 연결: 10 (총 최대 20)
|
|
pool_timeout=60, # 백그라운드는 대기 시간 여유있게
|
|
pool_recycle=3600,
|
|
pool_pre_ping=True,
|
|
pool_reset_on_return="rollback",
|
|
connect_args={
|
|
"connect_timeout": 10,
|
|
"charset": "utf8mb4",
|
|
},
|
|
)
|
|
|
|
# 백그라운드 세션 팩토리
|
|
BackgroundSessionLocal = async_sessionmaker(
|
|
bind=background_engine,
|
|
class_=AsyncSession,
|
|
expire_on_commit=False,
|
|
autoflush=False,
|
|
)
|
|
|
|
|
|
async def create_db_tables():
|
|
import asyncio
|
|
|
|
# 모델 import (테이블 메타데이터 등록용)
|
|
from app.home.models import Image, Project # noqa: F401
|
|
from app.lyric.models import Lyric # noqa: F401
|
|
from app.song.models import Song # noqa: F401
|
|
from app.video.models import Video # noqa: F401
|
|
|
|
print("Creating database tables...")
|
|
|
|
async with asyncio.timeout(10):
|
|
async with engine.begin() as connection:
|
|
await connection.run_sync(Base.metadata.create_all)
|
|
|
|
|
|
# FastAPI 의존성용 세션 제너레이터
|
|
async def get_session() -> AsyncGenerator[AsyncSession, None]:
|
|
# 커넥션 풀 상태 로깅 (디버깅용)
|
|
pool = engine.pool
|
|
print(f"[get_session] Pool status - size: {pool.size()}, checked_in: {pool.checkedin()}, checked_out: {pool.checkedout()}, overflow: {pool.overflow()}")
|
|
|
|
async with AsyncSessionLocal() as session:
|
|
try:
|
|
yield session
|
|
except Exception as e:
|
|
await session.rollback()
|
|
print(f"[get_session] Session rollback due to: {e}")
|
|
raise e
|
|
finally:
|
|
# 명시적으로 세션 종료 확인
|
|
print(f"[get_session] Session closing - Pool checked_out: {pool.checkedout()}")
|
|
|
|
|
|
# 앱 종료 시 엔진 리소스 정리 함수
|
|
async def dispose_engine() -> None:
|
|
await engine.dispose()
|
|
await background_engine.dispose()
|
|
print("Database engines disposed (main + background)")
|