From 0abfda7f2db5e701c843a03813cf234f40fa29b3 Mon Sep 17 00:00:00 2001 From: wwwwwwwww <526125649@qq.com> Date: Thu, 9 Jul 2026 16:21:56 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0chat=5Fgeneration=5Ftasks?= =?UTF-8?q?=E3=80=81generation=5Frecords=E6=97=B6=E9=97=B4generated=5Fat?= =?UTF-8?q?=E7=B4=A2=E5=BC=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ...2_add_generated_date_expression_indexes.py | 46 +++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 video-gen-api/alembic/versions/2026070902_add_generated_date_expression_indexes.py diff --git a/video-gen-api/alembic/versions/2026070902_add_generated_date_expression_indexes.py b/video-gen-api/alembic/versions/2026070902_add_generated_date_expression_indexes.py new file mode 100644 index 00000000..8520a8c6 --- /dev/null +++ b/video-gen-api/alembic/versions/2026070902_add_generated_date_expression_indexes.py @@ -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')