orbis/internal/workscheduler/queries.sql

-- name: Ping :one
SELECT 1;

-- name: GetCurrentQueueSize :one
SELECT 
    COUNT(*) current_queue_size
FROM
    work_schedule
WHERE
        worker_id = $1
    AND state <> 'archived';

-- name: InsertQueueItem :exec
INSERT INTO work_schedule
    (
          schedule_id
        , worker_id
        , start_run
        , end_run
        , state
    )
VALUES
    (
          $1
        , $2
        , $3
        , $4
        , 'pending'
    );


-- name: GetNext :one
SELECT 
    *
FROM 
    work_schedule
WHERE
        worker_id = $1
    AND state = 'pending'
ORDER BY updated_at DESC
LIMIT 1;

-- name: StartProcessing :exec
UPDATE work_schedule
SET
    state = 'processing'
WHERE
    schedule_id = $1;

-- name: Archive :exec
UPDATE work_schedule
SET
    state = 'archived'
WHERE
    schedule_id = $1;
   
-- name: GetUnattended :many
SELECT
    *
FROM 
    work_schedule
WHERE
        worker_id NOT IN (SELECT unnest(@worker_ids::uuid[]))
    AND state <> 'archived'
    --AND updated_at <= now() - INTERVAL '10 minutes'
ORDER BY updated_at DESC
LIMIT @amount::integer;

-- name: UpdateSchdule :exec
UPDATE work_schedule
SET
      state = 'pending'
    , worker_id = $1
    , updated_at = now()
WHERE
    schedule_id = $2;