query-orchestration/database/migrations/00000000000102_job_views.up.sql

CREATE OR REPLACE FUNCTION listJobDocumentIDs(
    _jobId uuid,
    _batchSize INTEGER,
    _offset INTEGER
)
RETURNS TABLE (id uuid, totalCount BIGINT) AS $$
DECLARE
    totalCount BIGINT := 0;
BEGIN
    SELECT COUNT(*)::BIGINT INTO totalCount FROM documents WHERE jobId = _jobId;
    
    RETURN QUERY
        SELECT d.id, totalCount
        FROM documents as d
        WHERE d.jobId = _jobId
        ORDER BY d.id
        LIMIT _batchSize
        OFFSET _offset;
END;
$$ LANGUAGE plpgsql;