jay brown
54 lines
1.6 KiB
SQL
54 lines
1.6 KiB
SQL
-- name: CreateBatchUpload :one
|
|
INSERT INTO batch_uploads (client_id, original_filename, total_documents)
|
|
VALUES ($1, $2, $3) RETURNING id;
|
|
|
|
-- name: GetBatchUpload :one
|
|
SELECT id, client_id, original_filename, total_documents, processed_documents,
|
|
failed_documents, invalid_type_documents, status, progress_percent,
|
|
failed_filenames, created_at, completed_at
|
|
FROM batch_uploads
|
|
WHERE id = $1 AND client_id = $2;
|
|
|
|
-- name: ListBatchUploads :many
|
|
SELECT id, client_id, original_filename, total_documents, processed_documents,
|
|
failed_documents, invalid_type_documents, status, progress_percent,
|
|
created_at, completed_at
|
|
FROM batch_uploads
|
|
WHERE client_id = $1
|
|
ORDER BY created_at DESC
|
|
LIMIT $2 OFFSET $3;
|
|
|
|
-- name: UpdateBatchProgress :exec
|
|
UPDATE batch_uploads
|
|
SET processed_documents = $2,
|
|
failed_documents = $3,
|
|
invalid_type_documents = $4,
|
|
progress_percent = $5
|
|
WHERE id = $1;
|
|
|
|
-- name: UpdateBatchStatus :exec
|
|
UPDATE batch_uploads
|
|
SET status = $2::batch_status,
|
|
completed_at = CASE WHEN $2::batch_status IN ('completed', 'failed', 'cancelled') THEN NOW() ELSE NULL END
|
|
WHERE id = $1;
|
|
|
|
-- name: AddFailedFilename :exec
|
|
UPDATE batch_uploads
|
|
SET failed_filenames = failed_filenames || jsonb_build_array($2::text)
|
|
WHERE id = $1;
|
|
|
|
-- name: CancelBatchUpload :exec
|
|
UPDATE batch_uploads
|
|
SET status = 'cancelled',
|
|
completed_at = NOW()
|
|
WHERE id = $1 AND client_id = $2 AND status = 'processing';
|
|
|
|
-- name: GetDocumentsByBatchId :many
|
|
SELECT id, clientId, hash
|
|
FROM documents
|
|
WHERE batch_id = $1;
|
|
|
|
-- name: CountDocumentsByBatchId :one
|
|
SELECT COUNT(*) as count
|
|
FROM documents
|
|
WHERE batch_id = $1; |