PostgreSQL plpgsql-变量列名

Question

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

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

但

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

给出一个错误：

错误：“ $ 1”或附近的语法错误LINE 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

您已经找到了我在 dba.SE 上推荐hstore操作员#=的答案。您可能还对 SO 上的相应参考答案感兴趣：

• 如何使用动态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');

提供不带引号且区分大小写的列名。

有关的：

• 从通用触发器中的不同列中获取值
• 使用动态字段名称触发