From 0ac9091ca1afc9d7e9516e71e0e2806c352d0d5d Mon Sep 17 00:00:00 2001 From: cloudroam <cloudroam> Date: 星期三, 23 十月 2024 17:35:25 +0800 Subject: [PATCH] fix:结算 --- src/main/java/com/mzl/flower/service/payment/OrderSettlementService.java | 63 +++++++++++++++++++------------ 1 files changed, 38 insertions(+), 25 deletions(-) diff --git a/src/main/java/com/mzl/flower/service/payment/OrderSettlementService.java b/src/main/java/com/mzl/flower/service/payment/OrderSettlementService.java index a012bb8..3fa2f6a 100644 --- a/src/main/java/com/mzl/flower/service/payment/OrderSettlementService.java +++ b/src/main/java/com/mzl/flower/service/payment/OrderSettlementService.java @@ -6,6 +6,7 @@ import com.mzl.flower.config.exception.ValidationException; import com.mzl.flower.config.security.SecurityUtils; import com.mzl.flower.constant.Constants; +import com.mzl.flower.constant.LockConstants; import com.mzl.flower.dto.request.payment.*; import com.mzl.flower.dto.response.payment.OrderSettlementDTO; import com.mzl.flower.dto.response.payment.OrderSettlementDetailDTO; @@ -27,8 +28,11 @@ import com.mzl.flower.mapper.wallet.WalletBillRecordMapper; import com.mzl.flower.mapper.wallet.WalletMapper; import com.mzl.flower.service.BaseService; +import com.mzl.flower.service.wallet.WalletService; import com.mzl.flower.utils.UUIDGenerator; import lombok.extern.slf4j.Slf4j; +import org.redisson.api.RLock; +import org.redisson.api.RedissonClient; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; @@ -40,6 +44,7 @@ import java.math.RoundingMode; import java.time.LocalDateTime; import java.util.*; +import java.util.concurrent.TimeUnit; @Slf4j @Service @@ -96,6 +101,12 @@ @Autowired private WalletBillRecordDetailMapper walletBillRecordDetailMapper; + + @Autowired + private WalletService walletService; + + @Autowired + RedissonClient redissonClient; public Page<OrderSettlementListDTO> selectSettlementList(Page page, OrderSettlementQueryDTO dto){ dto.setStartDate(parseLocalDateTime(dto.getStartDateStr(), true)); @@ -759,34 +770,36 @@ //旧逻辑转账的时候才会标记状态为结算中 //新逻辑不是立即转账。不走结算后转账逻辑。此时没有转账时间、openID、转账ID、转账明细ID,直接标记结算状态完成,存金额到钱包中 settlement.setStatus(Constants.SETTLEMENT_STATUS.COMPLETED.name()); - settlementMapper.insert(settlement); - //增加供应商结算金额保存到钱包 - walletBillRecord.setSupplierId(s.getId()); - walletBillRecord.setSettlementId(settlement.getId()); - walletBillRecord.setType(Constants.BILL_CHANGE_TYPE.settlement.name()); - walletBillRecord.setMethod(Constants.BILL_CHANGE_METHOD.add.name()); - WalletDO walletDOInfo = walletMapper.getTotalAmount(s.getUserId()); - if (ObjectUtils.isEmpty(walletDOInfo)) { - // 新增钱包 - WalletDO walletDO = new WalletDO(); - walletDO.setUserId(s.getUserId()); - walletDO.setSupplierId(s.getId()); - walletDO.setTotalAmount(BigDecimal.ZERO); - walletMapper.insert(walletDO); - walletBillRecord.setOriginalAmount(BigDecimal.ZERO); - walletBillRecord.setChangeAmount(settlementAmount); - walletBillRecord.setBalance(settlementAmount); - }else{ - walletBillRecord.setOriginalAmount(walletDOInfo.getTotalAmount()); - walletBillRecord.setChangeAmount(settlementAmount); - walletBillRecord.setBalance(walletDOInfo.getTotalAmount().add(settlementAmount)); - //更新钱包 - walletDOInfo.setTotalAmount(walletBillRecord.getBalance()); - walletMapper.updateById(walletDOInfo); + WalletDO walletDO = walletService.getCurrentSupplier(); + if (null == walletDO) { + throw new ValidationException("钱包不能为空"); } - walletBillRecordMapper.insert(walletBillRecord); + RLock lock = redissonClient.getLock(String.format(LockConstants.WALLET_ID_KEY, walletDO.getId())); + try { + if (lock.tryLock(10, 30, TimeUnit.SECONDS)) { + try { + WalletDO walletDOInfo = walletService.getCurrentSupplier(); + //增加供应商结算金额保存到钱包 + walletBillRecord.setSupplierId(s.getId()); + walletBillRecord.setSettlementId(settlement.getId()); + walletBillRecord.setType(Constants.BILL_CHANGE_TYPE.settlement.name()); + walletBillRecord.setMethod(Constants.BILL_CHANGE_METHOD.add.name()); + walletBillRecord.setOriginalAmount(walletDOInfo.getTotalAmount()); + walletBillRecord.setChangeAmount(settlementAmount); + walletBillRecord.setBalance(walletDOInfo.getTotalAmount().add(settlementAmount)); + //更新钱包 + walletDOInfo.setTotalAmount(walletBillRecord.getBalance()); + walletMapper.updateById(walletDOInfo); + walletBillRecordMapper.insert(walletBillRecord); + } finally { + lock.unlock(); + } + } + } catch (InterruptedException e) { + throw new RuntimeException(e); + } } -- Gitblit v1.9.3