PostgreSQL递归函数的调用深度、调用次数查询及迭代改造咨询

一、如何追踪递归函数的调用深度与调用次数 #

1. 实时获取调用深度 #

PostgreSQL本身没有直接提供递归调用深度的内置函数，但我们可以用会话级自定义变量来手动追踪：
在你的generategrid函数内部，每次进入时递增一个变量，退出时递减，这样就能实时获取当前的调用深度。示例代码如下：

CREATE OR REPLACE FUNCTION public.generategrid(parameters) RETURNS SETOF text AS $BODY$
DECLARE
  current_depth integer;
  -- 你的其他变量声明...
BEGIN
  -- 获取当前会话的深度值，默认初始为0
  current_depth := coalesce(current_setting('generategrid.depth')::integer, 0);
  -- 进入函数，深度+1
  PERFORM set_config('generategrid.depth', (current_depth + 1)::text, true);

  -- 这里是你原本的函数逻辑...

  -- 函数执行完毕，深度回退
  PERFORM set_config('generategrid.depth', current_depth::text, true);
  
  -- 你的返回逻辑...
END;
$BODY$ LANGUAGE plpgsql;

之后在函数运行过程中，你可以在同一个会话里执行SELECT current_setting('generategrid.depth');，就能看到当前的递归调用深度了。

2. 统计调用次数 #

有两种方式可以统计函数的调用次数：

• 会话内实时统计：同样用会话变量，每次调用函数时递增计数：
  在函数开头加入PERFORM set_config('generategrid.calls', (coalesce(current_setting('generategrid.calls')::integer, 0) + 1)::text, true);，之后用SELECT current_setting('generategrid.calls');查看累计次数。
• 系统视图统计：PostgreSQL的pg_stat_user_functions视图会记录所有用户函数的调用情况，执行以下查询就能拿到generategrid的累计调用次数：

SELECT calls 
FROM pg_stat_user_functions 
WHERE funcname = 'generategrid' AND schemaname = 'public';

注意这个视图的统计是从数据库启动或执行pg_stat_reset()后开始计数的，每次递归调用都会被算作一次调用。

二、把递归函数改成迭代方式的思路 #

递归改迭代的核心是用循环+任务队列/栈来模拟递归的调用栈，避免递归带来的栈溢出风险，同时通常能提升性能。针对你的generategrid函数，大概的改造步骤如下：

1. 定义任务存储结构：用数组、临时表或者自定义类型来存储需要处理的任务参数，比如如果你的递归是分块处理网格，就把每一块的参数存起来。
2. 初始化任务队列：把初始的网格参数放入队列中。
3. 循环处理任务：开启WHILE循环，只要队列不为空，就取出一个任务处理：
   • 处理当前任务，生成需要返回的结果（用RETURN NEXT或RETURN QUERY输出）。
   • 如果原本需要递归调用的场景，就把新的子任务参数加入队列，而不是直接调用递归函数。
4. 结束循环：当队列中的任务全部处理完毕，退出循环并结束函数。

给你一个简化的示例框架（你需要根据自己的实际参数和逻辑调整）：

CREATE OR REPLACE FUNCTION public.generategrid(parameters) RETURNS SETOF text AS $BODY$
DECLARE
  -- 假设用数组存储任务参数，根据你的实际参数类型调整
  task_queue your_parameter_type[];
  current_task your_parameter_type;
  output_text text;
BEGIN
  -- 初始化任务队列，放入初始参数
  task_queue := ARRAY[your_initial_parameters];

  WHILE array_length(task_queue, 1) > 0 LOOP
    -- 取出队列第一个任务
    current_task := task_queue[1];
    task_queue := task_queue[2:array_length(task_queue, 1)];

    -- 处理当前任务，生成输出内容
    output_text := your_processing_logic(current_task); -- 替换成你的实际处理代码

    -- 返回结果
    RETURN NEXT output_text;

    -- 生成子任务并加入队列（原本递归调用的地方）
    -- 比如假设处理current_task后得到两个子任务sub_task1、sub_task2
    task_queue := task_queue || ARRAY[sub_task1, sub_task2];
  END LOOP;

  RETURN;
END;
$BODY$ LANGUAGE plpgsql;

这种方式完全用循环替代了递归，避免了递归调用的栈开销，对于深层递归的场景性能提升会很明显。

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