Question

我正在创建一个触发器，它使用列的动态名称

NEW.name:=2222; -- works fine !

但

dynamic_column:='name';
EXECUTE '$1.'||dynamic_column||':=2222 ' USING NEW; -- raises error

给出错误：

错误：语法错误在或附近＆＃34; $ 1＆＃34;第1行：$ 1.name：= 2222

Answer 1

问题是这不是有效的SQL语句。

您可以使用动态SQL访问new中的列，如下所示：

EXECUTE 'SELECT $1.id' INTO v_id USING NEW;

在NEW中更改单个列时没有这种方法。

您可以使用TG_RELID来获取表的OID，查询列的pg_attribute，组成由NEW中的值和您的新值组成的行文字字符串，强制转换这是表格类型，并将结果分配给NEW。相当麻烦。

以下是执行此操作的示例代码（我对其进行了测试，但可能还有错误）：

CREATE OR REPLACE FUNCTION dyntrig() RETURNS trigger
   LANGUAGE plpgsql AS
$$DECLARE
   colname text;
   colval text;
   newrow text := '';
   fieldsep text := 'ROW(';
BEGIN
   /* loop through the columns of the table */
   FOR colname IN
      SELECT attname
      FROM pg_catalog.pg_attribute
      WHERE attrelid = TG_RELID
        AND attnum > 0
        AND NOT attisdropped
      ORDER BY attnum
   LOOP
      IF colname = 'name' THEN
         colval = '2222';
      ELSE
         /* all other columns than 'name' retain their value */
         EXECUTE 'SELECT CAST($1.' || quote_ident(colname) || ' AS text)'
            INTO colval USING NEW;
      END IF;

      /* compose a string that represents the new table row */
      IF colval IS NULL THEN
         newrow := newrow || fieldsep || 'NULL';
      ELSE
         newrow := newrow || fieldsep || '''' || colval || '''';
      END IF;
      fieldsep := ',';
   END LOOP;
   newrow := newrow || ')';

   /* assign the new table row to NEW */
   EXECUTE 'SELECT (CAST(' || newrow || ' AS '
           || quote_ident(TG_TABLE_SCHEMA) || '.' || quote_ident(TG_TABLE_NAME)
           || ')).*'
      INTO NEW;

   RETURN NEW;
END;$$;

Answer 2

我在这里找到了信息：Assign to NEW by key in a Postgres trigger

如果我们通过以下方式启用模块hstore：

CREATE EXTENSION hstore;

我们可以这样做：

dynamic_column:='name';    
temp_sql_string:='"'||dynamic_column||'"=>"2222"';
NEW := NEW #= temp_sql_string::hstore;

现在，RECORD NEW.name设置为值2222。

感谢您努力寻找解决方案@Laurenz Albe

Answer 3

您已找到my answer recommending the hstore operator #= on dba.SE。您可能还对SO上的相应参考答案感兴趣：

How to set value of composite variable field using dynamic SQL

由于您从变量构造辅助hstore值，我建议使用简单函数hstore()：

CREATE OR REPLACE FUNCTION dyn_trigger_func()
  RETURNS TRIGGER AS
$func$
DECLARE
   dyn_col_name text := 'name';
   dyn_col_val  text := '2222';
BEGIN
   NEW := NEW #= hstore(dyn_col_name, dyn_col_val);
   RETURN NEW;
END
$func$  LANGUAGE plpgsql;

这种方式更快/更简单/更清晰/更安全。

或者，因为它显然是一个触发器函数，你可能希望在CREATE TRIGGER语句中传递列名和值：

CREATE OR REPLACE FUNCTION dyn_trigger_func()
  RETURNS TRIGGER AS
$func$
BEGIN
   NEW := NEW #= hstore(TG_ARGV[0], TG_ARGV[1]);
   RETURN NEW;
END
$func$  LANGUAGE plpgsql;

和

CREATE TRIGGER ins_bef
BEFORE INSERT ON tbl
FOR EACH ROW EXECUTE PROCEDURE dyn_trigger_func('name', '2222');

提供未加引号和区分大小写的列名。

PostgreSQL plpgsql - 变量列名

3 个答案: