From 0a32a50a7d35a7865e70623bde75f7579585f515 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Sun, 24 Apr 2022 16:30:58 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |   12 +++++++++++-
 1 files changed, 11 insertions(+), 1 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
index b541147..9f0c845 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
@@ -3,6 +3,7 @@
 import cc.mrbird.febs.common.contants.AppContants;
 import cc.mrbird.febs.common.entity.QueryRequest;
 import cc.mrbird.febs.common.exception.FebsException;
+import cc.mrbird.febs.common.utils.FebsUtil;
 import cc.mrbird.febs.common.utils.LoginUserUtil;
 import cc.mrbird.febs.common.utils.RedisUtils;
 import cc.mrbird.febs.dapp.dto.RecordInPageDto;
@@ -11,6 +12,7 @@
 import cc.mrbird.febs.dapp.mapper.*;
 import cc.mrbird.febs.dapp.service.DappWalletService;
 import cc.mrbird.febs.dapp.vo.WalletInfoVo;
+import cn.hutool.core.collection.CollUtil;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import lombok.RequiredArgsConstructor;
@@ -22,7 +24,7 @@
 import java.util.List;
 
 /**
- * @author wzy
+ * @author 
  * @date 2022-03-18
  **/
 @Slf4j
@@ -45,6 +47,7 @@
         DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
 
         WalletInfoVo walletInfo = new WalletInfoVo();
+        walletInfo.setInviteId(member.getInviteId());
         walletInfo.setTotalMine(walletMine.getTotalAmount());
         walletInfo.setAvailableMine(walletMine.getAvailableAmount());
         walletInfo.setAvailableWallet(walletCoin.getAvailableAmount());
@@ -93,6 +96,11 @@
     public void withdraw(WalletOperateDto walletOperateDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
+        List<DappFundFlowEntity> fundFlows = dappFundFlowDao.selectListForMemberAndDay(member.getId(), 2);
+        if (CollUtil.isNotEmpty(fundFlows)) {
+            throw new FebsException("一天只能提现一次");
+        }
+
         DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
         if (walletOperateDto.getAmount().compareTo(walletCoin.getAvailableAmount()) > 0) {
             throw new FebsException("可用金额不足");
@@ -117,8 +125,10 @@
     public List<DappFundFlowEntity> recordInPage(RecordInPageDto recordInPageDto) {
         Page<DappFundFlowEntity> page = new Page<>(recordInPageDto.getPageNum(), recordInPageDto.getPageSize());
 
+        DappMemberEntity member = LoginUserUtil.getAppUser();
         DappFundFlowEntity dappFundFlowEntity = new DappFundFlowEntity();
         dappFundFlowEntity.setType(recordInPageDto.getType());
+        dappFundFlowEntity.setMemberId(member.getId());
         IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity);
         return records.getRecords();
     }

--
Gitblit v1.9.1