youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-16 12:27:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: add indexes for human_input_forms query patterns (#32849)

Browse Source 
Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Bond Zhu <37842169+MRZHUH@users.noreply.github.com>
This commit is contained in:
非法操作
2026-03-16 10:10:03 +08:00
committed by GitHub
parent c71ecd2fe0
commit 09dad78a5d
2 changed files with 87 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
api/migrations/versions/2026_03_02_1805-0ec65df55790_add_indexes_for_human_input_forms.py Normal file
View File

@ -0,0 +1,68 @@
"""add indexes for human_input_forms query patterns
Revision ID: 0ec65df55790
Revises: e288952f2994
Create Date: 2026-03-02 18:05:00.000000
"""
from alembic import op
# revision identifiers, used by Alembic.
revision = "0ec65df55790"
down_revision = "e288952f2994"
branch_labels = None
depends_on = None
def upgrade():
with op.batch_alter_table("human_input_forms", schema=None) as batch_op:
batch_op.create_index(
"human_input_forms_workflow_run_id_node_id_idx",
["workflow_run_id", "node_id"],
unique=False,
)
batch_op.create_index(
"human_input_forms_status_created_at_idx",
["status", "created_at"],
unique=False,
)
batch_op.create_index(
"human_input_forms_status_expiration_time_idx",
["status", "expiration_time"],
unique=False,
)
with op.batch_alter_table("human_input_form_deliveries", schema=None) as batch_op:
batch_op.create_index(
batch_op.f("human_input_form_deliveries_form_id_idx"),
["form_id"],
unique=False,
)
with op.batch_alter_table("human_input_form_recipients", schema=None) as batch_op:
batch_op.create_index(
batch_op.f("human_input_form_recipients_delivery_id_idx"),
["delivery_id"],
unique=False,
)
batch_op.create_index(
batch_op.f("human_input_form_recipients_form_id_idx"),
["form_id"],
unique=False,
)
def downgrade():
with op.batch_alter_table("human_input_forms", schema=None) as batch_op:
batch_op.drop_index("human_input_forms_workflow_run_id_node_id_idx")
batch_op.drop_index("human_input_forms_status_expiration_time_idx")
batch_op.drop_index("human_input_forms_status_created_at_idx")
with op.batch_alter_table("human_input_form_recipients", schema=None) as batch_op:
batch_op.drop_index(batch_op.f("human_input_form_recipients_form_id_idx"))
batch_op.drop_index(batch_op.f("human_input_form_recipients_delivery_id_idx"))
with op.batch_alter_table("human_input_form_deliveries", schema=None) as batch_op:
batch_op.drop_index(batch_op.f("human_input_form_deliveries_form_id_idx"))

19
api/models/human_input.py
View File

@ -30,6 +30,15 @@ def _generate_token() -> str:
class HumanInputForm(DefaultFieldsMixin, Base):
__tablename__ = "human_input_forms"
__table_args__ = (
sa.Index(
"human_input_forms_workflow_run_id_node_id_idx",
"workflow_run_id",
"node_id",
),
sa.Index("human_input_forms_status_expiration_time_idx", "status", "expiration_time"),
sa.Index("human_input_forms_status_created_at_idx", "status", "created_at"),
)
tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
app_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
@ -84,6 +93,12 @@ class HumanInputForm(DefaultFieldsMixin, Base):
class HumanInputDelivery(DefaultFieldsMixin, Base):
__tablename__ = "human_input_form_deliveries"
__table_args__ = (
sa.Index(
None,
"form_id",
),
)
form_id: Mapped[str] = mapped_column(
StringUUID,
@ -181,6 +196,10 @@ RecipientPayload = Annotated[
class HumanInputFormRecipient(DefaultFieldsMixin, Base):
__tablename__ = "human_input_form_recipients"
__table_args__ = (
sa.Index(None, "form_id"),
sa.Index(None, "delivery_id"),
)
form_id: Mapped[str] = mapped_column(
StringUUID,