增加chat_generation_tasks、generation_records时间generated_at索引

2026-07-09 16:21:56 +08:00
parent 3f1e84f0c9
commit 0abfda7f2d
1 changed files with 46 additions and 0 deletions
@@ -0,0 +1,46 @@
+"""2026070902_add_generated_date_expression_indexes
+
+Revision ID: 2026070902
+Revises: 2026070901
+Create Date: 2026-07-09 00:00:00.000000
+
+该文件包含 2026-07-09 的数据库迁移内容：
+1. chat_generation_tasks 表增加 generated_at 日期表达式索引，优化按日期分组/筛选
+2. generation_records 表增加 generated_at 日期表达式索引，优化按日期分组/筛选
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision: str = '2026070902'
+down_revision: Union[str, None] = '2026070901'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ========================================
+    # 2026-07-09 - 生成记录表增加日期表达式索引
+    # ========================================
+    op.create_index(
+        'idx_chat_generation_tasks_generated_date',
+        'chat_generation_tasks',
+        [sa.text("date(generated_at)")],
+        unique=False,
+    )
+    op.create_index(
+        'idx_generation_records_generated_date',
+        'generation_records',
+        [sa.text("date(generated_at)")],
+        unique=False,
+    )
+
+
+def downgrade() -> None:
+    # ========================================
+    # 2026-07-09 - 生成记录表增加日期表达式索引（回滚）
+    # ========================================
+    op.drop_index('idx_generation_records_generated_date', table_name='generation_records')
+    op.drop_index('idx_chat_generation_tasks_generated_date', table_name='chat_generation_tasks')