# database.py
# 本文件负责配置数据库连接、创建引擎、会话工厂，并提供依赖注入函数。
#
# 相比原版的三点改动：
#   1. 连接串不再硬编码 root/123456，改为读 settings（环境变量 / .env）；
#   2. 引擎补上 pool_pre_ping / pool_recycle —— 解决 MySQL "server has gone away"；
#   3. expire_on_commit=False —— 避免 commit 后返回 ORM 对象时又触发隐藏查询。

import logging

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

from settings import settings

logger = logging.getLogger(__name__)

# 1. 数据库引擎
engine = create_engine(
    settings.database_url,
    echo=settings.DB_ECHO,                  # 打开后会把每条 SQL 打到日志，排查问题用
    pool_size=settings.DB_POOL_SIZE,        # 连接池常驻连接数
    max_overflow=settings.DB_MAX_OVERFLOW,
    pool_recycle=settings.DB_POOL_RECYCLE,  # 超过 N 秒的连接主动回收，避免被 MySQL 断开
    pool_pre_ping=True,                     # 取连接前先 ping 一次，自动剔除死连接
    future=True,
)


# 2. 会话工厂
#    autocommit=False：不自动提交，事务边界由 DAO 显式控制
#    expire_on_commit=False：commit 后对象不过期，API 层序列化时不会再多打一次 SELECT
SessionLocal = sessionmaker(
    autocommit=False,
    autoflush=False,
    expire_on_commit=False,
    bind=engine,
)

# 3. 声明式基类
Base = declarative_base()


# 4. 依赖注入：每个 HTTP 请求一个独立会话，请求结束必定关闭
def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()