From 005411717184353746b237baf453507abe82b359 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 25 Jan 2025 04:17:12 +0800 Subject: [PATCH] Add workflow_oid_status_idx index to workflow table to index status (#1642) --- ...7_add_workflow_oid_status_idx_index_to_.py | 29 +++++++++++++++++++ skyvern/forge/sdk/db/models.py | 1 + 2 files changed, 30 insertions(+) create mode 100644 alembic/versions/2025_01_24_2014-957ad2d1d3f7_add_workflow_oid_status_idx_index_to_.py diff --git a/alembic/versions/2025_01_24_2014-957ad2d1d3f7_add_workflow_oid_status_idx_index_to_.py b/alembic/versions/2025_01_24_2014-957ad2d1d3f7_add_workflow_oid_status_idx_index_to_.py new file mode 100644 index 00000000..27067e7c --- /dev/null +++ b/alembic/versions/2025_01_24_2014-957ad2d1d3f7_add_workflow_oid_status_idx_index_to_.py @@ -0,0 +1,29 @@ +"""Add workflow_oid_status_idx index to workflow table to index status + +Revision ID: 957ad2d1d3f7 +Revises: 26dc22efaf0b +Create Date: 2025-01-24 20:14:49.790319+00:00 + +""" + +from typing import Sequence, Union + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "957ad2d1d3f7" +down_revision: Union[str, None] = "26dc22efaf0b" +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.create_index("workflow_oid_status_idx", "workflows", ["organization_id", "status"], unique=False) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_index("workflow_oid_status_idx", table_name="workflows") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index a2d1f08f..af10b018 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -195,6 +195,7 @@ class WorkflowModel(Base): ), Index("permanent_id_version_idx", "workflow_permanent_id", "version"), Index("organization_id_title_idx", "organization_id", "title"), + Index("workflow_oid_status_idx", "organization_id", "status"), ) workflow_id = Column(String, primary_key=True, index=True, default=generate_workflow_id)