PostgreSQL 13.2：如何为触发器函数的参数列名赋值？

错误原因 #

报错The record new has no field 'arg_column'的核心问题是：你直接将变量arg_column作为NEW的字段名引用，但PostgreSQL会把它解析成字面意义上的字段名（即找名为arg_column的字段），而非变量中存储的实际主键列名，所以才会提示字段不存在。

修正后的代码 #

declare 
    ll_maxid integer;
    arg_column text := TG_ARGV[0]::text;
begin
    if TG_WHEN = 'BEFORE' AND TG_OP = 'INSERT' THEN
        -- 获取指定主键列的最大值
        execute format('SELECT coalesce(max(%I), 0) + 1 FROM %I.%I', arg_column, TG_TABLE_SCHEMA, TG_TABLE_NAME) INTO ll_maxid;
        
        -- 动态给NEW的目标字段赋值
        execute format('SELECT $1.* %s', format('SET %I = $2', arg_column)) INTO new USING new, ll_maxid;
    end if;
    return new;
end;

关键修改说明 #

• 使用*%I*占位符处理列名、表名：这是PostgreSQL中处理标识符的安全写法，既能避免SQL注入，也能正确识别带特殊字符或大小写敏感的标识符。
• 动态赋值逻辑：通过EXECUTE构造修改NEW记录的语句，利用USING传递原NEW记录和计算出的最大值，避免直接拼接字符串带来的安全风险和语法错误。
• 语句逻辑：SELECT $1.* SET %I = $2会基于传入的原NEW记录，修改指定字段的值后生成新的记录，再赋值回NEW变量。

额外提示 #

• 确保传入的TG_ARGV[0]是目标表中真实存在的主键列名，否则会触发列不存在的错误。
• 这种手动生成自增ID的方式，性能和并发安全性远不如PostgreSQL原生的SERIAL或IDENTITY列，仅建议在无法使用原生自增的特殊场景下使用。

内容的提问来源于stack exchange，提问作者manaz