PostgreSQL函数如何实现条件分支返回不同表

核心前提 #

PostgreSQL 中返回表类型的函数，返回结果的列数量、字段名、字段类型必须在函数定义阶段固定，因此要求待返回的表A、表B结构必须完全一致，且和函数声明的返回表结构匹配，否则会触发类型校验错误，和原SQL Server存储过程要求分支返回同结构表的逻辑对齐。

标准PL/pgSQL实现 #

使用PL/pgSQL过程语言，通过RETURN QUERY语法在条件分支中返回对应表的查询结果即可，示例代码如下：

-- 替换函数名、入参为你实际的业务参数
CREATE OR REPLACE FUNCTION sp_xxx(row_state char(1))
RETURNS TABLE (
  -- 此处按表A、表B的共同列结构逐一定义返回字段，示例字段请替换为实际业务字段
  col1 int,
  col2 varchar(100),
  col3 timestamp
) AS $$
BEGIN
  IF row_state = 'L' THEN
    -- 返回表A的结果，字段顺序、类型必须和上方RETURNS TABLE定义完全对应
    RETURN QUERY SELECT a.col1, a.col2, a.col3 FROM table_a a;
  ELSIF row_state = 'C' THEN
    -- 返回表B的结果，字段顺序、类型必须和上方RETURNS TABLE定义完全对应
    RETURN QUERY SELECT b.col1, b.col2, b.col3 FROM table_b b;
  END IF;
  -- 若row_state既不是L也不是C，默认返回空结果集，和原SQL Server逻辑一致
END;
$$ LANGUAGE plpgsql;

注意事项 #

• 若需要某个分支返回结果后立刻终止函数后续逻辑，可在对应分支的RETURN QUERY语句后追加单独的RETURN;语句，直接跳出函数体，不会执行后续代码。
• 若表A、表B本身字段顺序、类型、名称完全匹配返回定义，可以直接用SELECT *写法简化代码，无需逐列写字段名。
• 禁止在分支中返回列结构不一致的结果，PostgreSQL会在函数创建阶段做静态类型校验，结构不匹配会直接导致函数创建失败。
• 如果需要对返回结果做额外过滤、加工，直接在RETURN QUERY后的SELECT语句中追加WHERE条件、计算字段即可，和普通查询写法完全一致。

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