feat(jobs): sqlc queries for jobs table

2026-05-05 16:59:03 +08:00
parent 215c1bb821
commit ff3f27c1dc
5 changed files with 613 additions and 0 deletions
@@ -0,0 +1,65 @@
+-- name: EnqueueJob :one
+INSERT INTO jobs (id, type, payload, status, scheduled_at, attempts, max_attempts)
+VALUES ($1, $2, $3, 'pending', $4, 0, $5)
+RETURNING *;
+
+-- name: LeaseNextJob :one
+-- SKIP LOCKED 让多 worker 并行抢不冲突；只返回单行。
+UPDATE jobs
+   SET status='running',
+       leased_at=now(),
+       leased_by=$1,
+       attempts=attempts+1,
+       updated_at=now()
+ WHERE id = (
+   SELECT id FROM jobs
+    WHERE status='pending' AND scheduled_at <= now()
+    ORDER BY scheduled_at ASC
+    FOR UPDATE SKIP LOCKED
+    LIMIT 1
+ )
+RETURNING *;
+
+-- name: CompleteJob :exec
+UPDATE jobs
+   SET status='completed', completed_at=now(),
+       leased_at=NULL, leased_by=NULL, last_error=NULL,
+       updated_at=now()
+ WHERE id=$1;
+
+-- name: FailJobWithRetry :exec
+UPDATE jobs
+   SET status='pending',
+       scheduled_at=$2,
+       leased_at=NULL, leased_by=NULL,
+       last_error=$3,
+       updated_at=now()
+ WHERE id=$1;
+
+-- name: DeadLetterJob :exec
+UPDATE jobs
+   SET status='dead', completed_at=now(),
+       leased_at=NULL, leased_by=NULL, last_error=$2,
+       updated_at=now()
+ WHERE id=$1;
+
+-- name: GetJob :one
+SELECT * FROM jobs WHERE id=$1;
+
+-- name: ReapStuckRunningJobs :many
+-- 复位 worker 崩了卡 running 的任务。
+UPDATE jobs
+   SET status='pending', leased_at=NULL, leased_by=NULL, scheduled_at=now(),
+       updated_at=now()
+ WHERE status='running' AND leased_at < $1
+RETURNING id;
+
+-- name: PurgeCompletedJobs :execrows
+DELETE FROM jobs
+ WHERE status IN ('completed','dead') AND completed_at < $1;
+
+-- name: ListDeadJobs :many
+SELECT * FROM jobs
+ WHERE status='dead' AND completed_at >= $1
+ ORDER BY completed_at DESC
+ LIMIT $2;