PostgreSQL中实现函数返回含原表全列与处理字段的动态表

首先修正你原函数中的几个关键错误：

• 字符串拼接语法错误：PostgreSQL中不能用||tablename||这种方式拼接变量，且ANYELEMENT类型参数需要通过pg_typeof()获取对应表名
• 参数引用错误：原函数中误用text作为参数名，实际参数是data
• 返回类型不匹配：声明返回SETOF ANYELEMENT（即传入表的行类型），但实际返回的是两个字符串，类型完全不匹配

要实现返回传入表的所有列+经过UPPER处理的new_data字段，必须借助动态SQL构造查询语句，同时处理动态返回结构的问题，具体实现如下：

正确的函数定义 #

CREATE OR REPLACE FUNCTION public.find_abc(tablename ANYELEMENT, data text)
RETURNS SETOF RECORD AS
$body$
DECLARE
    -- 从ANYELEMENT参数中提取表名
    target_table text := pg_typeof(tablename)::text;
    dynamic_query text;
BEGIN
    -- 校验传入的列是否存在于目标表中，避免无效查询
    IF NOT EXISTS (
        SELECT 1 
        FROM information_schema.columns
        WHERE table_schema = 'public'
          AND table_name = target_table
          AND column_name = data
    ) THEN
        RAISE EXCEPTION '列 "%" 不存在于表 "%"', data, target_table;
    END IF;

    -- 构造动态SQL：查询原表所有列 + 大写处理后的目标列
    dynamic_query := format(
        'SELECT g1.*, UPPER(g1.%I) AS new_data 
         FROM public.%I g1',
        data, -- 用%I自动处理列名的转义，避免SQL注入
        target_table
    );

    -- 执行动态查询并返回结果
    RETURN QUERY EXECUTE dynamic_query;
END;
$body$
LANGUAGE plpgsql;

调用方式 #

由于返回的是动态结构的RECORD，调用时需要显式指定返回列的名称和类型（与目标表结构+new_data字段对应）：

-- 示例：假设company表结构为(id int, name text, address text)
SELECT * 
FROM public.find_abc(NULL::company, 'name') 
AS result(id int, name text, address text, new_data text);

关键说明 #

1. ANYELEMENT的作用：通过传入NULL::表名的方式，让函数自动识别目标表的结构，避免硬编码表名
2. format()函数：使用%I占位符自动处理标识符（表名、列名）的转义，彻底避免SQL注入风险
3. 动态返回结构：借助SETOF RECORD实现动态列返回，调用时只需匹配目标表结构+新增的new_data字段即可

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