如何在PL/pgSQL的WHERE子句中用IF语句简化存储过程？

你可以通过以下几种方式简化重复的查询代码，避免写两次几乎一样的SELECT语句：

1. 合并WHERE条件（最简洁的方式） #

直接在WHERE子句中整合条件逻辑，不需要IF分支。利用逻辑运算符的短路特性，实现条件动态生效：

SELECT my_key, my_id
INTO l_my_key, l_my_id
FROM data
WHERE my_num = NEW.my_num 
  AND my_data = NEW.my_date
  AND (check_unit <> 'Y' OR my_unit = NEW.my_unit);

逻辑说明：

• 当check_unit不等于'Y'时，check_unit <> 'Y'为真，整个AND条件自动成立，不会校验my_unit
• 当check_unit等于'Y'时，必须满足my_unit = NEW.my_unit才能匹配数据

2. 使用动态SQL（适合复杂场景） #

如果后续还有更多动态变化的条件，可以用动态拼接SQL的方式，只修改差异部分：

DECLARE
  v_sql TEXT;
BEGIN
  -- 基础查询语句
  v_sql := 'SELECT my_key, my_id INTO l_my_key, l_my_id FROM data WHERE my_num = $1 AND my_data = $2';
  
  IF check_unit = 'Y' THEN
    -- 追加unit条件
    v_sql := v_sql || ' AND my_unit = $3';
    -- 带第三个参数执行
    EXECUTE v_sql USING NEW.my_num, NEW.my_date, NEW.my_unit;
  ELSE
    -- 直接执行基础语句
    EXECUTE v_sql USING NEW.my_num, NEW.my_date;
  END IF;
END;

注意：用USING传递参数可以避免SQL注入风险，不要直接拼接变量值到SQL字符串里。

3. 封装成公共函数（彻底解决多存储过程重复问题） #

既然这段查询在多个存储过程里重复出现，最好把它封装成一个独立函数，所有存储过程直接调用即可：

第一步：创建公共函数 #

CREATE OR REPLACE FUNCTION get_data_keys(
  p_my_num INT, 
  p_my_date DATE, 
  p_my_unit VARCHAR, 
  p_check_unit CHAR
) RETURNS TABLE(my_key INT, my_id INT) AS $$
BEGIN
  RETURN QUERY
  SELECT d.my_key, d.my_id
  FROM data d
  WHERE d.my_num = p_my_num
    AND d.my_data = p_my_date
    AND (p_check_unit <> 'Y' OR d.my_unit = p_my_unit);
END;
$$ LANGUAGE plpgsql;

第二步：在存储过程中调用 #

SELECT my_key, my_id INTO l_my_key, l_my_id 
FROM get_data_keys(NEW.my_num, NEW.my_date, NEW.my_unit, check_unit);

这种方式的好处是后续要修改查询逻辑时，只需要改这一个函数，所有调用它的存储过程都会自动生效，维护成本极低。

如果你的场景只是简单的单条件分支，第一种合并WHERE条件的写法就是PL/pgSQL里的标准简洁写法；如果多存储过程重复调用，优先用第三种封装函数的方案。

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