自定义变量替代会话启动函数调用，缓存RLS参数缓解CPU尖峰

算力阿峰

2026-6-27

问题分析与优化方案

问题背景

函数f1()每日执行超百万次，引发CPU尖峰。该函数会在每个网页请求、批处理作业建立数据库连接时被调用，用于获取RLS（行级安全）相关的会话参数。当前单次函数执行仅需2ms，但高调用量的累积消耗导致了资源瓶颈。

相关代码示例：
设置会话变量的逻辑：

PERFORM set_config (namespace || '.'||attribute,value,false);

f1()函数代码：

CREATE OR REPLACE FUNCTION f1() 
RETURNS character varying LANGUAGE plpgsql STABLE AS $function$ 
DECLARE
   v_status boolean; 
BEGIN 
   return current_setting('ctx_vpd'||'.'||'ctx_vpd_fil'); 
   exception when undefined_object then return 'ERROR'; 
END;
$function$;

可行的缓存优化方案

1. 基于连接复用的会话级缓存

PostgreSQL的会话变量本身是会话级持久化的，只要数据库连接不关闭，变量值就会保留。最核心的优化是：

应用层启用连接池（如PgBouncer、应用内置连接池），复用已建立的数据库连接，而非每次请求都新建连接
每个连接仅在首次建立时执行一次set_config和f1()调用，后续请求直接复用已配置好的会话变量，彻底避免重复执行函数

2. 连接池预配置会话变量

如果使用PgBouncer这类连接池，可以在配置中添加连接初始化语句，在连接分配给应用前自动完成RLS参数配置。应用拿到连接时，会话变量已就绪，无需调用f1()。

示例PgBouncer配置（pgbouncer.ini）：

[databases]
your_db = host=localhost port=5432 dbname=your_db user=your_user password=your_pass
init_connect = "SELECT set_config('ctx_vpd.ctx_vpd_fil', 'target_value', false);"

3. 函数轻量化改造

如果必须保留函数调用，可简化f1()的实现，进一步降低开销：

CREATE OR REPLACE FUNCTION f1() 
RETURNS character varying LANGUAGE sql STABLE AS $function$ 
SELECT current_setting('ctx_vpd.ctx_vpd_fil', true); -- 第二个参数设为true，未定义时返回null而非抛出异常
$function$;

改用SQL函数替代PL/pgSQL函数，减少函数调用的额外开销。

4. 应用层缓存

若RLS参数在较长时间内不会变更，可在应用层（如内存缓存、本地缓存）存储变量值，仅在首次获取或参数更新时调用f1()。需注意设置合理的缓存失效机制，确保与数据库端的参数变更同步。

关键注意事项

会话变量与数据库连接绑定，连接复用是最高效的优化方案，能完全消除重复调用的开销
若RLS参数需按用户动态调整，连接池需使用会话模式而非事务模式，确保不同用户的连接参数相互独立

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