PostgreSQL RLS策略含EXISTS时未定义参数报错求助

需求场景 #

要创建PostgreSQL行级安全（RLS）策略，满足以下两种生效模式，无需同时定义两个配置参数：

• 当user_param设为true时，允许查询全表，无需定义company_param
• 仅设置company_param时，按公司ID条件过滤表数据

正常工作的策略代码 #

仅使用current_setting的OR判断时，只定义其中一个参数即可正常查询：

using( (current_setting('user_param') = 'true') OR (current_setting('company_param') = 'true') )

出现问题的策略代码 #

将其中一个条件替换为EXISTS子查询后，必须同时定义两个参数，否则会报错ERROR: unrecognized configuration parameter "company_param" SQL state: 42704：

USING (((current_setting('user_param'::text) = 'true'::text) OR (EXISTS ( SELECT 1 FROM company c WHERE ((rls_test.company_id = c.company_id) AND (c.company_current_id = (current_setting('company_param'::text))::bigint))))))

问题原因 #

核心在于PostgreSQL的表达式求值逻辑与查询优化器的行为差异：

• 纯OR条件场景下，PostgreSQL会执行短路求值：如果第一个条件current_setting('user_param') = 'true'为真，就不会去计算第二个条件，因此即使company_param未定义也不会触发错误。
• 但当OR分支包含EXISTS子查询时，查询优化器可能会提前解析或执行子查询中的表达式（包括current_setting('company_param')），不会严格遵循短路求值逻辑。因为子查询的执行计划可能被优化器提前处理，导致无论第一个条件是否为真，都会先尝试解析company_param，参数未定义时直接抛出错误。

解决方法 #

给current_setting添加第三个参数missing_ok => true，当参数未定义时返回NULL而非报错，配合条件判断即可实现需求：

USING (
  (current_setting('user_param', true) = 'true') 
  OR 
  (EXISTS ( 
    SELECT 1 FROM company c 
    WHERE rls_test.company_id = c.company_id 
      AND c.company_current_id = (current_setting('company_param', true))::bigint
  ))
)

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