使用动态字段名称触发

时间:2016-08-02 07:14:09

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

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

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

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

ERROR:  syntax error at or near "NEW"

我如何实现这一目标?

3 个答案:

答案 0 :(得分:7)

您可以使用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;

触发器(为多个表重用相同的func):

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

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

答案 1 :(得分:1)

您的问题是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;

我在我的示例中使用了硬编码的idval,但这不是必需的。

答案 2 :(得分:0)

我找到了一个有效的解决方案: 触发器应在插入/更新后执行,而不是之前执行。然后所需的行采用

形式
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: