Citus单节点模式下事务/存储过程中UPDATE并行执行失效求助

问题根源 #

Citus默认在显式事务、存储过程（含SQL函数）内会限制分布式修改的并行性，这是出于多节点场景下事务一致性的考量，但单节点环境完全可以放开这个限制。

关键配置调整 #

1. 允许事务内并行修改 #

开启citus.enable_parallel_modifications_in_transaction，这是解决事务/存储过程中并行失效的核心开关：

-- 临时生效
SET citus.enable_parallel_modifications_in_transaction = on;
-- 永久生效需修改postgresql.conf，添加/修改后重启服务

2. 强制函数内并行化 #

针对LANGUAGE SQL函数，还需打开citus.force_function_parallelization，确保函数内的分布式查询走并行计划：

SET citus.force_function_parallelization = on;

3. 确认执行池容量 #

确保citus.max_adaptive_executor_pool_size不小于你的分片数（比如你这里是32分片，设为32或更高）：

SET citus.max_adaptive_executor_pool_size = 32;

验证步骤 #

1. 写个测试存储过程：

CREATE OR REPLACE PROCEDURE test_parallel_update()
LANGUAGE plpgsql
AS $$
BEGIN
    UPDATE my_distributed_table SET some_col = some_col + 1 WHERE filter_col < 10000;
END;
$$;

2. 调用后查会话：

CALL test_parallel_update();
-- 查看并行会话
SELECT pid, query FROM pg_stat_activity WHERE query LIKE '%my_distributed_table%';

此时应该能看到32个左右的并行会话在执行分片更新。

注意点 #

• 这些配置只适合单节点Citus，多节点集群开启可能引发跨节点事务一致性问题。
• 确保你的UPDATE语句WHERE条件能正确命中分片路由，避免全分片扫描拖慢效率。

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