feat: retry async image tasks

2026-07-09 14:46:09 +08:00
parent e3598bfe8b
commit 8878769ec5
16 changed files with 550 additions and 42 deletions
@@ -62,6 +62,7 @@ class ImageGenerationTask(models.Model):
    started_at = models.DateTimeField("开始时间", null=True, blank=True)
    finished_at = models.DateTimeField("完成时间", null=True, blank=True)
    expires_at = models.DateTimeField("任务元数据过期时间", null=True, blank=True)
+    next_attempt_at = models.DateTimeField("下次重试时间", null=True, blank=True)
    locked_at = models.DateTimeField("锁定时间", null=True, blank=True)
    lease_expires_at = models.DateTimeField("租约过期时间", null=True, blank=True)
    heartbeat_at = models.DateTimeField("心跳时间", null=True, blank=True)
@@ -89,6 +90,7 @@ class ImageGenerationTask(models.Model):
            models.Index(fields=("user", "created_at")),
            models.Index(fields=("api_key", "created_at")),
            models.Index(fields=("status", "created_at")),
+            models.Index(fields=("status", "next_attempt_at")),
            models.Index(fields=("status", "lease_expires_at")),
            models.Index(fields=("worker_id", "status")),
            models.Index(fields=("expires_at",)),