refactor(db): 修复异步连接池无界增长，并完成 SQLAlchemy 2.0 迁移与分层归位 (#6320)

app/startup/database_initializer.py

@@ -0,0 +1,50 @@
+from configparser import ConfigParser as _ConfigParser
+import traceback
+
+from alembic.command import upgrade
+from alembic.config import Config
+
+from app.runtime.config import settings
+from app.db import Base
+from app.runtime.log import logger
+
+
+def init_db():
+    """
+    初始化数据库
+    """
+    # 函数内导入而非模块级：写成模块级会让 import 本模块的一方也被迫拉起引擎模块。
+    # 引擎一律用 get_engine() 取——旧名字 `app.db.Engine` 只为仓库外插件保留，且它一经
+    # 属性访问就把引擎建出来，模块级写法会使本模块反过来依赖「数据库已在别处初始化完成」。
+    from app.db.engine import get_engine
+
+    # 确保所有模型都已注册到 Base.metadata 中
+    import app.db.models  # noqa: F401
+
+    # 全量建表
+    Base.metadata.create_all(bind=get_engine())
+
+
+def update_db():
+    """
+    更新数据库
+    """
+    script_location = settings.ROOT_PATH / 'database'
+    try:
+        alembic_cfg = Config()
+        alembic_cfg.file_config = _ConfigParser(interpolation=None)
+        alembic_cfg.set_main_option('script_location', str(script_location))
+        
+        # 与引擎构建使用同一套 URL 推导：两处各自拼接会在配置变更时悄悄漂移，
+        # 导致迁移连到与应用不同的库上
+        db_url = settings.DB_SQLITE_URL() if settings.DB_TYPE.lower() != "postgresql" \
+            else settings.DB_POSTGRESQL_URL()
+            
+        alembic_cfg.set_main_option('sqlalchemy.url', db_url)
+        upgrade(alembic_cfg, 'head')
+    except Exception as error:
+        logger.error(
+            f'数据库更新失败：{str(error)} - {traceback.format_exc()}'
+        )
+        raise
+