启用RLS策略的表全文搜索无法用索引的解决方案咨询

问题原因 #

你遇到的核心问题是：RLS策略中使用的自定义函数不是LEAKPROOF。PostgreSQL为防止未授权数据泄露，当RLS的USING条件包含非LEAKPROOF函数时，会放弃索引扫描，改为全表扫描后再过滤——因为它无法确认函数是否会在索引扫描过程中泄露敏感信息。而表所有者不受RLS限制，所以能正常使用索引。

由于RDS环境无法修改系统表来标记函数为LEAKPROOF，我们需要通过其他方式规避这个限制。

可行解决方案 #

1. 用内置表达式替代自定义RLS函数（优先推荐）

如果你的RLS逻辑可以用简单的内置表达式实现（比如示例中的inst_id < 4），直接替换掉自定义函数：

DROP POLICY pol_has_access ON mytable;
CREATE POLICY pol_has_access 
ON mytable 
FOR SELECT TO PUBLIC USING (inst_id < 4);

内置运算符（如<、=）都是LEAKPROOF的，PostgreSQL可以安全地将RLS过滤与全文索引扫描结合，此时非所有者查询会正常使用idx_text_vector索引。

2. 创建SECURITY DEFINER函数封装查询

如果RLS逻辑复杂，无法用简单表达式实现，可以创建一个SECURITY DEFINER函数来执行查询。函数会以表所有者身份运行，不受RLS限制，同时在内部加入权限检查逻辑：

CREATE OR REPLACE FUNCTION search_mytable(p_query TEXT)
RETURNS TABLE(inst_id INTEGER, text_vector TSVECTOR, rank REAL, q TSQUERY)
LANGUAGE plpgsql SECURITY DEFINER
AS $$
BEGIN
  -- 直接在函数内加入RLS权限检查，同时利用所有者身份使用索引
  RETURN QUERY
  SELECT m.inst_id, m.text_vector, ts_rank(m.text_vector, q) AS rank, q
  FROM mytable m, plainto_tsquery(p_query) q
  WHERE m.text_vector @@ q
    AND has_access(m.inst_id); -- 调用原权限检查函数
END;
$$;

-- 赋予普通用户函数执行权限
GRANT EXECUTE ON FUNCTION search_mytable(TEXT) TO PUBLIC;

普通用户通过调用该函数查询：

SELECT * FROM search_mytable('apple & orange');

这种方式既保证了数据权限控制，又能让查询利用全文索引。

3. 调整查询逻辑强制索引（临时应急方案）

如果上述方法都无法使用，可以临时强制PostgreSQL使用索引（不推荐长期使用，可能影响性能）：

SET LOCAL enable_seqscan = off;
SELECT inst_id, text_vector, ts_rank(text_vector, q) AS rank, q
FROM mytable, plainto_tsquery('apple & orange') q
WHERE text_vector @@ q;
SET LOCAL enable_seqscan = on;

这个方法只是临时绕过全表扫描，不适合生产环境长期使用。

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