Fix realtime queued task counts

2026-05-12 10:41:07 +08:00
parent 2a91b31d12
commit 9ea83be718
6 changed files with 128 additions and 22 deletions
@@ -189,7 +189,7 @@ WHERE t.id = picked.task_id
 RETURNING `+gatewayTaskColumns, workerID))
 }

-func (s *Store) RequeueTask(ctx context.Context, taskID string, delay time.Duration) (GatewayTask, error) {
+func (s *Store) RequeueTask(ctx context.Context, taskID string, delay time.Duration, queueKey string) (GatewayTask, error) {
 	if delay < time.Second {
 		delay = time.Second
 	}
@@ -204,12 +204,13 @@ SET status = 'queued',
    locked_at = NULL,
    heartbeat_at = NULL,
    next_run_at = $2::timestamptz,
+    queue_key = COALESCE(NULLIF($3::text, ''), queue_key),
    error = NULL,
    error_code = NULL,
    error_message = NULL,
    updated_at = now()
 WHERE id = $1::uuid
-RETURNING `+gatewayTaskColumns, taskID, nextRunAt))
+RETURNING `+gatewayTaskColumns, taskID, nextRunAt, strings.TrimSpace(queueKey)))
 }

 func (s *Store) SetTaskRiverJobID(ctx context.Context, taskID string, riverJobID int64) error {