如何在PostgreSQL函数中根据输入值执行不同查询？

方法1：将条件整合到CTE的WHERE子句中 #

这种写法无需分支逻辑，直接通过输入参数动态筛选数据，简洁高效：

CREATE OR REPLACE FUNCTION public.get_data(in_param smallint)
RETURNS TABLE (
    -- 请根据tbl1和tbl2的实际字段替换以下占位符
    tbl1_col1 int,
    tbl1_col2 text,
    tbl2_col1 int,
    tbl2_col3 numeric
) AS $$
BEGIN
    RETURN QUERY
    WITH cte1 AS (
        SELECT * FROM tbl1
    ),
    cte2 AS (
        SELECT * FROM tbl2
        WHERE 
            (in_param = 1 AND tbl2.col1 > 5)
            OR (in_param != 1 AND tbl2.col3 < 0)
    )
    SELECT 
        cte1.*, cte2.*
    FROM cte1
    JOIN cte2 ON -- 补充cte1和cte2的实际关联条件，避免笛卡尔积
        cte1.id = cte2.tbl1_id;
END;
$$ LANGUAGE plpgsql;

注意点： #

• 必须明确返回表的字段结构（也可以用RETURNS SETOF record，但调用时需指定字段，不如直接定义方便）
• 关联cte1和cte2时一定要加实际连接条件，原示例的笛卡尔积写法通常不符合业务需求
• 这种方式性能更优，PostgreSQL能对条件做优化

方法2：使用动态SQL（适合复杂分支场景） #

如果不同分支的CTE查询逻辑差异较大，比如字段选择、关联表不同，可以用动态SQL实现：

CREATE OR REPLACE FUNCTION public.get_data(in_param smallint)
RETURNS TABLE (
    tbl1_col1 int,
    tbl1_col2 text,
    tbl2_col1 int,
    tbl2_col3 numeric
) AS $$
DECLARE
    cte2_sql text;
BEGIN
    -- 根据参数构建cte2的查询语句
    IF in_param = 1 THEN
        cte2_sql := 'SELECT * FROM tbl2 WHERE col1 > 5';
    ELSE
        cte2_sql := 'SELECT * FROM tbl2 WHERE col3 < 0';
    END IF;

    -- 执行动态SQL并返回结果
    RETURN QUERY EXECUTE format(
        'WITH cte1 AS (SELECT * FROM tbl1), cte2 AS (%s)
         SELECT cte1.*, cte2.* FROM cte1 JOIN cte2 ON cte1.id = cte2.tbl1_id',
        cte2_sql
    );
END;
$$ LANGUAGE plpgsql;

注意点： #

• 用format函数拼接SQL，避免字符串拼接的安全风险
• 如果参数是字符串类型，必须用USING传递参数防止SQL注入（本示例参数为smallint，风险较低，但建议养成习惯）
• 动态SQL适合复杂分支场景，但调试难度相对高一些

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