0010_tenant_dispatch.py 782 B

1234567891011121314151617181920212223242526
  1. import sqlalchemy as sa
  2. from alembic import op
  3. from sqlalchemy.dialects import mysql
  4. revision = "0010_tenant_dispatch"
  5. down_revision = "0009_monitoring_indexes"
  6. branch_labels = None
  7. depends_on = None
  8. def upgrade():
  9. op.create_table(
  10. "tenant_dispatch",
  11. sa.Column("tenant_id", mysql.BINARY(16), nullable=False),
  12. sa.Column("selected_at", mysql.DATETIME(fsp=6), nullable=False),
  13. sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
  14. sa.PrimaryKeyConstraint("tenant_id"),
  15. )
  16. op.create_index(
  17. "ix_agent_runs_tenant_queue", "agent_runs", ["tenant_id", "status", "created_at", "id"]
  18. )
  19. def downgrade():
  20. op.drop_index("ix_agent_runs_tenant_queue", "agent_runs")
  21. op.drop_table("tenant_dispatch")