# database.py
# 本文件负责配置数据库连接、创建引擎、会话工厂，并提供依赖注入函数（同步模式）

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, declarative_base

from config import settings

# 1. 配置 MySQL 数据库连接 URL（从 config 读取，支持环境变量覆盖）
SQLALCHEMY_DATABASE_URL = settings.DATABASE_URL

# 2. 创建数据库引擎
#    - pool_pre_ping=True：每次取连接前先 ping，防止使用已断开的连接
#    - pool_recycle=3600：连接最长复用 1 小时，避免 MySQL wait_timeout 断连
engine = create_engine(
    SQLALCHEMY_DATABASE_URL,
    pool_pre_ping=True,
    pool_recycle=3600,
    echo=settings.DB_ECHO,  # True 会在控制台打印所有 SQL，便于调试
)

# 3. 创建会话工厂：不自动提交，需要手动 commit
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# 4. 声明式基类，所有模型类都继承自它
Base = declarative_base()


# 5. 依赖注入函数：用于 FastAPI 路由中获取数据库会话
def get_db():
    """每次请求创建一个数据库会话，请求结束后关闭"""
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()