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 |  100 +++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 95 insertions(+), 5 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 13ffa86..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,19 +1,30 @@
 package com.mzl.flower.service.impl.wallet;
 
-import com.mzl.flower.config.exception.ValidationException;
-import com.mzl.flower.config.security.SecurityUtils;
+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.response.supplier.SupplierDTO;
+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.service.supplier.SupplierService;
+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.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.ObjectUtils;
+import org.springframework.util.StringUtils;
+
+import javax.annotation.Resource;
+import java.util.List;
 
 /**
  * <p>
@@ -26,10 +37,89 @@
 @Service
 public class WalletBillRecordServiceImpl extends ServiceImpl<WalletBillRecordMapper, WalletBillRecordDO> implements WalletBillRecordService {
 
+    @Resource
+    private TransferMapper transferMapper;
 
+    @Resource
+    private TransferDetailMapper transferDetailMapper;
+
+    @Resource
+    private WalletBillRecordMapper walletBillRecordMapper;
+
+    @Resource
+    private WalletService walletService;
+
+    @Resource
+    private WalletMapper walletMapper;
+
+    @Resource
+    private WalletWithdrawRecordMapper walletWithdrawRecordMapper;
     @Override
     public void create(CreateWalletBillRecordDTO dto) {
 
 
     }
+
+    public void updateTransferStatus(String transferId) {
+        Transfer t = transferMapper.selectById(transferId);
+        if ("FINISHED".equals(t.getStatus())) {
+            WalletBillRecordDO walletBillRecordDO = walletBillRecordMapper.selectOne(new QueryWrapper<WalletBillRecordDO>().eq("transfer_id", transferId));
+            if (walletBillRecordDO == null) {
+                return;
+            }
+            String transferDetailId = walletBillRecordDO.getTransferDetailId();
+            TransferDetail td = transferDetailMapper.selectById(transferDetailId);
+            if (td == null) {
+                log.warn("未找到对应明细");
+                return;
+            }
+            String dStatus = td.getStatus();
+            if ("SUCCESS".equals(dStatus)) {
+                walletBillRecordDO.setTransferState(Constants.SETTLEMENT_STATUS.COMPLETED.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.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