60 lines
1.8 KiB
PL/PgSQL
60 lines
1.8 KiB
PL/PgSQL
-- Migration: Add NOTIFY trigger for enforcement creation
|
|
-- This enables real-time notifications when enforcements are created or updated
|
|
|
|
-- Function to send notifications on enforcement changes
|
|
CREATE OR REPLACE FUNCTION notify_enforcement_change()
|
|
RETURNS TRIGGER AS $$
|
|
DECLARE
|
|
payload JSONB;
|
|
operation TEXT;
|
|
BEGIN
|
|
-- Determine operation type
|
|
IF TG_OP = 'INSERT' THEN
|
|
operation := 'created';
|
|
ELSIF TG_OP = 'UPDATE' THEN
|
|
operation := 'updated';
|
|
ELSE
|
|
operation := 'deleted';
|
|
END IF;
|
|
|
|
-- Build JSON payload with enforcement details
|
|
payload := jsonb_build_object(
|
|
'entity_type', 'enforcement',
|
|
'entity_id', NEW.id,
|
|
'operation', operation,
|
|
'timestamp', NOW(),
|
|
'data', jsonb_build_object(
|
|
'id', NEW.id,
|
|
'rule', NEW.rule,
|
|
'rule_ref', NEW.rule_ref,
|
|
'trigger_ref', NEW.trigger_ref,
|
|
'event', NEW.event,
|
|
'status', NEW.status,
|
|
'condition', NEW.condition,
|
|
'conditions', NEW.conditions,
|
|
'config', NEW.config,
|
|
'payload', NEW.payload,
|
|
'created', NEW.created,
|
|
'updated', NEW.updated
|
|
)
|
|
);
|
|
|
|
-- Send notification to the attune_notifications channel
|
|
PERFORM pg_notify('attune_notifications', payload::text);
|
|
|
|
RETURN NEW;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
-- Trigger to send pg_notify on enforcement insert
|
|
CREATE TRIGGER notify_enforcement_change
|
|
AFTER INSERT OR UPDATE ON enforcement
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION notify_enforcement_change();
|
|
|
|
-- Add comments
|
|
COMMENT ON FUNCTION notify_enforcement_change() IS
|
|
'Sends PostgreSQL NOTIFY for enforcement changes to enable real-time notifications';
|
|
COMMENT ON TRIGGER notify_enforcement_change ON enforcement IS
|
|
'Broadcasts enforcement changes via pg_notify for real-time updates';
|