|
|
|
|
@ -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
2 years ago