Movatterモバイル変換

41.10. Trigger Functions#

This example trigger ensures that any time a row is inserted or updated in the table, the current user name and time are stamped into the row. And it checks that an employee's name is given and that the salary is a positive value.

CREATE TABLE emp (    empname           text,    salary            integer,    last_date         timestamp,    last_user         text);CREATE FUNCTION emp_stamp() RETURNS trigger AS $emp_stamp$    BEGIN        -- Check that empname and salary are given        IF NEW.empname IS NULL THEN            RAISE EXCEPTION 'empname cannot be null';        END IF;        IF NEW.salary IS NULL THEN            RAISE EXCEPTION '% cannot have null salary', NEW.empname;        END IF;        -- Who works for us when they must pay for it?        IF NEW.salary < 0 THEN            RAISE EXCEPTION '% cannot have a negative salary', NEW.empname;        END IF;        -- Remember who changed the payroll when        NEW.last_date := current_timestamp;        NEW.last_user := current_user;        RETURN NEW;    END;$emp_stamp$ LANGUAGE plpgsql;CREATE TRIGGER emp_stamp BEFORE INSERT OR UPDATE ON emp    FOR EACH ROW EXECUTE FUNCTION emp_stamp();

This example trigger ensures that any insert, update or delete of a row in theemp table is recorded (i.e., audited) in theemp_audit table. The current time and user name are stamped into the row, together with the type of operation performed on it.

CREATE TABLE emp (    empname           text NOT NULL,    salary            integer);CREATE TABLE emp_audit(    operation         char(1)   NOT NULL,    stamp             timestamp NOT NULL,    userid            text      NOT NULL,    empname           text      NOT NULL,    salary            integer);CREATE OR REPLACE FUNCTION process_emp_audit() RETURNS TRIGGER AS $emp_audit$    BEGIN        --        -- Create a row in emp_audit to reflect the operation performed on emp,        -- making use of the special variable TG_OP to work out the operation.        --        IF (TG_OP = 'DELETE') THEN            INSERT INTO emp_audit SELECT 'D', now(), current_user, OLD.*;        ELSIF (TG_OP = 'UPDATE') THEN            INSERT INTO emp_audit SELECT 'U', now(), current_user, NEW.*;        ELSIF (TG_OP = 'INSERT') THEN            INSERT INTO emp_audit SELECT 'I', now(), current_user, NEW.*;        END IF;        RETURN NULL; -- result is ignored since this is an AFTER trigger    END;$emp_audit$ LANGUAGE plpgsql;CREATE TRIGGER emp_auditAFTER INSERT OR UPDATE OR DELETE ON emp    FOR EACH ROW EXECUTE FUNCTION process_emp_audit();

This example uses a trigger on the view to make it updatable, and ensure that any insert, update or delete of a row in the view is recorded (i.e., audited) in theemp_audit table. The current time and user name are recorded, together with the type of operation performed, and the view displays the last modified time of each row.

CREATE TABLE emp (    empname           text PRIMARY KEY,    salary            integer);CREATE TABLE emp_audit(    operation         char(1)   NOT NULL,    userid            text      NOT NULL,    empname           text      NOT NULL,    salary            integer,    stamp             timestamp NOT NULL);CREATE VIEW emp_view AS    SELECT e.empname,           e.salary,           max(ea.stamp) AS last_updated      FROM emp e      LEFT JOIN emp_audit ea ON ea.empname = e.empname     GROUP BY 1, 2;CREATE OR REPLACE FUNCTION update_emp_view() RETURNS TRIGGER AS $$    BEGIN        --        -- Perform the required operation on emp, and create a row in emp_audit        -- to reflect the change made to emp.        --        IF (TG_OP = 'DELETE') THEN            DELETE FROM emp WHERE empname = OLD.empname;            IF NOT FOUND THEN RETURN NULL; END IF;            OLD.last_updated = now();            INSERT INTO emp_audit VALUES('D', current_user, OLD.*);            RETURN OLD;        ELSIF (TG_OP = 'UPDATE') THEN            UPDATE emp SET salary = NEW.salary WHERE empname = OLD.empname;            IF NOT FOUND THEN RETURN NULL; END IF;            NEW.last_updated = now();            INSERT INTO emp_audit VALUES('U', current_user, NEW.*);            RETURN NEW;        ELSIF (TG_OP = 'INSERT') THEN            INSERT INTO emp VALUES(NEW.empname, NEW.salary);            NEW.last_updated = now();            INSERT INTO emp_audit VALUES('I', current_user, NEW.*);            RETURN NEW;        END IF;    END;$$ LANGUAGE plpgsql;CREATE TRIGGER emp_auditINSTEAD OF INSERT OR UPDATE OR DELETE ON emp_view    FOR EACH ROW EXECUTE FUNCTION update_emp_view();

The schema detailed here is partly based on theGrocery Store example fromThe Data Warehouse Toolkit by Ralph Kimball.

---- Main tables - time dimension and sales fact.--CREATE TABLE time_dimension (    time_key                    integer NOT NULL,    day_of_week                 integer NOT NULL,    day_of_month                integer NOT NULL,    month                       integer NOT NULL,    quarter                     integer NOT NULL,    year                        integer NOT NULL);CREATE UNIQUE INDEX time_dimension_key ON time_dimension(time_key);CREATE TABLE sales_fact (    time_key                    integer NOT NULL,    product_key                 integer NOT NULL,    store_key                   integer NOT NULL,    amount_sold                 numeric(12,2) NOT NULL,    units_sold                  integer NOT NULL,    amount_cost                 numeric(12,2) NOT NULL);CREATE INDEX sales_fact_time ON sales_fact(time_key);---- Summary table - sales by time.--CREATE TABLE sales_summary_bytime (    time_key                    integer NOT NULL,    amount_sold                 numeric(15,2) NOT NULL,    units_sold                  numeric(12) NOT NULL,    amount_cost                 numeric(15,2) NOT NULL);CREATE UNIQUE INDEX sales_summary_bytime_key ON sales_summary_bytime(time_key);---- Function and trigger to amend summarized column(s) on UPDATE, INSERT, DELETE.--CREATE OR REPLACE FUNCTION maint_sales_summary_bytime() RETURNS TRIGGERAS $maint_sales_summary_bytime$    DECLARE        delta_time_key          integer;        delta_amount_sold       numeric(15,2);        delta_units_sold        numeric(12);        delta_amount_cost       numeric(15,2);    BEGIN        -- Work out the increment/decrement amount(s).        IF (TG_OP = 'DELETE') THEN            delta_time_key = OLD.time_key;            delta_amount_sold = -1 * OLD.amount_sold;            delta_units_sold = -1 * OLD.units_sold;            delta_amount_cost = -1 * OLD.amount_cost;        ELSIF (TG_OP = 'UPDATE') THEN            -- forbid updates that change the time_key -            -- (probably not too onerous, as DELETE + INSERT is how most            -- changes will be made).            IF ( OLD.time_key != NEW.time_key) THEN                RAISE EXCEPTION 'Update of time_key : % -> % not allowed',                                                      OLD.time_key, NEW.time_key;            END IF;            delta_time_key = OLD.time_key;            delta_amount_sold = NEW.amount_sold - OLD.amount_sold;            delta_units_sold = NEW.units_sold - OLD.units_sold;            delta_amount_cost = NEW.amount_cost - OLD.amount_cost;        ELSIF (TG_OP = 'INSERT') THEN            delta_time_key = NEW.time_key;            delta_amount_sold = NEW.amount_sold;            delta_units_sold = NEW.units_sold;            delta_amount_cost = NEW.amount_cost;        END IF;        -- Insert or update the summary row with the new values.        <<insert_update>>        LOOP            UPDATE sales_summary_bytime                SET amount_sold = amount_sold + delta_amount_sold,                    units_sold = units_sold + delta_units_sold,                    amount_cost = amount_cost + delta_amount_cost                WHERE time_key = delta_time_key;            EXIT insert_update WHEN found;            BEGIN                INSERT INTO sales_summary_bytime (                            time_key,                            amount_sold,                            units_sold,                            amount_cost)                    VALUES (                            delta_time_key,                            delta_amount_sold,                            delta_units_sold,                            delta_amount_cost                           );                EXIT insert_update;            EXCEPTION                WHEN UNIQUE_VIOLATION THEN                    -- do nothing            END;        END LOOP insert_update;        RETURN NULL;    END;$maint_sales_summary_bytime$ LANGUAGE plpgsql;CREATE TRIGGER maint_sales_summary_bytimeAFTER INSERT OR UPDATE OR DELETE ON sales_fact    FOR EACH ROW EXECUTE FUNCTION maint_sales_summary_bytime();INSERT INTO sales_fact VALUES(1,1,1,10,3,15);INSERT INTO sales_fact VALUES(1,2,1,20,5,35);INSERT INTO sales_fact VALUES(2,2,1,40,15,135);INSERT INTO sales_fact VALUES(2,3,1,10,1,13);SELECT * FROM sales_summary_bytime;DELETE FROM sales_fact WHERE product_key = 1;SELECT * FROM sales_summary_bytime;UPDATE sales_fact SET units_sold = units_sold * 2;SELECT * FROM sales_summary_bytime;

This example produces the same results asExample 41.4, but instead of using a trigger that fires for every row, it uses a trigger that fires once per statement, after collecting the relevant information in a transition table. This can be significantly faster than the row-trigger approach when the invoking statement has modified many rows. Notice that we must make a separate trigger declaration for each kind of event, since theREFERENCING clauses must be different for each case. But this does not stop us from using a single trigger function if we choose. (In practice, it might be better to use three separate functions and avoid the run-time tests onTG_OP.)

CREATE TABLE emp (    empname           text NOT NULL,    salary            integer);CREATE TABLE emp_audit(    operation         char(1)   NOT NULL,    stamp             timestamp NOT NULL,    userid            text      NOT NULL,    empname           text      NOT NULL,    salary            integer);CREATE OR REPLACE FUNCTION process_emp_audit() RETURNS TRIGGER AS $emp_audit$    BEGIN        --        -- Create rows in emp_audit to reflect the operations performed on emp,        -- making use of the special variable TG_OP to work out the operation.        --        IF (TG_OP = 'DELETE') THEN            INSERT INTO emp_audit                SELECT 'D', now(), current_user, o.* FROM old_table o;        ELSIF (TG_OP = 'UPDATE') THEN            INSERT INTO emp_audit                SELECT 'U', now(), current_user, n.* FROM new_table n;        ELSIF (TG_OP = 'INSERT') THEN            INSERT INTO emp_audit                SELECT 'I', now(), current_user, n.* FROM new_table n;        END IF;        RETURN NULL; -- result is ignored since this is an AFTER trigger    END;$emp_audit$ LANGUAGE plpgsql;CREATE TRIGGER emp_audit_ins    AFTER INSERT ON emp    REFERENCING NEW TABLE AS new_table    FOR EACH STATEMENT EXECUTE FUNCTION process_emp_audit();CREATE TRIGGER emp_audit_upd    AFTER UPDATE ON emp    REFERENCING OLD TABLE AS old_table NEW TABLE AS new_table    FOR EACH STATEMENT EXECUTE FUNCTION process_emp_audit();CREATE TRIGGER emp_audit_del    AFTER DELETE ON emp    REFERENCING OLD TABLE AS old_table    FOR EACH STATEMENT EXECUTE FUNCTION process_emp_audit();

This example trigger simply raises aNOTICE message each time a supported command is executed.

CREATE OR REPLACE FUNCTION snitch() RETURNS event_trigger AS $$BEGIN    RAISE NOTICE 'snitch: % %', tg_event, tg_tag;END;$$ LANGUAGE plpgsql;CREATE EVENT TRIGGER snitch ON ddl_command_start EXECUTE FUNCTION snitch();