从表中获取总和小于值的行

Question

考虑我在表中有以下行

|  id |   user_id   |  amount |      date      |
------------------------------------------------
|  1  |     1       |  100    |   2019-09-30   |
------------------------------------------------
|  2  |     2       |  100    |   2019-09-30   |
------------------------------------------------
|  3  |     1       |  100    |   2019-09-30   |
------------------------------------------------
|  4  |     3       |  100    |   2019-10-01   |
------------------------------------------------
|  5  |     1       |  75     |   2019-10-01   |
------------------------------------------------
|  6  |     3       |  100    |   2019-10-01   |
------------------------------------------------
|  7  |     1       |  35     |   2019-10-01   |
------------------------------------------------

我正在尝试找到一种方法来获取带有user_id = 1 的所有行，其中sum(amount) < 300 和date <= '2019-10-01'。

我想要做的是只处理满足某个阈值总和的记录。我不太确定从哪里开始。

预期结果

|  id |   user_id   |  amount |      date      |
------------------------------------------------
|  1  |     1       |  100    |   2019-09-30   |
------------------------------------------------
|  3  |     1       |  100    |   2019-09-30   |
------------------------------------------------
|  5  |     1       |   75    |   2019-10-01   |
------------------------------------------------

这是我目前尝试过的

SELECT id, SUM(amount) as total_sum 
   FROM table 
   WHERE date <= '2019-10-01' AND user_id = 1
     ORDER BY date ASC
   HAVING total_sum <= 300

根据上述查询，我​​没有得到所需的输出。

当前使用的 MySQL 版本：5.7.25

我确实看过这个问题MySQL select records with sum greater than threshold 假设他们正在尝试做同样的事情，但这不是我正在看的

Answer 1

如果您不喜欢使用会话变量（一些有经验的 SO 用户非常不喜欢它们），您可以使用基于“Self-Join”的技术，然后使用 GROUP BY 和 HAVING 进行过滤。

一般的思路是，我们left join 获取特定user_id 的前几行，然后聚合得到rolling sum，然后使用Having 子句过滤。

查询

SELECT 
  t1.* 
FROM 
  your_table AS t1 
LEFT JOIN your_table AS t2 
  ON t2.user_id = t1.user_id 
     AND t2.date <= t1.date 
     AND t2.id <= t1.id 
WHERE t1.user_id = 1 
  AND t1.date <= '2019-10-31'
GROUP BY t1.user_id, t1.date, t1.id, t1.amount
HAVING COALESCE(SUM(t2.amount),0) < 300;

结果

| id  | user_id | amount | date       |
| --- | ------- | ------ | ---------- |
| 1   | 1       | 100    | 2019-09-30 |
| 3   | 1       | 100    | 2019-09-30 |
| 5   | 1       | 75     | 2019-10-01 |

View on DB Fiddle

您可以对这两种方法进行基准测试，然后决定哪种方法合适。

对于此查询，您将需要复合索引：(user_id, date)

Answer 2

这是一个滚动求和问题。在 MySQL 8.0.2 及更高版本中，您可以使用Window functions with Frames 解决此问题。在旧版本中，我们可以使用用户定义的 Session 变量来做同样的事情。

• 我们首先使用Session variables 计算滚动和。
• 然后，使用Derived table中的结果集，找到总和越过“障碍”300的id。当新滚动总和大于300时达到障碍。我们将障碍值设置为此时为 1，之前的行为 0，之后的行为 2 或更多。
• 我们将只考虑障碍为 0 的行。

尝试（适用于所有 MySQL 版本）：

查询 #1

SELECT dt.id, 
       dt.user_id, 
       dt.amount, 
       dt.date
FROM 
(
  SELECT 
    t.id, 
    t.user_id, 
    t.amount, 
    t.date, 
    @barrier := CASE
                  WHEN 
                       (@tot_qty := @tot_qty + t.amount) > 300
                  THEN (@barrier + 1)
                  ELSE 0
                END AS barrier 
  FROM 
    your_table AS t 
  CROSS JOIN (SELECT @tot_qty := 0, 
                     @barrier := 0) AS user_init
    WHERE t.user_id = 1 
        AND t.date <= '2019-10-01' 
  ORDER BY t.user_id, t.date, t.id
) AS dt 
WHERE dt.barrier = 0 
ORDER BY dt.user_id, dt.date, dt.id;

结果

| id  | user_id | amount | date       |
| --- | ------- | ------ | ---------- |
| 1   | 1       | 100    | 2019-09-30 |
| 3   | 1       | 100    | 2019-09-30 |
| 5   | 1       | 75     | 2019-10-01 |

---

View on DB Fiddle