SQL Server转PostgreSQL：如何用同一函数返回不同数据集？

没有一键转换的简便工具能完美适配这类逻辑，不过可以通过PostgreSQL的PL/pgSQL函数（或存储过程）实现完全等价的功能，以下是具体转换方案：

核心转换思路 #

原SQL Server存储过程根据参数@GetResultCount判断返回计数还是具体数据，PostgreSQL中用函数替代存储过程更适合这类返回结果集的场景，同时调整类型适配PostgreSQL语法规范：

• 将SQL Server的BIT类型替换为PostgreSQL的boolean（更符合逻辑判断语义）
• 将DateTime参数映射为PostgreSQL的timestamp（或timestamptz，根据业务时区需求选择）
• 用RETURN QUERY语句返回查询结果集，这是PL/pgSQL中返回多行数据的标准方式

转换后的函数代码 #

CREATE OR REPLACE FUNCTION GetMyData(
    p_start_time timestamp,
    p_end_time timestamp,
    p_get_result_count boolean DEFAULT false
)
RETURNS SETOF record AS $$
BEGIN
    IF p_get_result_count THEN
        RETURN QUERY SELECT COUNT(*)::bigint FROM myTable WHERE xDate >= p_start_time AND xDate < p_end_time;
    ELSE
        RETURN QUERY SELECT x, y, z FROM myTable WHERE xDate >= p_start_time AND xDate < p_end_time;
    END IF;
END;
$$ LANGUAGE plpgsql;

调用方式 #

• 获取计数：

SELECT * FROM GetMyData('2024-01-01 00:00:00'::timestamp, '2024-02-01 00:00:00'::timestamp, true) AS cnt_result(total bigint);

• 获取具体数据：

SELECT * FROM GetMyData('2024-01-01 00:00:00'::timestamp, '2024-02-01 00:00:00'::timestamp);

可选：用PostgreSQL存储过程实现（11+版本支持） #

如果需要事务控制等存储过程特性，也可以用以下写法：

CREATE OR REPLACE PROCEDURE GetMyData(
    p_start_time timestamp,
    p_end_time timestamp,
    p_get_result_count boolean DEFAULT false
)
LANGUAGE plpgsql AS $$
BEGIN
    IF p_get_result_count THEN
        SELECT COUNT(*) FROM myTable WHERE xDate >= p_start_time AND xDate < p_end_time;
    ELSE
        SELECT x, y, z FROM myTable WHERE xDate >= p_start_time AND xDate < p_end_time;
    END IF;
END;
$$;

调用时使用CALL语句：

CALL GetMyData('2024-01-01 00:00:00'::timestamp, '2024-02-01 00:00:00'::timestamp, true);

注意事项 #

• 若原表中xDate字段是带时区的时间类型（timestamptz），需将函数参数也改为timestamptz，避免时区转换错误
• COUNT(*)在PostgreSQL中返回bigint类型，显式转换::bigint可避免类型匹配问题

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