agent-platform/services/scheduler-service/alembic/versions/20260426_0001_init_scheduler_models.py

"""init scheduler models

Revision ID: 20260426_0001
Revises:
Create Date: 2026-04-26 00:20:00
"""

from collections.abc import Sequence

from alembic import op
import sqlalchemy as sa


revision: str = "20260426_0001"
down_revision: str | None = None
branch_labels: Sequence[str] | None = None
depends_on: Sequence[str] | None = None


def upgrade() -> None:
    op.create_table(
        "scheduled_job",
        sa.Column("job_type", sa.String(length=32), nullable=False),
        sa.Column("status", sa.String(length=32), nullable=False),
        sa.Column("name", sa.String(length=128), nullable=False),
        sa.Column("description", sa.Text(), nullable=True),
        sa.Column("target_service", sa.String(length=64), nullable=True),
        sa.Column("target_url", sa.Text(), nullable=True),
        sa.Column("method", sa.String(length=16), nullable=True),
        sa.Column("payload_json", sa.JSON(), nullable=False),
        sa.Column("schedule_time", sa.DateTime(), nullable=False),
        sa.Column("lease_expire_time", sa.DateTime(), nullable=True),
        sa.Column("claimed_by", sa.String(length=128), nullable=True),
        sa.Column("claimed_time", sa.DateTime(), nullable=True),
        sa.Column("completed_time", sa.DateTime(), nullable=True),
        sa.Column("attempt_count", sa.Integer(), nullable=False),
        sa.Column("max_attempts", sa.Integer(), nullable=False),
        sa.Column("last_error_message", sa.Text(), nullable=True),
        sa.Column("metadata_json", sa.JSON(), nullable=False),
        sa.Column("id", sa.String(length=36), nullable=False),
        sa.Column("tenant_id", sa.String(length=36), nullable=False),
        sa.Column("created_by", sa.String(length=36), nullable=True),
        sa.Column("updated_by", sa.String(length=36), nullable=True),
        sa.Column("created_time", sa.DateTime(), nullable=False),
        sa.Column("updated_time", sa.DateTime(), nullable=False),
        sa.Column("deleted_time", sa.DateTime(), nullable=True),
        sa.Column("version", sa.Integer(), nullable=False),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index("ix_scheduled_job_tenant_id", "scheduled_job", ["tenant_id"])
    op.create_index("ix_scheduled_job_job_type", "scheduled_job", ["job_type"])
    op.create_index("ix_scheduled_job_status", "scheduled_job", ["status"])
    op.create_index("ix_scheduled_job_target_service", "scheduled_job", ["target_service"])
    op.create_index("ix_scheduled_job_schedule_time", "scheduled_job", ["schedule_time"])
    op.create_index("ix_scheduled_job_claimed_by", "scheduled_job", ["claimed_by"])


def downgrade() -> None:
    op.drop_index("ix_scheduled_job_claimed_by", table_name="scheduled_job")
    op.drop_index("ix_scheduled_job_schedule_time", table_name="scheduled_job")
    op.drop_index("ix_scheduled_job_target_service", table_name="scheduled_job")
    op.drop_index("ix_scheduled_job_status", table_name="scheduled_job")
    op.drop_index("ix_scheduled_job_job_type", table_name="scheduled_job")
    op.drop_index("ix_scheduled_job_tenant_id", table_name="scheduled_job")
    op.drop_table("scheduled_job")