feat: 综合平台services多服务支持+SW网络优先缓存修复+适配层/定时任务完善

app/database.py

@@ -17,12 +17,30 @@ ASSETS_DB_URL = f"sqlite:///{DATA_DIR / 'assets.db'}"
 METRICS_DB_URL = f"sqlite:///{DATA_DIR / 'metrics.db'}"
 
 assets_engine = create_engine(
-    ASSETS_DB_URL, echo=False, connect_args={"check_same_thread": False}
+    ASSETS_DB_URL,
+    echo=False,
+    connect_args={"check_same_thread": False},
+    pool_pre_ping=True,
 )
 metrics_engine = create_engine(
-    METRICS_DB_URL, echo=False, connect_args={"check_same_thread": False}
+    METRICS_DB_URL,
+    echo=False,
+    connect_args={"check_same_thread": False},
+    pool_pre_ping=True,
 )
 
+
+def _enable_wal(engine) -> None:
+    """启用 WAL 模式，提升 SQLite 并发读写能力"""
+    import sqlalchemy as sa
+    with engine.connect() as conn:
+        conn.execute(sa.text("PRAGMA journal_mode=WAL"))
+        conn.execute(sa.text("PRAGMA busy_timeout=5000"))
+
+
+_enable_wal(assets_engine)
+_enable_wal(metrics_engine)
+
 # 兼容旧代码：默认 engine 指向资产库
 engine = assets_engine
 
@@ -37,8 +55,9 @@ def init_db() -> None:
         ServerInfo,
     )
     from app.models.provider import Provider  # noqa: F401
+    from app.models.ssl import SiteCert, Subdomain  # noqa: F401
 
-    asset_models = [Provider, Asset, VPSDetail, DomainDetail, AIAccount, CloudflareDetail]
+    asset_models = [Provider, Asset, VPSDetail, DomainDetail, AIAccount, CloudflareDetail, Subdomain, SiteCert]
     metric_models = [MetricPoint, ServerInfo, SecurityCheck, EventLog]
 
     SQLModel.metadata.create_all(
@@ -48,6 +67,7 @@ def init_db() -> None:
         metrics_engine, tables=[m.__table__ for m in metric_models]
     )
     _migrate_assets_db()
+    _migrate_indexes()
 
 
 def _migrate_assets_db() -> None:
@@ -68,10 +88,67 @@ def _migrate_assets_db() -> None:
             cols = {c["name"] for c in insp.get_columns("providers")}
             if "last_synced_at" not in cols:
                 conn.execute(sa.text("ALTER TABLE providers ADD COLUMN last_synced_at DATETIME"))
+            if "services" not in cols:
+                conn.execute(sa.text("ALTER TABLE providers ADD COLUMN services VARCHAR DEFAULT ''"))
+            # 为已有预设平台补充 services（仅补空值，用户自定义行不动）
+            _backfill_provider_services(conn)
         if insp.has_table("ai_accounts"):
             cols = {c["name"] for c in insp.get_columns("ai_accounts")}
             if "api_key_encrypted" not in cols:
                 conn.execute(sa.text("ALTER TABLE ai_accounts ADD COLUMN api_key_encrypted VARCHAR"))
+            # 迁移：将明文 api_key 加密后存入 api_key_encrypted，并清空原字段
+            _migrate_plaintext_api_keys(conn)
+
+
+def _migrate_indexes() -> None:
+    """为已有数据库补充复合索引（create_all 不会为已存在的表补索引，IF NOT EXISTS 幂等）"""
+    import sqlalchemy as sa
+
+    stmts = [
+        (assets_engine, "CREATE INDEX IF NOT EXISTS ix_assets_provider_ext_type ON assets (provider_id, external_id, asset_type)"),
+        (metrics_engine, "CREATE INDEX IF NOT EXISTS ix_metric_points_asset_ts ON metric_points (asset_id, ts)"),
+        (metrics_engine, "CREATE INDEX IF NOT EXISTS ix_security_checks_asset_ts ON security_checks (asset_id, ts)"),
+    ]
+    for eng, sql in stmts:
+        with eng.begin() as conn:
+            conn.execute(sa.text(sql))
+
+
+def _migrate_plaintext_api_keys(conn) -> None:
+    """一次性迁移：将 ai_accounts 中残留的明文 api_key 加密后存入 api_key_encrypted，并清空原字段"""
+    import sqlalchemy as sa
+    from app.core.crypto import encrypt
+
+    rows = conn.execute(
+        sa.text("SELECT id, api_key FROM ai_accounts WHERE api_key IS NOT NULL AND api_key != ''")
+    ).fetchall()
+    if not rows:
+        return
+    for row in rows:
+        encrypted = encrypt(row[1])
+        if encrypted:
+            conn.execute(
+                sa.text("UPDATE ai_accounts SET api_key_encrypted = :enc, api_key = NULL WHERE id = :id"),
+                {"enc": encrypted, "id": row[0]},
+            )
+
+
+def _backfill_provider_services(conn) -> None:
+    """为已有预设平台补充 services 服务列表（只更新 services 为空的预设 slug）"""
+    import sqlalchemy as sa
+    from app.core.seed import PRESET_PROVIDERS
+
+    for data in PRESET_PROVIDERS:
+        services = data.get("services")
+        if not services:
+            continue
+        conn.execute(
+            sa.text(
+                "UPDATE providers SET services = :services "
+                "WHERE slug = :slug AND (services IS NULL OR services = '')"
+            ),
+            {"services": services, "slug": data["slug"]},
+        )
 
 
 def get_session() -> Generator[Session, None, None]: