From 6f9dc440fdf9933dc087f8e0041b09aacaadc8c2 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Fri, 10 Jun 2022 09:55:46 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |   24 ++++++++++++++----------
 1 files changed, 14 insertions(+), 10 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 03eb81a..4ec3e35 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
@@ -48,18 +48,20 @@
 
     @Override
     public WalletInfoVo walletInfo() {
-//        DappMemberEntity member = LoginUserUtil.getAppUser();
-//        DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
-//        DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
+        DappMemberEntity member = LoginUserUtil.getAppUser();
 
+        Map<String, BigDecimal> map = dappFundFlowDao.selectAmountTotalByType(member.getId());
         WalletInfoVo walletInfo = new WalletInfoVo();
-//        walletInfo.setInviteId(member.getInviteId());
-//        walletInfo.setTotalMine(walletMine.getTotalAmount());
-//        walletInfo.setAvailableMine(walletMine.getAvailableAmount());
-//        walletInfo.setAvailableWallet(walletCoin.getAvailableAmount());
+        List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1);
+        List<DappMemberEntity> notDirect = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 2);
+        BigDecimal childHoldAmount = dappMemberDao.selectChildHoldAmount(member.getInviteId());
 
-        walletInfo.setTotalChild(10000);
-        walletInfo.setTotalChild(41451);
+        walletInfo.setTotalChild(notDirect.size());
+        walletInfo.setDirectCnt(direct.size());
+        walletInfo.setTotalChildCoin(childHoldAmount);
+        walletInfo.setTeamReward(map.get("teamReward"));
+        walletInfo.setMiningAmount(map.get("mine"));
+        walletInfo.setInviteId(member.getInviteId());
         return walletInfo;
     }
 
@@ -136,7 +138,9 @@
 
         DappMemberEntity member = LoginUserUtil.getAppUser();
         DappFundFlowEntity dappFundFlowEntity = new DappFundFlowEntity();
-        dappFundFlowEntity.setType(recordInPageDto.getType());
+        if (recordInPageDto.getType() != null && recordInPageDto.getType() != 0) {
+            dappFundFlowEntity.setType(recordInPageDto.getType());
+        }
         dappFundFlowEntity.setMemberId(member.getId());
         IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity);
         return records.getRecords();

--
Gitblit v1.9.1