From f77c25900aac7231d006d8f4e08aaa12d72c62b4 Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期一, 28 十月 2024 16:21:39 +0800
Subject: [PATCH] add: 钱包供应商财务

---
 src/main/java/com/mzl/flower/service/impl/wallet/WalletBillRecordServiceImpl.java |  107 +++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 94 insertions(+), 13 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 7d1aa37..fab1396 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,8 +1,13 @@
 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.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.mzl.flower.constant.Constants;
+import com.mzl.flower.constant.LockConstants;
 import com.mzl.flower.dto.request.wallet.CreateWalletBillRecordDTO;
+import com.mzl.flower.dto.request.wallet.QueryWalletBillRecordDTO;
+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;
@@ -14,13 +19,22 @@
 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 com.mzl.flower.utils.ExcelExportUtil;
+import org.redisson.api.RLock;
+import org.redisson.api.RedissonClient;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.ObjectUtils;
 import org.springframework.util.StringUtils;
 
 import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import java.net.URLEncoder;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.TimeUnit;
 
 /**
  * <p>
@@ -30,7 +44,9 @@
  * @author @TaoJie
  * @since 2024-10-22
  */
+
 @Service
+@Transactional
 public class WalletBillRecordServiceImpl extends ServiceImpl<WalletBillRecordMapper, WalletBillRecordDO> implements WalletBillRecordService {
 
     @Resource
@@ -50,6 +66,9 @@
 
     @Resource
     private WalletWithdrawRecordMapper walletWithdrawRecordMapper;
+    @Resource
+    RedissonClient redissonClient;
+
     @Override
     public void create(CreateWalletBillRecordDTO dto) {
 
@@ -76,12 +95,23 @@
                 walletBillRecordMapper.updateById(walletBillRecordDO);
                 //更新钱包
                 WalletDO walletDO = walletService.getBySupplierId(walletBillRecordDO.getWalletId());
-                if(!ObjectUtils.isEmpty(walletDO)){
-                    //提现中金额:减少提现中金额
-                    walletDO.setWithdrawingAmount(walletDO.getWithdrawingAmount().subtract(walletBillRecordDO.getChangeAmount()));
-                    //已提现金额:增加已提现金额
-                    walletDO.setWithdrawnAmount(walletDO.getWithdrawnAmount().add(walletBillRecordDO.getChangeAmount()));
-                    walletMapper.updateById(walletDO);
+                RLock lock = redissonClient.getLock(String.format(LockConstants.WALLET_ID_KEY, walletDO.getId()));
+                try {
+                    if (lock.tryLock(10, 30, TimeUnit.SECONDS)) {
+                        try {
+                            if (!ObjectUtils.isEmpty(walletDO)) {
+                                //提现中金额:减少提现中金额
+                                walletDO.setWithdrawingAmount(walletDO.getWithdrawingAmount().subtract(walletBillRecordDO.getChangeAmount()));
+                                //已提现金额:增加已提现金额
+                                walletDO.setWithdrawnAmount(walletDO.getWithdrawnAmount().add(walletBillRecordDO.getChangeAmount()));
+                                walletMapper.updateById(walletDO);
+                            }
+                        } finally {
+                            lock.unlock();
+                        }
+                    }
+                } catch (InterruptedException e) {
+                    throw new RuntimeException(e);
                 }
                 //回写提现状态
                 if (StringUtils.isEmpty(walletBillRecordDO.getWithdrawRecordId())) {
@@ -95,12 +125,23 @@
                 walletBillRecordDO.update("sys");
                 walletBillRecordMapper.updateById(walletBillRecordDO);
                 WalletDO walletDO = walletService.getBySupplierId(walletBillRecordDO.getWalletId());
-                if(!ObjectUtils.isEmpty(walletDO)){
-                    //提现中金额:减少提现中金额
-                    walletDO.setWithdrawingAmount(walletDO.getWithdrawingAmount().subtract(walletBillRecordDO.getChangeAmount()));
-                    //可提现金额:增加可提现金额
-                    walletDO.setWithdrawableAmount(walletDO.getWithdrawableAmount().add(walletBillRecordDO.getChangeAmount()));
-                    walletMapper.updateById(walletDO);
+                RLock lock = redissonClient.getLock(String.format(LockConstants.WALLET_ID_KEY, walletDO.getId()));
+                try {
+                    if (lock.tryLock(10, 30, TimeUnit.SECONDS)) {
+                        try {
+                            if (!ObjectUtils.isEmpty(walletDO)) {
+                                //提现中金额:减少提现中金额
+                                walletDO.setWithdrawingAmount(walletDO.getWithdrawingAmount().subtract(walletBillRecordDO.getChangeAmount()));
+                                //可提现金额:增加可提现金额
+                                walletDO.setWithdrawableAmount(walletDO.getWithdrawableAmount().add(walletBillRecordDO.getChangeAmount()));
+                                walletMapper.updateById(walletDO);
+                            }
+                        } finally {
+                            lock.unlock();
+                        }
+                    }
+                } catch (InterruptedException e) {
+                    throw new RuntimeException(e);
                 }
                 //回写提现状态
                 if (StringUtils.isEmpty(walletBillRecordDO.getWithdrawRecordId())) {
@@ -111,4 +152,44 @@
             }
         }
     }
+
+    @Override
+    public Page<WalletBillRecordVO> queryPage(QueryWalletBillRecordDTO queryWalletBillRecordDTO, Page page) {
+        List<WalletBillRecordVO> list = walletBillRecordMapper.queryPage(queryWalletBillRecordDTO, page);
+        page.setRecords(list);
+        return page;
+    }
+
+    @Override
+    public void exportSupplierFinanceList(HttpServletResponse response, QueryWalletBillRecordDTO queryWalletBillRecordDTO) {
+
+        List<WalletBillRecordVO> list = walletBillRecordMapper.queryWalletBillRecordList(queryWalletBillRecordDTO);
+        String[] rowsName = new String[]{"序号","供应商信息", "变动类型", "原金额", "变动金额", "现余额", "提交时间", "审核时间", "备注"};
+        List<Object[]> dataList = new ArrayList<>();
+        int sn = 1;
+        for (WalletBillRecordVO o : list) {
+            Object[] objs = new Object[rowsName.length];
+            int a = 0;
+            objs[a++] = sn; // 序号
+            objs[a++] = o.getSupplierName()+":"+o.getSupplierId(); // 供应商信息
+            objs[a++] = o.getTypeStr(); // 变动类型
+            objs[a++] = o.getOriginalAmount(); // 原金额
+            objs[a++] = o.getChangeAmount(); // 变动金额
+            objs[a++] = o.getBalance(); //现余额
+            objs[a++] = o.getCreateTime();//提交时间
+            objs[a++] = o.getApproveTime(); // 审核时间
+            objs[a++] = o.getRemark(); //备注
+            dataList.add(objs);
+            sn++;
+        }
+
+        ExcelExportUtil excelExportUtil = new ExcelExportUtil("供应商财务", rowsName, dataList, response);
+        try {
+            response.addHeader("filename", URLEncoder.encode("供应商财务.xls", "UTF-8"));
+            response.addHeader("Access-Control-Expose-Headers", "filename");
+            excelExportUtil.export();
+        } catch (Exception e) {
+            log.error(e.getMessage(), e);
+        }
+    }
 }

--
Gitblit v1.9.3