1 """Add index for scheduled deployments
2
3 Revision ID: a0284438370e
4 Revises: af52717cf201
5 Create Date: 2022-11-05 18:06:19.568896
6
7 """
8
9 from alembic import op 1 ctx 1a
10
11 # revision identifiers, used by Alembic.
12 revision = "a0284438370e" 1 ctx 1a
13 down_revision = "af52717cf201" 1 ctx 1a
14 branch_labels = None 1 ctx 1a
15 depends_on = None 1 ctx 1a
16
17
18 def upgrade ( ) : 1 ctx 1a
19 op . execute ( 1 ctx 1a
20 """
21 CREATE INDEX ix_flow_run__scheduler_deployment_id_auto_scheduled_next_scheduled_start_time
22 ON flow_run (deployment_id, auto_scheduled, next_scheduled_start_time)
23 WHERE state_type = 'SCHEDULED';
24 """
25 )
26
27
28 def downgrade ( ) : 1 ctx 1a
29 op . execute (
30 """
31 DROP INDEX ix_flow_run__scheduler_deployment_id_auto_scheduled_next_scheduled_start_time;
32 """
33 )