51 lines
1.6 KiB
PL/PgSQL
51 lines
1.6 KiB
PL/PgSQL
-- Migration: 003_jobs_scheduling
|
|
-- Description: Retry/scheduling columns for patch_job_hosts, and NOTIFY trigger
|
|
-- for immediate patch job dispatch.
|
|
|
|
-- ============================================================
|
|
-- Add retry-scheduling columns to patch_job_hosts
|
|
-- ============================================================
|
|
|
|
-- When the retry engine should next attempt this host; NULL = not scheduled
|
|
ALTER TABLE patch_job_hosts
|
|
ADD COLUMN IF NOT EXISTS retry_next_at TIMESTAMPTZ;
|
|
|
|
-- Last failure reason captured by the worker for display in the UI
|
|
ALTER TABLE patch_job_hosts
|
|
ADD COLUMN IF NOT EXISTS last_error TEXT;
|
|
|
|
-- ============================================================
|
|
-- pg_notify trigger: fires when an immediate job is inserted
|
|
-- ============================================================
|
|
|
|
CREATE OR REPLACE FUNCTION notify_job_enqueued()
|
|
RETURNS TRIGGER
|
|
LANGUAGE plpgsql
|
|
AS $$
|
|
BEGIN
|
|
IF NEW.immediate = TRUE THEN
|
|
PERFORM pg_notify('job_enqueued', NEW.id::text);
|
|
END IF;
|
|
RETURN NEW;
|
|
END;
|
|
$$;
|
|
|
|
DO $$ BEGIN
|
|
IF NOT EXISTS (
|
|
SELECT 1 FROM pg_trigger WHERE tgname = 'trg_job_enqueued'
|
|
) THEN
|
|
CREATE TRIGGER trg_job_enqueued
|
|
AFTER INSERT ON patch_jobs
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION notify_job_enqueued();
|
|
END IF;
|
|
END $$;
|
|
|
|
-- ============================================================
|
|
-- Index: efficiently find hosts due for retry
|
|
-- ============================================================
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_pjh_retry
|
|
ON patch_job_hosts (retry_next_at)
|
|
WHERE retry_next_at IS NOT NULL;
|