fix(worker): 隔离任务创建与队列恢复窗口

高媒体并发下,任务创建后的请求恢复和候选准入可能持续数十秒;通用 River 恢复扫描会在准入记录落库前误把新任务当成崩溃遗留任务,提前创建 Job 并占满 Worker 执行槽。\n\n为无活动 River Job 的通用兜底恢复增加 5 分钟新任务保护窗。正常提交、准入调度以及已有 River Job 的 Worker 强杀恢复保持原路径;新增 PostgreSQL 集成回归,验证新任务不被抢占且超过保护窗后仍可恢复。\n\n验证:Go 全量测试、go vet、runner/store race、gofmt、迁移安全检查通过。
This commit is contained in:
2026-07-31 07:46:47 +08:00
parent b625edcd71
commit 14d13ad3a7
2 changed files with 37 additions and 1 deletions
@@ -471,6 +471,36 @@ WHERE id = $1::uuid`, queuedAtomicTask.ID).Scan(&riverJobID, &queuedLeases); err
t.Fatal("generic River recovery claimed a task owned by the async admission dispatcher")
}
}
recoveryGraceTask := createTask(true)
recoverableTasks, err = first.ListRecoverableAsyncTasks(ctx, 1000)
if err != nil {
t.Fatalf("list fresh recoverable tasks: %v", err)
}
for _, item := range recoverableTasks {
if item.ID == recoveryGraceTask.ID {
t.Fatal("generic River recovery claimed a task still preparing admission")
}
}
if _, err := first.pool.Exec(ctx, `
UPDATE gateway_tasks
SET created_at = now() - interval '6 minutes'
WHERE id = $1::uuid`, recoveryGraceTask.ID); err != nil {
t.Fatalf("age task beyond River recovery grace period: %v", err)
}
recoverableTasks, err = first.ListRecoverableAsyncTasks(ctx, 1000)
if err != nil {
t.Fatalf("list aged recoverable tasks: %v", err)
}
recoveryGraceFound := false
for _, item := range recoverableTasks {
if item.ID == recoveryGraceTask.ID {
recoveryGraceFound = true
break
}
}
if !recoveryGraceFound {
t.Fatal("generic River recovery did not claim a task beyond the recovery grace period")
}
result, err = second.TryTaskAdmission(ctx, inputFor(queuedAtomicTask, 100, ""))
if err != nil || !result.Admitted || len(result.Leases) != 1 {
t.Fatalf("worker-time queued admission result=%+v err=%v", result, err)
+7 -1
View File
@@ -1043,11 +1043,17 @@ func (s *Store) ListRecoverableAsyncTasks(ctx context.Context, limit int) ([]Asy
if limit <= 0 {
limit = 500
}
// API handlers create the durable task before they finish media restoration
// and register distributed admission. Give that normal submission path
// ownership of fresh tasks so the generic crash recovery scan cannot create
// a River job inside the create-to-admission window.
const minimumRecoveryAge = 5 * time.Minute
rows, err := s.pool.Query(ctx, `
SELECT task.id::text, task.priority, task.next_run_at
FROM gateway_tasks task
LEFT JOIN river_job job ON job.id = task.river_job_id
WHERE task.async_mode = true
AND task.created_at <= now() - $2::interval
AND (
task.status = 'queued'
OR (
@@ -1067,7 +1073,7 @@ WHERE task.async_mode = true
AND admission.status = 'waiting'
)
ORDER BY task.priority ASC, task.created_at ASC
LIMIT $1`, limit)
LIMIT $1`, limit, minimumRecoveryAge.String())
if err != nil {
return nil, err
}