如何让PostgreSQL的DO块不受statement_timeout语句超时限制？

问题描述 #

预期执行以下代码完成300次循环（每次休眠1秒并提交）：

SET statement_timeout = 30000;
DO language plpgsql $$
DECLARE
  min_id integer := 1;
  max_id integer := 300;
BEGIN  
  FOR index IN min_id..max_id LOOP
    PERFORM pg_sleep(1);
    COMMIT;
    RAISE NOTICE '%', index;
  END LOOP;
END;
$$;

但实际执行到第29次时触发超时：

NOTICE:  1
NOTICE:  2
...
NOTICE:  29
ERROR:  canceling statement due to statement timeout
CONTEXT:  SQL statement "SELECT pg_sleep(1)"
PL/pgSQL function inline_code_block line 7 at PERFORM

核心问题：DO块被当作单条语句处理，即使内部执行COMMIT，整个DO块的总执行时间仍受statement_timeout限制。需要实现：既保留对普通长时语句的超时限制，又允许这类带内部提交的DO块不受限制运行。

解决方案 #

1. 在DO块内部临时调整超时 #

这是最直接的方法，在DO块开始时将statement_timeout设为0（无限制），执行结束后恢复原设置：

SET statement_timeout = 30000;
DO language plpgsql $$
DECLARE
  original_timeout integer;
  min_id integer := 1;
  max_id integer := 300;
BEGIN  
  -- 保存原超时设置
  SELECT current_setting('statement_timeout')::integer INTO original_timeout;
  -- 临时禁用超时
  SET LOCAL statement_timeout = 0;
  
  FOR index IN min_id..max_id LOOP
    PERFORM pg_sleep(1);
    COMMIT;
    RAISE NOTICE '%', index;
  END LOOP;
  
  -- 恢复原超时设置
  SET LOCAL statement_timeout = original_timeout;
END;
$$;

SET LOCAL的作用范围仅当前事务，而DO块内部每次COMMIT会开启新事务，所以每次循环都会继承这个设置，无需重复调整。

2. 改用函数而非DO块（更灵活） #

如果需要频繁执行这类任务，建议创建一个带事务控制的函数，在调用时临时调整超时：

CREATE OR REPLACE FUNCTION batch_process()
RETURNS void AS $$
DECLARE
  min_id integer := 1;
  max_id integer := 300;
BEGIN  
  FOR index IN min_id..max_id LOOP
    PERFORM pg_sleep(1);
    COMMIT;
    RAISE NOTICE '%', index;
  END LOOP;
END;
$$ LANGUAGE plpgsql;

调用时临时取消超时：

SET statement_timeout = 30000;
-- 调用前临时禁用超时
SET LOCAL statement_timeout = 0;
SELECT batch_process();
-- 自动恢复原超时（因为SET LOCAL作用于当前事务）

关键原理说明 #

PostgreSQL的statement_timeout是针对整个语句的执行时长，而非单个事务。DO块本身就是一条完整的语句，内部的COMMIT只是结束当前子事务，但整个DO块的执行时间仍在被持续统计，所以总时长超过30秒就会触发超时。

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