SQL 计算账户余额

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

代码语言:javascript
复制
    id  op      amount  
------  ------  --------
     1  in          1000
     2  exp          124
     3  exp           68
     4  exp          256
     5  in            88
     6  in           200
     7  exp           11
     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 。

代码语言:javascript
复制
SELECT 
  *,
  (SELECT 
    SUM(IF(op = 'exp', - 1 * amount, amount)) 
  FROM
    t_account b 
  WHERE b.id <= a.id) AS balance 
FROM
  t_account a

结果呈现 >>>

代码语言:javascript
复制
    id  op      amount  balance  
------  ------  ------  ---------
     1  in        1000  1000     
     2  exp        124  876      
     3  exp         68  808      
     4  exp        256  552      
     5  in          88  640      
     6  in         200  840      
     7  exp         11  829      
     8  exp        404  425      

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

代码语言:javascript
复制
SELECT 
  *,
  SUM(IF(op = 'exp', - 1 * amount, amount)) 
  OVER (ORDER BY id) AS balance 
FROM
  t_account a