PostgreSQL函数内遍历表名数组存储多表查询结果的实现咨询

• 动态SQL赋值语法不合法：PL/pgSQL中不能直接将EXECUTE的返回值写在赋值语句右侧，必须通过EXECUTE ... INTO 目标变量的语法承接单值查询结果，原代码直接写current_height:=EXECUTE quer会触发语法错误。
• 标识符拼接不规范：format函数使用%s直接拼接表名，既存在SQL注入风险，也无法处理带特殊字符、大小写敏感、与SQL关键字重名的表名场景，应改用专门处理标识符的%I占位符，自动完成合法转义。
• 数组更新逻辑失效：array_append()是纯函数，会返回拼接后的新数组，原代码仅裸写SELECT array_append(heights, current_height)但没有通过赋值语句将返回值存回heights变量，查询结果会被直接丢弃，heights数组永远不会被填充。
• 动态参数传递缺失：动态SQL的执行上下文与外层函数完全隔离，原代码在SQL文本中写的$2位置占位符没有配套USING子句传入函数入参idd，运行时会触发“参数不存在”的错误。

CREATE OR REPLACE FUNCTION array_iter(tables_name text[], idd integer)
RETURNS integer[]
LANGUAGE plpgsql
AS $BODY$
DECLARE
    current_table text;
    current_height integer;
    quer text;
    heights integer[] := '{}'; -- 初始化空数组，避免null值引发的逻辑异常
BEGIN
    -- 遍历表名数组逐次执行查询
    FOREACH current_table IN ARRAY tables_name
    LOOP
        -- 用%I占位符对表名做合法转义，避免拼接语法错误
        quer := format('SELECT height FROM %I WHERE idd = $1', current_table);
        
        -- 执行动态SQL，通过USING传入查询参数，结果赋值给变量
        EXECUTE quer INTO current_height USING idd;
        
        -- 显式将array_append的返回值赋值回原数组
        heights := array_append(heights, current_height);
    END LOOP;

    RAISE NOTICE '逐表查询收集的高度值：%', heights;
    RETURN heights; -- 直接返回数组结果供调用方使用，兼顾日志打印与功能实用性
END;
$BODY$;

• 执行查询前可以增加表存在性校验，比如通过查询information_schema.tables判断传入的表名是否有效，避免因无效表名直接中断函数执行。
• 所有外部传入的参数值统一通过USING子句传递给动态SQL，不要直接拼接进SQL文本，从根源规避SQL注入风险。
• 如果查询需求是合并多表的查询结果而非单纯攒值存数组，可以改用RETURN QUERY EXECUTE直接返回多表合并的结果集，比手动循环攒数组的执行效率更高。
• 对可能返回空结果的查询，可通过COALESCE对承接的变量做默认值处理，避免null值混入结果数组。

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