BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 3a6150f544444495a39a14087389fad278be5844

Authored by 芋道源码
Committed by GitHub
2 parents 3dcbf382 62a13d49

Merge pull request #1258 from dadiyang/fix/a60-wallet-balance-chain 

[BUG] 钱包并发消费：同钱包两笔并发扣款后流水余额链不连贯
Inline Side-by-side
Showing 1 changed file with 15 additions and 9 deletions
yudao-module-pay/src/main/java/cn/iocoder/yudao/module/pay/service/wallet/PayWalletServiceImpl.java
  View file @3a6150f
... ... @@ -153,7 +153,7 @@ public class PayWalletServiceImpl implements PayWalletService {
153 153  
154 154 // 2. 加锁，更新钱包余额（目的：避免钱包流水的并发更新时，余额变化不连贯）
155 155 return lockRedisDAO.lock(walletId, UPDATE_TIMEOUT_MILLIS, () -> {
156   - // 2. 扣除余额
  156 + // 2.1 扣除余额（CAS 原子扣减，先于重读，确保重读拿到扣减后最新值）
157 157 int updateCounts;
158 158 switch (bizType) {
159 159 case PAYMENT: {
... ... @@ -173,12 +173,14 @@ public class PayWalletServiceImpl implements PayWalletService {
173 173 throw exception(WALLET_BALANCE_NOT_ENOUGH);
174 174 }
175 175  
  176 + // 2.2 CAS 后重读钱包（事务内读自身更新，拿到扣减后最新余额；锁外预读的 balance 在并发时已过期，
  177 + // 且 @Transactional REPEATABLE READ 下 CAS 前重读仍读事务快照旧值，须 CAS 后重读）
  178 + PayWalletDO currentWallet = getWallet(walletId);
  179 +
176 180 // 3. 生成钱包流水
177   - // 情况一：充值退款：balance 在冻结时已扣，updateWhenRechargeRefund 只扣 freeze_price，所以 afterBalance 不变。https://t.zsxq.com/OJk9m
178   - // 情况二：消费支付：updateWhenConsumption 从 balance 扣，所以 afterBalance = balance - price
179   - Integer afterBalance = bizType == PayWalletBizTypeEnum.RECHARGE_REFUND
180   - ? payWallet.getBalance()
181   - : payWallet.getBalance() - price;
  181 + // 情况一：充值退款：balance 在冻结时已扣，updateWhenRechargeRefund 只扣 freeze_price，balance 不变，afterBalance=当前余额
  182 + // 情况二：消费支付：updateWhenConsumption 从 balance 扣 price，CAS 后重读的 balance 即扣减后值，afterBalance=当前余额
  183 + Integer afterBalance = currentWallet.getBalance();
182 184 WalletTransactionCreateReqBO bo = new WalletTransactionCreateReqBO().setWalletId(payWallet.getId())
183 185 .setPrice(-price).setBalance(afterBalance).setBizId(String.valueOf(bizId))
184 186 .setBizType(bizType.getType()).setTitle(bizType.getDescription());
... ... @@ -200,7 +202,7 @@ public class PayWalletServiceImpl implements PayWalletService {
200 202  
201 203 // 2. 加锁，更新钱包余额（目的：避免钱包流水的并发更新时，余额变化不连贯）
202 204 return lockRedisDAO.lock(walletId, UPDATE_TIMEOUT_MILLIS, () -> {
203   - // 3. 更新钱包金额
  205 + // 2.1 更新钱包金额（CAS 原子加，先于重读，确保重读拿到加值后最新值）
204 206 switch (bizType) {
205 207 case PAYMENT_REFUND: { // 退款更新
206 208 walletMapper.updateWhenConsumptionRefund(payWallet.getId(), price);
... ... @@ -219,9 +221,13 @@ public class PayWalletServiceImpl implements PayWalletService {
219 221 }
220 222 }
221 223  
222   - // 4. 生成钱包流水
  224 + // 2.2 CAS 后重读钱包（事务内读自身更新，拿到加值后最新余额；锁外预读的 balance 在并发时已过期，
  225 + // 且 @Transactional REPEATABLE READ 下 CAS 前重读仍读事务快照旧值，须 CAS 后重读）
  226 + PayWalletDO currentWallet = getWallet(walletId);
  227 +
  228 + // 3. 生成钱包流水（afterBalance = CAS 后最新余额，无需再 +price）
223 229 WalletTransactionCreateReqBO transactionCreateReqBO = new WalletTransactionCreateReqBO()
224   - .setWalletId(payWallet.getId()).setPrice(price).setBalance(payWallet.getBalance() + price)
  230 + .setWalletId(payWallet.getId()).setPrice(price).setBalance(currentWallet.getBalance())
225 231 .setBizId(bizId).setBizType(bizType.getType()).setTitle(bizType.getDescription());
226 232 return walletTransactionService.createWalletTransaction(transactionCreateReqBO);
227 233 });
... ...