PostgreSQL函数如何返回多表关联的动态全字段查询结果？

问题分析 #

你原有代码存在三个核心错误：

• format函数定义了2个表占位符，仅传入了第一个表的类型参数，触发参数不匹配报错
• WHERE条件语法错误，直接拼接_id变量到SQL字符串既存在SQL注入风险，也错误构造了s1.'||_id||''的非法语法，没有用到预留的USING传参逻辑
• 返回值声明为SETOF anyelement，仅能匹配单表的行结构，两表JOIN后的行结构不属于任意单个传入表的类型，触发类型不匹配报错

实现方案 #

方案1：固定关联表的简化版本

如果你的需求固定为查询table1关联table2的结果，不需要动态适配其他表，可直接使用以下极简实现，无需指定任何列名：

CREATE OR REPLACE FUNCTION get_data(_id int)
  RETURNS SETOF record
  LANGUAGE plpgsql AS
$func$
BEGIN
   RETURN QUERY
   SELECT * FROM table1 t1 
   JOIN table2 t2 ON t1.id = t2.t2id 
   WHERE t1.id = _id;
END
$func$;

调用时通过LIKE语法复用表结构，无需手动枚举所有列：

SELECT * FROM get_data(123) AS t(LIKE table1, LIKE table2);

方案2：支持动态指定关联表的通用版本

如果需要适配不同的关联表场景，可使用多态参数实现通用逻辑：

CREATE OR REPLACE FUNCTION get_data(_tbl1 anyelement, _tbl2 anyelement, _id int)
  RETURNS SETOF record
  LANGUAGE plpgsql AS
$func$
BEGIN
   RETURN QUERY EXECUTE format(
      'SELECT *
       FROM %I s1
       JOIN %I s2 ON s1.id = s2.t2id
       WHERE s1.id = $1',
      pg_typeof(_tbl1)::text,
      pg_typeof(_tbl2)::text
   )
   USING _id;
END
$func$;

调用方式：

SELECT * FROM get_data(NULL::table1, NULL::table2, 123) AS t(LIKE table1, LIKE table2);

若两表存在重名列，可在查询的SELECT部分为重名列指定别名，或在调用时的t()结构中显式声明重名列的名称与类型即可。

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