PostgreSQL中如何在包装函数内调用两个表返回函数并解决报错

错误原因分析 #

你遇到的query has no destination for result data错误，是因为在PL/pgSQL函数中直接用SELECT调用返回表的函数时，没有指定结果去向——PL/pgSQL要求SELECT的结果要么存入变量，要么明确返回出去，否则会触发该错误。而提示的PERFORM仅用于丢弃结果，不符合你需要返回数据的需求。

解决方案 #

根据get_table_data和get_table_data2返回的表结构是否一致，有两种实用处理方式：

情况1：两个函数返回的表结构完全相同 #

如果两个函数返回的是同一种表类型（比如都是user_info表结构），可以用RETURN QUERY将两个函数的结果合并成一个结果集返回，C#调用时可直接处理统一的结果集。

示例代码：

CREATE OR REPLACE FUNCTION get_table_data3()
RETURNS SETOF your_target_table_type AS $$ -- 替换为实际表类型名
BEGIN
    -- 返回第一个函数的结果
    RETURN QUERY SELECT * FROM get_table_data();
    -- 返回第二个函数的结果，与前一个结果合并
    RETURN QUERY SELECT * FROM get_table_data2();
END;
$$ LANGUAGE plpgsql;

调用方式：

SELECT * FROM get_table_data3();

情况2：两个函数返回的表结构不同 #

如果两个函数的返回结构差异较大，可选择以下两种方式：

方案A：返回JSON格式的合并结果

将两个函数的结果分别转为JSON数组，再打包成一个JSON对象返回，C#可通过JSON序列化工具解析。

示例代码：

CREATE OR REPLACE FUNCTION get_table_data3()
RETURNS JSON AS $$
DECLARE
    data_set1 JSON;
    data_set2 JSON;
BEGIN
    -- 将第一个函数结果转为JSON数组
    SELECT array_to_json(array_agg(t)) INTO data_set1 FROM get_table_data() t;
    -- 将第二个函数结果转为JSON数组
    SELECT array_to_json(array_agg(t)) INTO data_set2 FROM get_table_data2() t;
    -- 打包成包含两个数据集的JSON对象返回
    RETURN json_build_object('table1_data', data_set1, 'table2_data', data_set2);
END;
$$ LANGUAGE plpgsql;

调用方式：

SELECT get_table_data3();

方案B：使用游标返回多结果集

通过游标分别返回两个函数的结果，适合C#驱动（如Npgsql）处理多结果集的场景。

示例代码：

CREATE OR REPLACE FUNCTION get_table_data3(ref_cursor1 refcursor, ref_cursor2 refcursor)
RETURNS SETOF refcursor AS $$
BEGIN
    -- 打开第一个游标，指向第一个函数的结果
    OPEN ref_cursor1 FOR SELECT * FROM get_table_data();
    RETURN NEXT ref_cursor1;
    
    -- 打开第二个游标，指向第二个函数的结果
    OPEN ref_cursor2 FOR SELECT * FROM get_table_data2();
    RETURN NEXT ref_cursor2;
END;
$$ LANGUAGE plpgsql;

调用方式（需在事务中执行）：

BEGIN;
SELECT * FROM get_table_data3('cursor1', 'cursor2');
FETCH ALL IN cursor1; -- 获取第一个数据集
FETCH ALL IN cursor2; -- 获取第二个数据集
COMMIT;

C#中可通过NpgsqlDataReader依次读取多个结果集。

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