diff --git a/internal/store/store.go b/internal/store/store.go index a387acb..8a71cef 100644 --- a/internal/store/store.go +++ b/internal/store/store.go @@ -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 {