使用动态字段名称触发答案

【问题标题】：Trigger with dynamic field name使用动态字段名称触发
【发布时间】：2017-07-22 10:06:31
【问题描述】：

我在更新表上创建 PostgreSQL (9.3) 触发器时遇到问题。我想在循环中设置新值

EXECUTE 'NEW.'|| fieldName || ':=''some prepend data'' || NEW.' || fieldName || ';';

其中 fieldName 是动态设置的。但是这个字符串会引发错误

ERROR:  syntax error at or near "NEW"

我该如何实现这一目标？

【问题讨论】：

提供一个完整的触发器函数和触发器定义会更有用。

标签： postgresql triggers plpgsql dynamic-sql postgresql-9.3

【解决方案1】：

您可以使用hstore 运算符#= 相当方便地实现它：

确保在 search_path 中包含的架构中正确安装了附加模块（每个数据库一次）：

触发功能：

CREATE OR REPLACE FUNCTION tbl_insup_bef()
  RETURNS TRIGGER AS
$func$
DECLARE
   _prefix CONSTANT text := 'some prepend data'; -- your prefix here
   _prelen CONSTANT int  := 17;  -- length of above string (optional optimization)
   _col text := quote_ident(TG_ARGV[0]);
   _val text;
BEGIN
   EXECUTE 'SELECT $1.' || _col
   USING NEW
   INTO _val;

   IF left(_val, _prelen) = _prefix THEN 
      -- do nothing: prefix already there!
   ELSE
      NEW := NEW #= hstore(_col, _prefix || _val);  
   END IF;

   RETURN NEW;
END
$func$  LANGUAGE plpgsql;

触发器（对多个表重复使用相同的函数）：

CREATE TRIGGER insup_bef
BEFORE INSERT OR UPDATE ON tbl
FOR EACH ROW
EXECUTE PROCEDURE tbl_insup_bef('fieldName');  -- unquoted, case-sensitive column name

与更多解释和建议密切相关：

【讨论】：

是的，这是一个非常有用的示例。 hstore 扩展存在，但在扩展中找不到运算符#=。
@FlyBot：这很奇怪，the operator is definitely included in the 9.3 version。您确定将它安装在 您的数据库 中并且架构包含在您的 search_path 中吗？我添加了带有详细信息的链接。

【解决方案2】：

您的问题是 EXECUTE 只能用于执行 SQL 语句，而不是像您问题中的赋值那样的 PL/pgSQL 语句。

您可以像这样解决这个问题：

假设表 testtab 是这样定义的：

CREATE TABLE testtab (
   id integer primary key,
   val text
);

然后像下面这样的触发函数将起作用：

BEGIN
   EXECUTE 'SELECT $1.id, ''prefix '' || $1.val' INTO NEW USING NEW;
   RETURN NEW;
END;

我在示例中使用了硬编码的id和val，但这不是必需的。

【讨论】：

谢谢，非常有用的例子。但在我的情况下，字段名称是动态变化的，我无法将它们硬编码到触发器中。我需要将该字段的名称替换为 NEW.$1 作为示例。我可以写 EXECUTE 'SELECT $1.*, ''prefix '' || $1.$2' INTO NEW USING NEW, fieldName; ?
否，但您可以使用自省（例如使用information_schema.columns）来构建字符串，然后在EXECUTE 中使用它。

【解决方案3】：

我找到了一个可行的解决方案：触发器应该在插入/更新之后执行，而不是之前。然后所需的行采用形式

EXECUTE 'UPDATE ' || TG_TABLE_SCHEMA || '.' || TG_TABLE_NAME ||
                ' SET ' || fieldName || '= ''prefix:'' ||''' || fieldValue || ''' WHERE id = ' || NEW.id;

fieldName 和 fieldValue 我下一个方法得到：

FOR fieldName,fieldValue IN select key,value from each(hstore(NEW)) LOOP
       IF .... THEN
END LOOP:

【讨论】：