Fix AI job claim returning columns
All checks were successful
CI / test (push) Successful in 13s
Build and Deploy / build-and-deploy (push) Successful in 22s

This commit is contained in:
Grendgi
2026-06-08 14:02:15 +03:00
parent 7452b1d5f2
commit eb59298135

View File

@@ -26,6 +26,13 @@ scheduled_at, started_at, completed_at, worker_id, heartbeat_at,
created_at, updated_at, idempotency_key
`
const jobSelectColumnsFromJobAlias = `
j.id, j.owner_service, j.owner_ref, j.task_type, j.model_profile, j.priority, j.status,
j.attempts, j.max_attempts, j.input, j.result, j.error_code, j.error_message,
j.scheduled_at, j.started_at, j.completed_at, j.worker_id, j.heartbeat_at,
j.created_at, j.updated_at, j.idempotency_key
`
func Open(ctx context.Context, databaseURL string) (*Store, error) {
if strings.TrimSpace(databaseURL) == "" {
return nil, errors.New("DATABASE_URL is required")
@@ -339,7 +346,7 @@ SET status = 'running',
updated_at = NOW()
FROM picked
WHERE j.id = picked.id
RETURNING ` + jobSelectColumns + `
RETURNING ` + jobSelectColumnsFromJobAlias + `
`
rows, err := s.pool.Query(ctx, q, in.TaskTypes, in.ModelProfiles, in.Limit, workerID)
if err != nil {