Бесконечный цикл в функции триггера

Это триггер, вызываемый либо вставкой, либо обновлением, либо удалением в таблице. Гарантируется, что в вызывающей таблице все столбцы затронуты, и также существует таблица удалений.

CREATE OR REPLACE FUNCTION sample_trigger_func() RETURNS TRIGGER AS $$
DECLARE
    operation_code char;
    table_name varchar(50);
    delete_table_name varchar(50);
    old_id integer; 

BEGIN
table_name = TG_TABLE_NAME;
delete_table_name = TG_TABLE_NAME || '_deletes';

SELECT SUBSTR(TG_OP, 1, 1)::CHAR INTO operation_code;

IF TG_OP = 'DELETE' THEN
    OLD.mod_op = operation_code;
    OLD.mod_date = now();

    RAISE INFO 'OLD: %', (OLD).name;

    EXECUTE format('INSERT INTO %s VALUES %s', delete_table_name, (OLD).*);

ELSE
    EXECUTE format('UPDATE TABLE %s SET mod_op = %s AND mod_date = %s'
                  , TG_TABLE_NAME, operation_code, now());
END IF;

RETURN NEW;
END;

$$ LANGUAGE plpgsql;

Разветвление ELSE запускает бесконечный цикл. Там может быть больше проблем. Как это исправить?

Ответы

Ответ 1

Ветвь ELSE может быть радикально упрощена. Но еще пара вещей неэффективна/неточна/опасна:

CREATE OR REPLACE FUNCTION sample_trigger_func()
  RETURNS TRIGGER AS
$func$
BEGIN
   IF TG_OP = 'DELETE' THEN
      RAISE INFO 'OLD: %', OLD.name;

      EXECUTE format('INSERT INTO %I SELECT ($1).*', TG_TABLE_NAME || '_deletes')
      USING OLD #= hstore('{mod_op, mod_datetime}'::text[]
                         , ARRAY[left(TG_OP, 1), now()::text]);
      RETURN OLD;
   ELSE  -- insert, update
      NEW.mod_op       := left(TG_OP, 1);
      NEW.mod_datetime := now();

      RETURN NEW;
   END IF;
END
$func$  LANGUAGE plpgsql;

  • В ветке ELSE просто назначьте NEW напрямую. Нет необходимости в более динамическом SQL - это приведет к тому, что один и тот же триггер снова вызовет бесконечный цикл. Это основная ошибка.

  • RETURN NEW; вне конструкции IF будет разбивать вашу триггерную функцию для DELETE, так как NEW не назначается для DELETE.

  • Ключевой особенностью является использование hstore и оператора hstore #= для динамического изменения двух выбранных полей известный тип строки - неизвестный во время написания кода. Таким образом, вы не нарушаете оригинальное значение OLD, которое может иметь неожиданный побочный эффект, если у вас больше триггеров по цепочке событий.

    OLD #= hstore('{mod_op, mod_datetime}'::text[]
             , ARRAY[left(TG_OP, 1), now()::text]);

    Должен быть установлен дополнительный модуль hstore. Подробности:

    Используя hstore(text[], text[]) здесь, чтобы построить значение hstore с несколькими полями "на лету".

  • Оператор присваивания в plpgsql равен :=:

  • Обратите внимание, что я использовал имя столбца mod_datetime вместо вводящего в заблуждение mod_date, поскольку столбец, очевидно, является timestamp, а не a date.

Я добавил несколько других улучшений, находясь на нем. И сам триггер должен выглядеть так:

CREATE TRIGGER insupdel_bef
BEFORE INSERT OR UPDATE OR DELETE ON table_name
FOR EACH ROW EXECUTE PROCEDURE sample_trigger_func();

SQL Fiddle.