import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import mysql

revision = "0010_tenant_dispatch"
down_revision = "0009_monitoring_indexes"
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        "tenant_dispatch",
        sa.Column("tenant_id", mysql.BINARY(16), nullable=False),
        sa.Column("selected_at", mysql.DATETIME(fsp=6), nullable=False),
        sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
        sa.PrimaryKeyConstraint("tenant_id"),
    )
    op.create_index(
        "ix_agent_runs_tenant_queue", "agent_runs", ["tenant_id", "status", "created_at", "id"]
    )


def downgrade():
    op.drop_index("ix_agent_runs_tenant_queue", "agent_runs")
    op.drop_table("tenant_dispatch")