|
|
@@ -43,81 +43,124 @@ public class PayWalletServiceImpl implements PayWalletService {
|
|
|
private static final String WALLET_REFUND_NO_PREFIX = "WR";
|
|
|
|
|
|
@Resource
|
|
|
- private PayWalletMapper payWalletMapper;
|
|
|
+ private PayWalletMapper walletMapper;
|
|
|
@Resource
|
|
|
private PayNoRedisDAO noRedisDAO;
|
|
|
|
|
|
@Resource
|
|
|
- private PayWalletTransactionService payWalletTransactionService;
|
|
|
+ private PayWalletTransactionService walletTransactionService;
|
|
|
@Resource
|
|
|
@Lazy
|
|
|
- private PayOrderService payOrderService;
|
|
|
+ private PayOrderService orderService;
|
|
|
@Resource
|
|
|
@Lazy
|
|
|
- private PayRefundService payRefundService;
|
|
|
+ private PayRefundService refundService;
|
|
|
|
|
|
@Override
|
|
|
- public PayWalletDO getOrCreatePayWallet(Long userId, Integer userType) {
|
|
|
- PayWalletDO payWalletDO = payWalletMapper.selectByUserIdAndType(userId, userType);
|
|
|
- if (payWalletDO == null) {
|
|
|
- payWalletDO = new PayWalletDO();
|
|
|
- payWalletDO.setUserId(userId);
|
|
|
- payWalletDO.setUserType(userType);
|
|
|
- payWalletDO.setBalance(0);
|
|
|
- payWalletDO.setTotalExpense(0L);
|
|
|
- payWalletDO.setTotalRecharge(0L);
|
|
|
- payWalletDO.setCreateTime(LocalDateTime.now());
|
|
|
- payWalletMapper.insert(payWalletDO);
|
|
|
+ public PayWalletDO getOrCreateWallet(Long userId, Integer userType) {
|
|
|
+ PayWalletDO wallet = walletMapper.selectByUserIdAndType(userId, userType);
|
|
|
+ if (wallet == null) {
|
|
|
+ wallet = new PayWalletDO().setUserId(userId).setUserType(userType)
|
|
|
+ .setBalance(0).setTotalExpense(0L).setTotalRecharge(0L);
|
|
|
+ wallet.setCreateTime(LocalDateTime.now());
|
|
|
+ walletMapper.insert(wallet);
|
|
|
}
|
|
|
- return payWalletDO;
|
|
|
+ return wallet;
|
|
|
}
|
|
|
|
|
|
-
|
|
|
@Override
|
|
|
@Transactional(rollbackFor = Exception.class)
|
|
|
public PayWalletTransactionDO orderPay(Long userId, Integer userType, String outTradeNo, Integer price) {
|
|
|
- // 判断支付交易拓展单是否存
|
|
|
- PayOrderExtensionDO orderExtension = payOrderService.getOrderExtensionByNo(outTradeNo);
|
|
|
+ // 1. 判断支付交易拓展单是否存
|
|
|
+ PayOrderExtensionDO orderExtension = orderService.getOrderExtensionByNo(outTradeNo);
|
|
|
if (orderExtension == null) {
|
|
|
throw exception(ORDER_EXTENSION_NOT_FOUND);
|
|
|
}
|
|
|
+ // 2. 扣减余额
|
|
|
return reduceWalletBalance(userId, userType, orderExtension.getOrderId(), PAYMENT, price);
|
|
|
}
|
|
|
|
|
|
+ @Override
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public PayWalletTransactionDO orderRefund(String outRefundNo, Integer refundPrice, String reason) {
|
|
|
+ // 1.1 判断退款单是否存在
|
|
|
+ PayRefundDO payRefund = refundService.getRefundByNo(outRefundNo);
|
|
|
+ if (payRefund == null) {
|
|
|
+ throw exception(REFUND_NOT_FOUND);
|
|
|
+ }
|
|
|
+ // 1.2 校验是否可以退款
|
|
|
+ Long walletId = validateWalletCanRefund(payRefund.getId(), payRefund.getChannelOrderNo(), refundPrice);
|
|
|
+ PayWalletDO wallet = walletMapper.selectById(walletId);
|
|
|
+ Assert.notNull(wallet, "钱包 {} 不存在", walletId);
|
|
|
+ // 2. 增加余额
|
|
|
+ return addWalletBalance(wallet.getUserId(), wallet.getUserType(), payRefund.getId(), PAYMENT_REFUND, refundPrice);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 校验是否能退款
|
|
|
+ *
|
|
|
+ * @param refundId 支付退款单 id
|
|
|
+ * @param walletPayNo 钱包支付 no
|
|
|
+ */
|
|
|
+ private Long validateWalletCanRefund(Long refundId, String walletPayNo, Integer refundPrice) {
|
|
|
+ // 1. 校验钱包支付交易存在
|
|
|
+ PayWalletTransactionDO walletTransaction = walletTransactionService.getWalletTransactionByNo(walletPayNo);
|
|
|
+ if (walletTransaction == null) {
|
|
|
+ throw exception(WALLET_TRANSACTION_NOT_FOUND);
|
|
|
+ }
|
|
|
+ // 原来的支付金额
|
|
|
+ // TODO @jason：应该允许多次退款哈；
|
|
|
+ int amount = - walletTransaction.getPrice();
|
|
|
+ if (refundPrice != amount) {
|
|
|
+ throw exception(WALLET_REFUND_AMOUNT_ERROR);
|
|
|
+ }
|
|
|
+ PayWalletTransactionDO refundTransaction = walletTransactionService.getWalletTransaction(
|
|
|
+ String.valueOf(refundId), PAYMENT_REFUND);
|
|
|
+ if (refundTransaction != null) {
|
|
|
+ throw exception(WALLET_REFUND_EXIST);
|
|
|
+ }
|
|
|
+ return walletTransaction.getWalletId();
|
|
|
+ }
|
|
|
+
|
|
|
@Override
|
|
|
public PayWalletTransactionDO reduceWalletBalance(Long userId, Integer userType,
|
|
|
Long bizId, PayWalletBizTypeEnum bizType, Integer price) {
|
|
|
// 1.1 获取钱包
|
|
|
- PayWalletDO payWallet = getOrCreatePayWallet(userId, userType);
|
|
|
+ PayWalletDO payWallet = getOrCreateWallet(userId, userType);
|
|
|
// 1.2 判断余额是否足够
|
|
|
int afterBalance = payWallet.getBalance() - price;
|
|
|
if (afterBalance < 0) {
|
|
|
throw exception(WALLET_BALANCE_NOT_ENOUGH);
|
|
|
}
|
|
|
|
|
|
+ // TODO jason：建议基于 where price >= 来做哈；然后抛出 WALLET_BALANCE_NOT_ENOUGH
|
|
|
// 2.1 扣除余额
|
|
|
- int number = payWalletMapper.updateWhenDecBalance(bizType,payWallet.getBalance(), payWallet.getTotalRecharge(),
|
|
|
- payWallet.getTotalExpense(), price, payWallet.getId());
|
|
|
+ int number = walletMapper.updateWhenDecBalance(bizType, payWallet.getBalance(),
|
|
|
+ payWallet.getTotalRecharge(), payWallet.getTotalExpense(), price, payWallet.getId());
|
|
|
if (number == 0) {
|
|
|
throw exception(TOO_MANY_REQUESTS);
|
|
|
}
|
|
|
+
|
|
|
// 2.2 生成钱包流水
|
|
|
+ // TODO @jason：walletNo 交给 payWalletTransactionService 自己生成哈；
|
|
|
String walletNo = generateWalletNo(bizType);
|
|
|
PayWalletTransactionDO walletTransaction = new PayWalletTransactionDO().setWalletId(payWallet.getId())
|
|
|
.setNo(walletNo).setPrice(-price).setBalance(afterBalance)
|
|
|
.setBizId(String.valueOf(bizId)).setBizType(bizType.getType()).setTitle(bizType.getDescription());
|
|
|
- payWalletTransactionService.createWalletTransaction(walletTransaction);
|
|
|
+ // TODO @jason：是不是可以 createWalletTransaction 搞个 bo 参数，然后 PayWalletTransactionDO 交回给 walletTransactionService 更好；然后把参数简化下
|
|
|
+ walletTransactionService.createWalletTransaction(walletTransaction);
|
|
|
return walletTransaction;
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public PayWalletTransactionDO addWalletBalance(Long userId, Integer userType, Long bizId,
|
|
|
- PayWalletBizTypeEnum bizType, Integer price) {
|
|
|
+ public PayWalletTransactionDO addWalletBalance(Long userId, Integer userType,
|
|
|
+ Long bizId, PayWalletBizTypeEnum bizType, Integer price) {
|
|
|
// 1.1 获取钱包
|
|
|
- PayWalletDO payWallet = getOrCreatePayWallet(userId, userType);
|
|
|
+ PayWalletDO payWallet = getOrCreateWallet(userId, userType);
|
|
|
|
|
|
// 2.1 增加余额
|
|
|
- int number = payWalletMapper.updateWhenIncBalance(bizType, payWallet.getBalance(), payWallet.getTotalRecharge(),
|
|
|
+ // TODO @jason：类似上面的思路哈；
|
|
|
+ int number = walletMapper.updateWhenIncBalance(bizType, payWallet.getBalance(), payWallet.getTotalRecharge(),
|
|
|
payWallet.getTotalExpense(), price, payWallet.getId());
|
|
|
if (number == 0) {
|
|
|
throw exception(TOO_MANY_REQUESTS);
|
|
|
@@ -129,63 +172,23 @@ public class PayWalletServiceImpl implements PayWalletService {
|
|
|
.setNo(walletNo).setPrice(price).setBalance(payWallet.getBalance()+price)
|
|
|
.setBizId(String.valueOf(bizId)).setBizType(bizType.getType())
|
|
|
.setTitle(bizType.getDescription());
|
|
|
- payWalletTransactionService.createWalletTransaction(newWalletTransaction);
|
|
|
+ walletTransactionService.createWalletTransaction(newWalletTransaction);
|
|
|
return newWalletTransaction;
|
|
|
}
|
|
|
|
|
|
private String generateWalletNo(PayWalletBizTypeEnum bizType) {
|
|
|
+ // TODO @jason：对于余额来说，是不是直接 W+序号就行了，它其实不关注业务；；；不然就耦合啦
|
|
|
String no = "";
|
|
|
switch(bizType){
|
|
|
- case PAYMENT :
|
|
|
+ case PAYMENT:
|
|
|
no = noRedisDAO.generate(WALLET_PAY_NO_PREFIX);
|
|
|
break;
|
|
|
- case PAYMENT_REFUND :
|
|
|
+ case PAYMENT_REFUND:
|
|
|
no = noRedisDAO.generate(WALLET_REFUND_NO_PREFIX);
|
|
|
break;
|
|
|
default :
|
|
|
- // TODO 待增加
|
|
|
}
|
|
|
return no;
|
|
|
}
|
|
|
|
|
|
- @Override
|
|
|
- @Transactional(rollbackFor = Exception.class)
|
|
|
- public PayWalletTransactionDO orderRefund(String outRefundNo, Integer refundPrice, String reason) {
|
|
|
- // 1.1 判断退款单是否存在
|
|
|
- PayRefundDO payRefund = payRefundService.getRefundByNo(outRefundNo);
|
|
|
- if (payRefund == null) {
|
|
|
- throw exception(REFUND_NOT_FOUND);
|
|
|
- }
|
|
|
- // 1.2 校验是否可以退款
|
|
|
- Long walletId = validateWalletCanRefund(payRefund.getId(), payRefund.getChannelOrderNo(), refundPrice);
|
|
|
-
|
|
|
- PayWalletDO payWallet = payWalletMapper.selectById(walletId);
|
|
|
- Assert.notNull(payWallet, "钱包 {} 不存在", walletId);
|
|
|
- return addWalletBalance(payWallet.getUserId(), payWallet.getUserType(),payRefund.getId(), PAYMENT_REFUND, refundPrice);
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * 校验是否能退款
|
|
|
- *
|
|
|
- * @param refundId 支付退款单 id
|
|
|
- * @param walletPayNo 钱包支付 no
|
|
|
- */
|
|
|
- private Long validateWalletCanRefund(Long refundId, String walletPayNo, Integer refundPrice) {
|
|
|
- // 查询钱包支付交易
|
|
|
- PayWalletTransactionDO payWalletTransaction = payWalletTransactionService.getWalletTransactionByNo(walletPayNo);
|
|
|
- if (payWalletTransaction == null) {
|
|
|
- throw exception(WALLET_TRANSACTION_NOT_FOUND);
|
|
|
- }
|
|
|
- // 原来的支付金额
|
|
|
- int amount = - payWalletTransaction.getPrice();
|
|
|
- if (refundPrice != amount) {
|
|
|
- throw exception(WALLET_REFUND_AMOUNT_ERROR);
|
|
|
- }
|
|
|
- PayWalletTransactionDO refundTransaction = payWalletTransactionService.getWalletTransaction(
|
|
|
- String.valueOf(refundId), PAYMENT_REFUND);
|
|
|
- if (refundTransaction != null) {
|
|
|
- throw exception(WALLET_REFUND_EXIST);
|
|
|
- }
|
|
|
- return payWalletTransaction.getWalletId();
|
|
|
- }
|
|
|
}
|
YunaiV