SQL 计算账户余额

有一张简单的账户表 t_account，它记录了每次支出（或收入）的金额，只是缺了余额字段，我们需要在每笔账单后面计算出当时的账户余额。 t_account 的部分数据：

1    id  op      amount  
2------  ------  --------
3     1  in          1000
4     2  exp          124
5     3  exp           68
6     4  exp          256
7     5  in            88
8     6  in           200
9     7  exp           11
10     8  exp          404

其中，id 越大表示这条记录产生的时间越近，op 字段是操作类型，‘in’ 表示收入，‘exp’ 表示支出，amount 是每次操作的金额。

具体来说，当 id = 1 时，账户上增加了 1000，此时余额是 1000；

当 id = 2 时，账户减去了 124，此时余额是 1000 - 124 = 876；

当 id = 3 时，账户又减去了 68，此时余额是 1000 - 124 - 68 = 808；

直到 id = 5，账户上才又有了一笔收入，此时余额是 1000 - 124 - 68 - 256 + 88 = 640 。

最终算到 id = 8 时，账户的余额是 425 。

实际上，当 id = x 时，它余额就是将 id 小于等于 x 的所有记录的 amount 累加，如果遇到 op 的类型是 ‘exp’ 的记录，则相应的加法操作变成减法。

话不多说，直接上 SQL 。

1SELECT 
2  *,
3  (SELECT 
4    SUM(IF(op = 'exp', - 1 * amount, amount)) 
5  FROM
6    t_account b 
7  WHERE b.id <= a.id) AS balance 
8FROM
9  t_account a

结果呈现 >>>

1    id  op      amount  balance  
2------  ------  ------  ---------
3     1  in        1000  1000     
4     2  exp        124  876      
5     3  exp         68  808      
6     4  exp        256  552      
7     5  in          88  640      
8     6  in         200  840      
9     7  exp         11  829      
10     8  exp        404  425      

如果用窗口函数，那累加的写法会更简单些。

1SELECT 
2  *,
3  SUM(IF(op = 'exp', - 1 * amount, amount)) 
4  OVER (ORDER BY id) AS balance 
5FROM
6  t_account a 

本文分享自微信公众号 - SQL实现（gh_684ee9235a26）。
如有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一起分享。