PostgreSQL：选择行，直到单列的总和减少超过阈值

问题描述

我想编写一个从有序表中获取行的查询，同时聚合一列的值，直到所述聚合值达到所需的阈值。

另一个条件是，查询结果中应包括超过阈值的违规行。

我寻找了在PostgreSQL中完成的其他解决方案，因此创建了以下查询：

SELECT * FROM (
    SELECT *, SUM(amount) OVER (ORDER BY amount DESC) AS running_amount
    FROM public.orders WHERE price = 0.09) AS t
WHERE t.running_amount <= 15;

但是，此查询的问题是它代表的是一个PostgreSQL窗口查询，如果给定行的列值不是唯一的，则该查询跳过对所有行的列值的聚合。

遗憾的是，窗口查询不支持考虑不同值列。

我听说仍能实现这一点的一些替代方法是通过创建一个PostgreSQL函数，尽管我不知道从哪里开始执行这种聚合查询。

如果有人有什么想法或诀窍，我将不胜感激。

推荐答案

在窗口函数的ORDER BY子句中添加唯一列(主键)，例如：

SELECT * FROM (
    SELECT *, SUM(amount) OVER (ORDER BY amount DESC, id) AS running_amount
    FROM public.orders WHERE price = 0.09
) AS t
WHERE t.running_amount <= 15;

如果缺少唯一列，可以使用系统列ctid.

---

您可以使用UNION ALL获取超过阈值的违规行，例如：

WITH cte AS (
    SELECT *, SUM(amount) OVER (ORDER BY amount DESC, id) AS running_amount
    FROM public.orders 
    WHERE price = 0.09
)
SELECT * 
FROM cte
WHERE running_amount <= 15
UNION ALL (
    SELECT * 
    FROM cte
    WHERE running_amount > 15
    LIMIT 1
);

这篇关于PostgreSQL：选择行，直到单列的总和减少超过阈值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！