递归累积函数 - 重复使用结果行作为输入

Question

我尝试过数组、横向视图、递归视图、自定义函数、变量......我失去了希望。

我正在寻找一个可以从下表中产生“result_good”的函数。

“result_good”是“trx”和其自身之前的行的累积乘以“event”。

with t(id, trx, event, result_good) as
(values 
(1, 20, 0.1, 2.000000),
(2,-10, 0.1, 1.200000),
(3, 20,-0.1,-3.320000),
(4,-10, 0.1, 1.988000),
(5, 20, 0.1, 4.186800),
(6,-10,-0.1,-3.605480),
(7, 20, 0.1, 5.244932)
) 

-- non-recursive approximation of intended result
select *,
prev + event*sum(prev) over(
    order by id range between unbounded preceding and 1 preceding
    ) as not_quite_my_tempo
from
    (select t.*, event*sum(trx) over(order by id) as prev
     from t
    ) t
order by id

x86_64-pc-linux-gnu 64 位上的 PostgreSQL 13.6

Answer 1

窗口函数不进行递归操作，并且递归 CTE 不允许在递归项中进行聚合。

每个结果都会递归地贡献给下一行。此函数写入临时表并重用结果：

CREATE OR REPLACE FUNCTION f_special_running_sum()
  RETURNS TABLE(id int, result text)
  LANGUAGE plpgsql AS
$func$
DECLARE
   t record;
BEGIN
   DROP TABLE IF EXISTS pg_temp.result;
   CREATE TEMP TABLE result (id int, trx int, result float8) ON COMMIT DROP;

   FOR t IN 
      TABLE tbl ORDER BY id  -- your actual table name here!
   LOOP
      INSERT INTO result(id, trx, result)
      SELECT t.id, t.trx
           , (COALESCE(sum(r.trx + r.result), 0) + t.trx) * t.event
      FROM   result r;
   END LOOP;

   -- format output   
   RETURN QUERY
   SELECT r.id, to_char(r.result, 'FM9999990.000000')
   FROM   result r;
END
$func$;

db<>fiddle 此处< /a>

呼叫：

SELECT * FROM f_special_running_sum();

我将结果号码格式化为文本以匹配您想要的结果准确。您可能需要数字或双精度。相应地进行调整。

对于大型源表，性能将会恶化，因为下一行的成本不断增长。类似于 O(N²)。

小心地对所有查询中的列名称进行表限定，因为相同的名称可用于多种目的。

---

或者，递归函数也可以工作。示例：

• 从递归树结构创建 JSON 对象