feat: track generation jobs

2026-04-18 16:29:22 +08:00
parent 16fafe0fe0
commit 96dfc677e2
18 changed files with 709 additions and 71 deletions
--- a/backend/alembic/versions/0011_add_generation_jobs.py
+++ b/backend/alembic/versions/0011_add_generation_jobs.py
@@ -0,0 +1,77 @@
+"""add generation jobs
+
+Revision ID: 0011_add_generation_jobs
+Revises: 0010_add_story_audio_cache_path
+Create Date: 2026-04-18
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+
+revision = "0011_add_generation_jobs"
+down_revision = "0010_add_story_audio_cache_path"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "generation_jobs",
+        sa.Column("id", sa.String(length=36), nullable=False),
+        sa.Column("user_id", sa.String(length=255), nullable=False),
+        sa.Column("story_id", sa.Integer(), nullable=True),
+        sa.Column("output_mode", sa.String(length=32), nullable=False),
+        sa.Column("input_type", sa.String(length=32), nullable=False),
+        sa.Column("status", sa.String(length=32), nullable=False, server_default="running"),
+        sa.Column(
+            "current_step",
+            sa.String(length=64),
+            nullable=False,
+            server_default="request_accepted",
+        ),
+        sa.Column("request_payload", sa.JSON(), nullable=False, server_default="{}"),
+        sa.Column("result_snapshot", sa.JSON(), nullable=False, server_default="{}"),
+        sa.Column("error_message", sa.Text(), nullable=True),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
+        sa.ForeignKeyConstraint(["story_id"], ["stories.id"], ondelete="SET NULL"),
+        sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_generation_jobs_user_id", "generation_jobs", ["user_id"])
+    op.create_index("ix_generation_jobs_story_id", "generation_jobs", ["story_id"])
+    op.create_index("ix_generation_jobs_status", "generation_jobs", ["status"])
+    op.create_index("ix_generation_jobs_created_at", "generation_jobs", ["created_at"])
+
+    op.create_table(
+        "generation_job_events",
+        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
+        sa.Column("job_id", sa.String(length=36), nullable=False),
+        sa.Column("story_id", sa.Integer(), nullable=True),
+        sa.Column("event_type", sa.String(length=64), nullable=False),
+        sa.Column("status", sa.String(length=32), nullable=False),
+        sa.Column("message", sa.Text(), nullable=True),
+        sa.Column("event_metadata", sa.JSON(), nullable=False, server_default="{}"),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
+        sa.ForeignKeyConstraint(["job_id"], ["generation_jobs.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["story_id"], ["stories.id"], ondelete="SET NULL"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_generation_job_events_job_id", "generation_job_events", ["job_id"])
+    op.create_index("ix_generation_job_events_story_id", "generation_job_events", ["story_id"])
+    op.create_index("ix_generation_job_events_created_at", "generation_job_events", ["created_at"])
+
+
+def downgrade() -> None:
+    op.drop_index("ix_generation_job_events_created_at", table_name="generation_job_events")
+    op.drop_index("ix_generation_job_events_story_id", table_name="generation_job_events")
+    op.drop_index("ix_generation_job_events_job_id", table_name="generation_job_events")
+    op.drop_table("generation_job_events")
+
+    op.drop_index("ix_generation_jobs_created_at", table_name="generation_jobs")
+    op.drop_index("ix_generation_jobs_status", table_name="generation_jobs")
+    op.drop_index("ix_generation_jobs_story_id", table_name="generation_jobs")
+    op.drop_index("ix_generation_jobs_user_id", table_name="generation_jobs")
+    op.drop_table("generation_jobs")