From 66f57ebef36c03353609ad3b2b6623396b5061a4 Mon Sep 17 00:00:00 2001
From: 陶杰 <1378534974@qq.com>
Date: 星期一, 28 十月 2024 15:06:54 +0800
Subject: [PATCH] 1.供应商-钱包创建并发控制

---
 src/main/java/com/mzl/flower/service/impl/wallet/WalletBillRecordServiceImpl.java |   47 ++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/mzl/flower/service/impl/wallet/WalletBillRecordServiceImpl.java b/src/main/java/com/mzl/flower/service/impl/wallet/WalletBillRecordServiceImpl.java
index b8ab819..557116d 100644
--- a/src/main/java/com/mzl/flower/service/impl/wallet/WalletBillRecordServiceImpl.java
+++ b/src/main/java/com/mzl/flower/service/impl/wallet/WalletBillRecordServiceImpl.java
@@ -1,22 +1,30 @@
 package com.mzl.flower.service.impl.wallet;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.mzl.flower.constant.Constants;
 import com.mzl.flower.dto.request.wallet.CreateWalletBillRecordDTO;
+import com.mzl.flower.dto.request.wallet.QueryWalletBillDTO;
+import com.mzl.flower.dto.response.wallet.WalletBillRecordVO;
 import com.mzl.flower.entity.payment.Transfer;
 import com.mzl.flower.entity.payment.TransferDetail;
 import com.mzl.flower.entity.wallet.WalletBillRecordDO;
 import com.mzl.flower.entity.wallet.WalletDO;
+import com.mzl.flower.entity.wallet.WalletWithdrawRecordDO;
 import com.mzl.flower.mapper.payment.TransferDetailMapper;
 import com.mzl.flower.mapper.payment.TransferMapper;
 import com.mzl.flower.mapper.wallet.WalletBillRecordMapper;
+import com.mzl.flower.mapper.wallet.WalletMapper;
+import com.mzl.flower.mapper.wallet.WalletWithdrawRecordMapper;
 import com.mzl.flower.service.wallet.WalletBillRecordService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.mzl.flower.service.wallet.WalletService;
 import org.springframework.stereotype.Service;
 import org.springframework.util.ObjectUtils;
+import org.springframework.util.StringUtils;
 
 import javax.annotation.Resource;
+import java.util.List;
 
 /**
  * <p>
@@ -41,6 +49,11 @@
     @Resource
     private WalletService walletService;
 
+    @Resource
+    private WalletMapper walletMapper;
+
+    @Resource
+    private WalletWithdrawRecordMapper walletWithdrawRecordMapper;
     @Override
     public void create(CreateWalletBillRecordDTO dto) {
 
@@ -66,19 +79,47 @@
                 walletBillRecordDO.update("sys");
                 walletBillRecordMapper.updateById(walletBillRecordDO);
                 //更新钱包
-                WalletDO walletDO = walletService.getBySupplierId(walletBillRecordDO.getWalletId());
+                WalletDO walletDO = walletService.getOrCreateBySupplierId(walletBillRecordDO.getWalletId());
                 if(!ObjectUtils.isEmpty(walletDO)){
-                    //体现中金额
+                    //提现中金额:减少提现中金额
                     walletDO.setWithdrawingAmount(walletDO.getWithdrawingAmount().subtract(walletBillRecordDO.getChangeAmount()));
-                    //已提现金额
+                    //已提现金额:增加已提现金额
                     walletDO.setWithdrawnAmount(walletDO.getWithdrawnAmount().add(walletBillRecordDO.getChangeAmount()));
+                    walletMapper.updateById(walletDO);
+                }
+                //回写提现状态
+                if (StringUtils.isEmpty(walletBillRecordDO.getWithdrawRecordId())) {
+                    WalletWithdrawRecordDO withdrawRecordDO = walletWithdrawRecordMapper.selectById(walletBillRecordDO.getWithdrawRecordId());
+                    withdrawRecordDO.setWithdrawState(Constants.WALLET_WITHDRAW_STATE.SUCCESS.name());
+                    walletWithdrawRecordMapper.updateById(withdrawRecordDO);
                 }
 
             } else if ("FAIL".equals(dStatus)) {
                 walletBillRecordDO.setTransferState(Constants.SETTLEMENT_STATUS.FAILED.name());
                 walletBillRecordDO.update("sys");
                 walletBillRecordMapper.updateById(walletBillRecordDO);
+                WalletDO walletDO = walletService.getOrCreateBySupplierId(walletBillRecordDO.getWalletId());
+                if(!ObjectUtils.isEmpty(walletDO)){
+                    //提现中金额:减少提现中金额
+                    walletDO.setWithdrawingAmount(walletDO.getWithdrawingAmount().subtract(walletBillRecordDO.getChangeAmount()));
+                    //可提现金额:增加可提现金额
+                    walletDO.setWithdrawableAmount(walletDO.getWithdrawableAmount().add(walletBillRecordDO.getChangeAmount()));
+                    walletMapper.updateById(walletDO);
+                }
+                //回写提现状态
+                if (StringUtils.isEmpty(walletBillRecordDO.getWithdrawRecordId())) {
+                    WalletWithdrawRecordDO withdrawRecordDO = walletWithdrawRecordMapper.selectById(walletBillRecordDO.getWithdrawRecordId());
+                    withdrawRecordDO.setWithdrawState(Constants.WALLET_WITHDRAW_STATE.FAILURE.name());
+                    walletWithdrawRecordMapper.updateById(withdrawRecordDO);
+                }
             }
         }
     }
+
+    @Override
+    public Page<WalletBillRecordVO> getPage(Page page, QueryWalletBillDTO dto) {
+        List<WalletBillRecordVO> list = walletBillRecordMapper.getPage(page,dto);
+        page.setRecords(list);
+        return page;
+    }
 }

--
Gitblit v1.9.3