admin/ai-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Grendgi
2026-06-08 14:02:15 +03:00
parent 7452b1d5f2
commit eb59298135
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/store/store.go
View File

@@ -26,6 +26,13 @@ scheduled_at, started_at, completed_at, worker_id, heartbeat_at,
created_at, updated_at, idempotency_key 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) { func Open(ctx context.Context, databaseURL string) (*Store, error) {
if strings.TrimSpace(databaseURL) == "" { if strings.TrimSpace(databaseURL) == "" {
return nil, errors.New("DATABASE_URL is required") return nil, errors.New("DATABASE_URL is required")
@@ -339,7 +346,7 @@ SET status = 'running',
updated_at = NOW() updated_at = NOW()
FROM picked FROM picked
WHERE j.id = picked.id WHERE j.id = picked.id
RETURNING ` + jobSelectColumns + ` RETURNING ` + jobSelectColumnsFromJobAlias + `
` `
rows, err := s.pool.Query(ctx, q, in.TaskTypes, in.ModelProfiles, in.Limit, workerID) rows, err := s.pool.Query(ctx, q, in.TaskTypes, in.ModelProfiles, in.Limit, workerID)
if err != nil { if err != nil {