Postgresql 触发器在 NEW 和 OLD 记录上运行

Question

这主要是一个风格问题。

我有一个 AFTER INSERT OR UPDATE 触发器。我想针对任何可用的 NEW/OLD 记录运行 same 查询。与其让条件检查 TG_OP 并复制查询，确定哪些可用并循环它们的最简洁方法是什么？

示例代码：

CREATE FUNCTION myfunc() RETURNS TRIGGER AS $$
  DECLARE
    id int;
    ids int[];
  BEGIN
    IF TG_OP IN ('INSERT', 'UPDATE') THEN
      ids := ids || NEW.id;
    END IF;

    IF TG_OP IN ('UPDATE', 'DELETE') THEN
      ids := ids || OLD.id;
    END IF;

    FOREACH id IN ARRAY ARRAY(SELECT DISTINCT UNNEST(ids))
    LOOP
      RAISE NOTICE 'myfunc called for % on id %', TG_OP, id;
      /* RUN QUERY REFERENCING id */
    END LOOP;

    RETURN NULL;
  END;
$$ LANGUAGE plpgsql;

是否有更短/更简单/更惯用的方法来实现这一目标？

Answer 1

数组处理和单独的SELECT DISTINCT 对于这项工作来说似乎太昂贵了。这应该更便宜：

CREATE FUNCTION myfunc()
  RETURNS TRIGGER AS
$func$
DECLARE
   _id int;
BEGIN
   CASE TG_OP
      WHEN 'INSERT', 'UPDATE' THEN
         _id := NEW.id;
      WHEN 'DELETE' THEN
         _id := OLD.id;
   END CASE;

   FOR i IN 1..2  -- max 2 iterations
   LOOP
      RAISE NOTICE 'myfunc called for % on id %', TG_OP, _id;
      /* RUN QUERY REFERENCING _id */

      EXIT WHEN TG_OP <> 'UPDATE' OR i = 2;  -- only continue in 1st round for UPDATE  
      EXIT WHEN _id = OLD.id;                -- only continue for different value
      _id := OLD.id;
   END LOOP;
   RETURN NULL;
END
$func$  LANGUAGE plpgsql;

相关：

• BREAK statement in PL/pgSQL

但我可能只是为每个 DML 语句编写一个单独的触发器函数和触发器。而是三个非常简单和快速的函数，而不是一个通用但复杂且速度较慢的函数。