支持 MySQL 迁移与性能优化

web/backend/migrations/versions/20260725_0009_database_performance_indexes.py

@@ -0,0 +1,67 @@
+"""补充高频查询组合索引
+
+Revision ID: 20260725_0009
+Revises: 20260724_0008
+Create Date: 2026-07-25
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision: str = "20260725_0009"
+down_revision: Union[str, None] = "20260724_0008"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def _has_table(bind, table_name: str) -> bool:
+    return sa.inspect(bind).has_table(table_name)
+
+
+def _indexes(bind, table_name: str) -> set[str]:
+    if not _has_table(bind, table_name):
+        return set()
+    return {index["name"] for index in sa.inspect(bind).get_indexes(table_name)}
+
+
+def _create_index_if_missing(bind, name: str, table_name: str, columns: list[str]) -> None:
+    if name not in _indexes(bind, table_name):
+        op.create_index(name, table_name, columns)
+
+
+def _drop_index_if_exists(bind, name: str, table_name: str) -> None:
+    if name in _indexes(bind, table_name):
+        op.drop_index(name, table_name=table_name)
+
+
+INDEXES = [
+    ("ix_accounts_assigned_to_id", "accounts", ["assigned_to", "id"]),
+    ("ix_accounts_tag_id", "accounts", ["tag", "id"]),
+    ("ix_login_tasks_account_status_id", "login_tasks", ["account_id", "status", "id"]),
+    ("ix_login_tasks_status_finished_at", "login_tasks", ["status", "finished_at"]),
+    ("ix_douyu_tasks_account_id", "douyu_tasks", ["account_id", "id"]),
+    ("ix_douyu_tasks_created_by_id", "douyu_tasks", ["created_by", "id"]),
+    ("ix_douyu_tasks_batch_status_id", "douyu_tasks", ["batch_id", "status", "id"]),
+    ("ix_huya_accounts_assigned_to_id", "huya_accounts", ["assigned_to", "id"]),
+    ("ix_huya_accounts_updated_at_id", "huya_accounts", ["updated_at", "id"]),
+    ("ix_huya_tasks_account_id", "huya_tasks", ["account_id", "id"]),
+    ("ix_huya_tasks_created_by_id", "huya_tasks", ["created_by", "id"]),
+    ("ix_huya_tasks_batch_status_id", "huya_tasks", ["batch_id", "status", "id"]),
+    ("ix_huya_register_items_batch_db_line", "huya_register_items", ["batch_db_id", "line"]),
+    ("ix_huya_register_success_logs_batch_id_id", "huya_register_success_logs", ["batch_id", "id"]),
+]
+
+
+def upgrade() -> None:
+    bind = op.get_bind()
+    for name, table_name, columns in INDEXES:
+        _create_index_if_missing(bind, name, table_name, columns)
+
+
+def downgrade() -> None:
+    bind = op.get_bind()
+    for name, table_name, _ in reversed(INDEXES):
+        _drop_index_if_exists(bind, name, table_name)