From c25b0632e4a360f7499f1c79ff1d70a67919e008 Mon Sep 17 00:00:00 2001 From: gongzuming <gongzuming> Date: 星期二, 29 十月 2024 11:45:53 +0800 Subject: [PATCH] 重复点击问题 --- src/main/java/com/mzl/flower/service/impl/wallet/WalletBillRecordServiceImpl.java | 141 ++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 126 insertions(+), 15 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 d22cb11..902a138 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,23 +1,43 @@ 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.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.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 com.mzl.flower.utils.ExcelExportUtil; +import org.redisson.api.RLock; +import org.redisson.api.RedissonClient; 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; +import java.util.List; /** * <p> @@ -27,7 +47,9 @@ * @author @TaoJie * @since 2024-10-22 */ + @Service +@Transactional public class WalletBillRecordServiceImpl extends ServiceImpl<WalletBillRecordMapper, WalletBillRecordDO> implements WalletBillRecordService { @Resource @@ -44,6 +66,12 @@ @Resource private WalletMapper walletMapper; + + @Resource + private WalletWithdrawRecordMapper walletWithdrawRecordMapper; + @Resource + RedissonClient redissonClient; + @Override public void create(CreateWalletBillRecordDTO dto) { @@ -69,28 +97,111 @@ walletBillRecordDO.update("sys"); 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); + WalletDO walletDO = walletService.getOrCreateBySupplierId(walletBillRecordDO.getWalletId()); + 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())) { + WalletWithdrawRecordDO withdrawRecordDO = walletWithdrawRecordMapper.selectById(walletBillRecordDO.getWithdrawRecordId()); + withdrawRecordDO.setWithdrawState(Constants.WALLET_WITHDRAW_STATE.SUCCESS.name()); + walletBillRecordDO.setRemark("账户资金提现完成"); + walletWithdrawRecordMapper.updateById(withdrawRecordDO); } } else if ("FAIL".equals(dStatus)) { walletBillRecordDO.setTransferState(Constants.SETTLEMENT_STATUS.FAILED.name()); 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); + WalletDO walletDO = walletService.getOrCreateBySupplierId(walletBillRecordDO.getWalletId()); + 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())) { + WalletWithdrawRecordDO withdrawRecordDO = walletWithdrawRecordMapper.selectById(walletBillRecordDO.getWithdrawRecordId()); + withdrawRecordDO.setWithdrawState(Constants.WALLET_WITHDRAW_STATE.FAILURE.name()); + walletBillRecordDO.setRemark("账户资金提现失败"); + walletWithdrawRecordMapper.updateById(withdrawRecordDO); } } } } + + @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.getTypeName(); // 变动类型 + 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); + } + } + + @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