PostgreSQL动态拼接表名并将查询结果赋值给变量的方法

在PostgreSQL中，静态SQL无法直接用表达式作为表名，必须通过动态SQL实现。你可以借助EXECUTE语句配合format()函数（推荐）安全拼接表名，同时将查询结果赋值给变量。

正确实现代码示例 #

CREATE OR REPLACE FUNCTION your_custom_function()
RETURNS void AS $$
DECLARE
    ref_A int := 123;  -- 替换为你的实际变量值
    ref_B int := 456;
    ref_C int;
    target_table text;
BEGIN
    -- 拼接带空格的目标表名："table refA refB"
    target_table := format('table %s %s', ref_A, ref_B);
    
    -- 执行动态查询并将结果赋值给ref_C
    EXECUTE format('SELECT count(*) FROM %I', target_table)
    INTO ref_C;
    
    -- 可添加调试或业务逻辑，比如输出结果
    RAISE NOTICE '查询到的记录数：%', ref_C;
END;
$$ LANGUAGE plpgsql;

核心要点说明 #

• format()函数的%I占位符会自动将字符串处理为合法的PostgreSQL标识符（给带空格/特殊字符的表名自动添加双引号），既避免语法错误，也能防范SQL注入风险。
• EXECUTE ... INTO是PostgreSQL中动态SQL赋值变量的标准写法，能把动态执行的查询结果存入指定变量。
• 如果表名格式是下划线分隔（比如table_123_456），只需调整拼接规则：target_table := format('table_%s_%s', ref_A, ref_B)即可。

替代写法（使用quote_ident） #

如果你偏好quote_ident函数，也可以这样实现：

EXECUTE 'SELECT count(*) FROM ' || quote_ident('table ' || ref_A || ' ' || ref_B)
INTO ref_C;

不过format()的可读性和安全性更优，更推荐使用。

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