celery 容灾升级

video-gen-api/app/models/shot_replicate_task_set.py

@@ -1,8 +1,9 @@
 from __future__ import annotations
 
+from datetime import datetime
 from typing import Any
 
-from sqlalchemy import Float, ForeignKey, Index, Integer, JSON, String, Text, text
+from sqlalchemy import DateTime, Float, ForeignKey, Index, Integer, JSON, String, Text, text
 from sqlalchemy.dialects.postgresql import JSONB
 from sqlalchemy.orm import Mapped, mapped_column
 
@@ -19,6 +20,7 @@ class ShotReplicateTaskSet(Base, TimestampMixin, SoftDeleteMixin):
         Index("idx_shot_replicate_task_sets_user_created", "user_id", "created_at"),
         Index("idx_shot_replicate_task_sets_status", "status"),
         Index("idx_shot_replicate_task_sets_analysis_status", "analysis_status"),
+        Index("idx_shot_task_sets_analysis_lease", "analysis_status", "analysis_lease_until"),
         Index("idx_shot_replicate_task_sets_split_status", "split_status"),
         Index(
             "uq_shot_replicate_task_sets_user_idempotency",
@@ -55,6 +57,10 @@ class ShotReplicateTaskSet(Base, TimestampMixin, SoftDeleteMixin):
     completed_segment_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
     failed_segment_count: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
 
+    analysis_attempt_no: Mapped[int] = mapped_column(Integer, default=1, nullable=False)
+    analysis_claim_token: Mapped[str | None] = mapped_column(String(64), nullable=True)
+    analysis_started_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+    analysis_lease_until: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
     analysis_error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
     split_error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
     idempotency_key: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)