celery 容灾升级
This commit is contained in:
@@ -22,6 +22,8 @@ class ShotReplicateSegment(Base, TimestampMixin, SoftDeleteMixin):
|
||||
Index("idx_shot_replicate_segments_source_mode", "source_mode"),
|
||||
Index("idx_shot_replicate_segments_split_status", "split_status"),
|
||||
Index("idx_shot_replicate_segments_analysis_status", "analysis_status"),
|
||||
Index("idx_shot_segments_analysis_lease", "analysis_status", "analysis_lease_until"),
|
||||
Index("idx_shot_segments_split_lease", "split_status", "split_lease_until"),
|
||||
Index("idx_shot_replicate_segments_replicate_status", "replicate_status"),
|
||||
Index("idx_shot_replicate_segments_project", "module_project_id"),
|
||||
Index(
|
||||
@@ -68,6 +70,7 @@ class ShotReplicateSegment(Base, TimestampMixin, SoftDeleteMixin):
|
||||
ai_suggestion_json: Mapped[dict[str, Any] | list[Any] | None] = mapped_column(_JSON_TYPE, nullable=True)
|
||||
module_project_id: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
|
||||
|
||||
split_claim_token: Mapped[str | None] = mapped_column(String(64), nullable=True)
|
||||
split_celery_task_id: Mapped[str | None] = mapped_column(String(160), nullable=True, index=True)
|
||||
split_enqueued_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
split_started_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
@@ -77,4 +80,8 @@ class ShotReplicateSegment(Base, TimestampMixin, SoftDeleteMixin):
|
||||
split_last_error: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
split_completed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user