feat: 🎉 postgresql broker first commit of forked prototype from my original code

2026-01-02 20:56:39 +02:00
parent e90e5902cd
commit 19e469ff54
29 changed files with 3325 additions and 2 deletions
--- a/pkg/broker/install/sql/procedures/02_broker_run.sql
+++ b/pkg/broker/install/sql/procedures/02_broker_run.sql
@@ -0,0 +1,113 @@
+-- broker_run function
+-- Executes a job by its ID
+-- Returns: p_retval (0=success, >0=error), p_errmsg (error message)
+
+CREATE OR REPLACE FUNCTION broker_run(
+    p_job_id BIGINT,
+    OUT p_retval INTEGER,
+    OUT p_errmsg TEXT
+)
+RETURNS RECORD
+LANGUAGE plpgsql
+AS $$
+DECLARE
+    v_job_record RECORD;
+    v_execute_result TEXT;
+    v_error_occurred BOOLEAN := false;
+BEGIN
+    p_retval := 0;
+    p_errmsg := '';
+    v_execute_result := '';
+
+    -- Validate job ID
+    IF p_job_id IS NULL OR p_job_id <= 0 THEN
+        p_retval := 1;
+        p_errmsg := 'Invalid job ID';
+        RETURN;
+    END IF;
+
+    -- Get job details
+    SELECT id_broker_jobs, execute_str, job_language, run_as, complete_status
+    INTO v_job_record
+    FROM broker_jobs
+    WHERE id_broker_jobs = p_job_id
+    FOR UPDATE;
+
+    -- Check if job exists
+    IF NOT FOUND THEN
+        p_retval := 2;
+        p_errmsg := 'Job not found';
+        RETURN;
+    END IF;
+
+    -- Check if job is in running state
+    IF v_job_record.complete_status != 1 THEN
+        p_retval := 3;
+        p_errmsg := format('Job is not in running state (status: %s)', v_job_record.complete_status);
+        RETURN;
+    END IF;
+
+    -- Execute the job
+    BEGIN
+        -- For SQL/PLPGSQL jobs, execute directly
+        IF v_job_record.job_language IN ('sql', 'plpgsql') THEN
+            EXECUTE v_job_record.execute_str;
+            v_execute_result := 'Success';
+        ELSE
+            -- Other languages would need external execution
+            p_retval := 4;
+            p_errmsg := format('Unsupported job language: %s', v_job_record.job_language);
+            v_error_occurred := true;
+        END IF;
+
+    EXCEPTION
+        WHEN OTHERS THEN
+            v_error_occurred := true;
+            p_retval := 5;
+            p_errmsg := SQLERRM;
+            v_execute_result := format('Error: %s', SQLERRM);
+    END;
+
+    -- Update job with results
+    IF v_error_occurred THEN
+        UPDATE broker_jobs
+        SET complete_status = 3,  -- failed
+            error_msg = p_errmsg,
+            execute_result = v_execute_result,
+            completed_at = NOW(),
+            updated_at = NOW()
+        WHERE id_broker_jobs = p_job_id;
+    ELSE
+        UPDATE broker_jobs
+        SET complete_status = 2,  -- completed
+            execute_result = v_execute_result,
+            error_msg = NULL,
+            completed_at = NOW(),
+            updated_at = NOW()
+        WHERE id_broker_jobs = p_job_id;
+    END IF;
+
+EXCEPTION
+    WHEN OTHERS THEN
+        p_retval := 6;
+        p_errmsg := SQLERRM;
+        RAISE WARNING 'broker_run error: %', SQLERRM;
+
+        -- Try to update job status to failed
+        BEGIN
+            UPDATE broker_jobs
+            SET complete_status = 3,  -- failed
+                error_msg = SQLERRM,
+                completed_at = NOW(),
+                updated_at = NOW()
+            WHERE id_broker_jobs = p_job_id;
+        EXCEPTION
+            WHEN OTHERS THEN
+                -- Ignore update errors
+                NULL;
+        END;
+END;
+$$;
+
+-- Comments
+COMMENT ON FUNCTION broker_run IS 'Executes a job by its ID and updates the status';