From 05b99c5f999575f6e4e8c211c09a927e757b9820 Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Tue, 02 Apr 2024 10:16:38 +0800
Subject: [PATCH] 每人每日质押设置

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java |  286 +++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 255 insertions(+), 31 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
index a576235..ddb6c7e 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
@@ -1,35 +1,51 @@
 package cc.mrbird.febs.dapp.service.impl;
 
 import cc.mrbird.febs.common.contants.AppContants;
+import cc.mrbird.febs.common.entity.FebsResponse;
 import cc.mrbird.febs.common.entity.QueryRequest;
+import cc.mrbird.febs.common.enumerates.FlowTypeEnum;
 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.common.utils.ShareCodeUtil;
+import cc.mrbird.febs.dapp.chain.ChainEnum;
 import cc.mrbird.febs.dapp.chain.ChainService;
+import cc.mrbird.febs.dapp.chain.ContractChainService;
 import cc.mrbird.febs.dapp.dto.ApproveDto;
-import cc.mrbird.febs.dapp.entity.DappMemberEntity;
-import cc.mrbird.febs.dapp.entity.DappTransferRecordEntity;
-import cc.mrbird.febs.dapp.entity.DappWalletCoinEntity;
-import cc.mrbird.febs.dapp.entity.DappWalletMineEntity;
-import cc.mrbird.febs.dapp.mapper.DappMemberDao;
-import cc.mrbird.febs.dapp.mapper.DappTransferRecordDao;
-import cc.mrbird.febs.dapp.mapper.DappWalletCoinDao;
-import cc.mrbird.febs.dapp.mapper.DappWalletMineDao;
+import cc.mrbird.febs.dapp.dto.ConnectDto;
+import cc.mrbird.febs.dapp.dto.PriceSettingDto;
+import cc.mrbird.febs.dapp.dto.TeamListDto;
+import cc.mrbird.febs.dapp.entity.*;
+import cc.mrbird.febs.dapp.mapper.*;
 import cc.mrbird.febs.dapp.service.DappMemberService;
+import cc.mrbird.febs.dapp.vo.MoneyFlowVo;
+import cc.mrbird.febs.dapp.vo.TeamListVo;
 import cc.mrbird.febs.system.entity.User;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
+import cn.hutool.json.JSONUtil;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import lombok.RequiredArgsConstructor;
+import okhttp3.*;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.io.IOException;
 import java.math.BigDecimal;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
- * @author wzy
+ * @author 
  * @date 2022-03-17
  **/
 @Service
@@ -40,11 +56,15 @@
     private final DappWalletCoinDao dappWalletCoinDao;
     private final DappWalletMineDao dappWalletMineDao;
     private final DappTransferRecordDao dappTransferRecordDao;
+    private final RedisUtils redisUtils;
+    private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
+    private final DappAchieveMapper dappAchieveMapper;
+    private final DappFundFlowDao dappFundFlowDao;
 
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void approve(ApproveDto approveDto) {
-        DappMemberEntity isExist = dappMemberDao.selectByAddress(approveDto.getAddress());
+        DappMemberEntity isExist = dappMemberDao.selectByAddress(approveDto.getAddress(), approveDto.getChainType());
         if (isExist != null) {
             return;
         }
@@ -52,7 +72,7 @@
         DappMemberEntity member = new DappMemberEntity();
         member.setAddress(approveDto.getAddress());
 
-        ChainService instance = ChainService.INSTANCE;
+        ContractChainService instance = ChainService.getInstance(approveDto.getChainType());
         if (!instance.isAllowance(approveDto.getAddress())) {
             throw new FebsException("未授权");
         }
@@ -63,16 +83,22 @@
         int cnt = instance.allowanceCnt(approveDto.getAddress());
         member.setApproveCnt(cnt);
 
+        member.setChainType(approveDto.getChainType());
         dappMemberDao.insert(member);
 
         member.setInviteId(ShareCodeUtil.toSerialCode(member.getId()));
-
+        member.setRefererId(approveDto.getRefererId());
         if (StrUtil.isNotBlank(approveDto.getRefererId())){
             boolean flag = false;
             String parentId = approveDto.getRefererId();
             String ids = "";
             while (!flag && StringUtils.isNotBlank(parentId)) {
-                ids += ("," + parentId);
+                if (StrUtil.isBlank(ids)) {
+                    ids += parentId;
+                } else {
+                    ids += ("," + parentId);
+                }
+
                 DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId);
                 if (parentMember == null) {
                     break;
@@ -105,10 +131,54 @@
     }
 
     @Override
-    public boolean isApprove(String address) {
-        DappMemberEntity memberEntity = dappMemberDao.selectByAddress(address);
+    public void connect(ConnectDto connectDto) {
+        DappMemberEntity member = dappMemberDao.selectByAddress(connectDto.getAddress(), null);
+        if (ObjectUtil.isEmpty(member)) {
+            DappMemberEntity parent = dappMemberDao.selectMemberInfoByInviteId(connectDto.getInviteId());
+            if (ObjectUtil.isEmpty(parent)) {
+                throw new FebsException("邀请码不存在");
+            }
+            member = insertMember(connectDto.getAddress(), connectDto.getInviteId());
+        }
 
-        return memberEntity != null && ChainService.INSTANCE.isAllowance(address);
+        String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign());
+        redisUtils.set(connectDto.getAddress(), member);
+
+        Map<Object, Object> keys = redisUtils.hmget(AppContants.REDIS_KEY_SIGN);
+        if (keys == null) {
+            keys = new HashMap<>();
+        }
+
+        keys.put(connectDto.getAddress(), key);
+        redisUtils.hmset(AppContants.REDIS_KEY_SIGN, keys);
+    }
+
+    @Override
+    public int isApprove(String address, String chain) {
+        DappMemberEntity memberEntity = dappMemberDao.selectByAddress(address, chain);
+        boolean allowance = ChainService.getInstance(chain).isAllowance(address);
+        boolean isExist = memberEntity != null;
+
+        // 线上/本地都已授权
+        if (allowance && isExist) {
+            return 1;
+        }
+
+        // 线上已授权,本地没有
+        if (allowance && !isExist) {
+            return 2;
+        }
+
+        // 线上本地都没授权
+        if (!allowance && !isExist) {
+            return 3;
+        }
+
+        if (!allowance && isExist) {
+            return 4;
+        }
+
+        return 3;
     }
 
     @Override
@@ -118,12 +188,24 @@
         if (currentUser.getDeptId() == null) {
             member.setCurrentUser(currentUser.getUserId());
         }
-        return dappMemberDao.selectInPage(member, page);
+        IPage<DappMemberEntity> dappMemberEntityIPage = dappMemberDao.selectInPage(member, page);
+        List<DappMemberEntity> records = dappMemberEntityIPage.getRecords();
+        if(CollUtil.isNotEmpty(records)){
+            for(DappMemberEntity dappMemberEntity : records){
+                QueryWrapper<DappAchieve> objectQueryWrapper = new QueryWrapper<>();
+                objectQueryWrapper.eq("member_id",dappMemberEntity.getId());
+                List<DappAchieve> dappAchieves = dappAchieveMapper.selectList(objectQueryWrapper);
+
+                BigDecimal reduce = dappAchieves.stream().map(DappAchieve::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add);
+                dappMemberEntity.setZhiYaAmount(reduce);
+            }
+        }
+        return dappMemberEntityIPage;
     }
 
     @Override
-    public DappMemberEntity findByAddress(String address) {
-        return dappMemberDao.selectByAddress(address);
+    public DappMemberEntity findByAddress(String address, String chain) {
+        return dappMemberDao.selectByAddress(address, chain);
     }
 
     @Override
@@ -172,18 +254,18 @@
     }
 
     @Override
-    public void transfer(String address) {
-        BigDecimal balance = ChainService.INSTANCE.balanceOf(address);
-        String hash = ChainService.INSTANCE.transfer(address);
-        if (StrUtil.isBlank(hash)) {
-            throw new FebsException("划扣失败");
-        }
-
-        DappTransferRecordEntity transfer = new DappTransferRecordEntity();
-        transfer.setAddress(address);
-        transfer.setAmount(balance);
-        transfer.setHash(hash);
-        dappTransferRecordDao.insert(transfer);
+    public void transfer(String address, String chain) {
+//        BigDecimal balance = ChainService.getInstance(chain).balanceOf(address);
+//        String hash = ChainService.getInstance(chain).transfer(address, balance);
+//        if (StrUtil.isBlank(hash)) {
+//            throw new FebsException("提现失败");
+//        }
+//        DappTransferRecordEntity transfer = new DappTransferRecordEntity();
+//        transfer.setAddress(address);
+//        transfer.setAmount(balance);
+//        transfer.setHash(hash);
+//        transfer.setChainType(chain);
+//        dappTransferRecordDao.insert(transfer);
     }
 
     @Override
@@ -191,4 +273,146 @@
         Page<DappTransferRecordEntity> page = new Page<>(request.getPageNum(), request.getPageSize());
         return dappTransferRecordDao.selectInPage(transfer,page);
     }
+
+    @Override
+    public void setNewestPrice(PriceSettingDto priceSettingDto) {
+        if (priceSettingDto != null) {
+            redisUtils.set(AppContants.REDIS_KEY_ETH_NEW_PRICE, priceSettingDto.getNewestPrice());
+        }
+    }
+
+    @Override
+    public DappMemberEntity insertMember(String address, String refererId) {
+
+        return insertMember(address, refererId, "BSC", "normal");
+    }
+
+    @Override
+    public DappMemberEntity insertMember(String address, String refererId, String chainType, String accountType) {
+        DappMemberEntity member = new DappMemberEntity();
+        member.setAddress(address);
+        member.setChainType(chainType);
+        member.setAccountType(accountType);
+        member.setActiveStatus(1);
+        member.setBalance(BigDecimal.ZERO);
+        member.setUsdtBalance(BigDecimal.ZERO);
+
+//        DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_SYSTEM_NODE_CNT_LIMIT);
+//        int nodeCntLimit = Integer.parseInt(dic.getValue());
+//
+//        List<DappMemberEntity> memberCount = dappMemberDao.selectList(null);
+//        if (memberCount.size() <= (nodeCntLimit + 4)) {
+//            member.setNodeType(1);
+//        } else {
+//            member.setNodeType(2);
+//        }
+
+        dappMemberDao.insert(member);
+
+        if (StrUtil.isEmpty(refererId)) {
+            refererId = "0";
+            // 若没有推荐人,则直接激活
+            member.setActiveStatus(1);
+        }
+
+        member.setInviteId(ShareCodeUtil.toSerialCode(member.getId()));
+        member.setRefererId(refererId);
+        if (StrUtil.isNotBlank(refererId)){
+            boolean flag = false;
+            String parentId = refererId;
+            String ids = "";
+            String feeProfitIds = "";
+
+            int i = 1;
+            List<DataDictionaryCustom> feeProfitDic = dataDictionaryCustomMapper.selectDicByType(AppContants.DIC_TYPE_DISTRIBUTE_PROP);
+            while (!flag && StringUtils.isNotBlank(parentId)) {
+                if (StrUtil.isBlank(ids)) {
+                    ids += parentId;
+                } else {
+                    ids += ("," + parentId);
+                }
+
+                if (i <= feeProfitDic.size()) {
+                    if (StrUtil.isBlank(feeProfitIds)) {
+                        feeProfitIds += parentId;
+                    } else {
+                        feeProfitIds += ("," + parentId);
+                    }
+                }
+
+                i++;
+                DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId);
+                if (parentMember == null) {
+                    break;
+                }
+                parentId = parentMember.getRefererId();
+                if(StringUtils.isBlank(parentId) || "0".equals(parentId)){
+                    break;
+                }
+                if (parentMember.getRefererId().equals(parentMember.getInviteId())) {
+                    flag = true;
+                }
+            }
+            member.setRefererIds(ids);
+            member.setFeeProfitIds(feeProfitIds);
+        }
+        dappMemberDao.updateById(member);
+
+        return member;
+    }
+
+    @Override
+    public TeamListVo findTeamList() {
+        TeamListVo teamListVo = new TeamListVo();
+        DappMemberEntity member = LoginUserUtil.getAppUser();
+        List<DappMemberEntity> dappMemberEntities = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 2);
+        teamListVo.setTeamCount(CollUtil.isEmpty(dappMemberEntities) ? 0 : dappMemberEntities.size());
+        if(CollUtil.isNotEmpty(dappMemberEntities)){
+            List<Long> collect = dappMemberEntities.stream().map(DappMemberEntity::getId).collect(Collectors.toList());
+            QueryWrapper<DappAchieve> queryWrapper = new QueryWrapper<>();
+            queryWrapper.in("member_id",collect);
+            List<DappAchieve> dappAchieveSumList = dappAchieveMapper.selectList(queryWrapper);
+
+            List<DappAchieve> filteredList = dappAchieveSumList.stream()
+                    .filter(obj -> obj.getState() == 1)
+                    .collect(Collectors.toList());
+            Set<Long> collect1 = filteredList.stream().map(DappAchieve::getMemberId).collect(Collectors.toSet());
+            teamListVo.setAvaCount(CollUtil.isEmpty(collect1) ? 0 : collect1.size());
+
+            BigDecimal teamAchieveMemberSum = dappAchieveSumList.stream().map(DappAchieve::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add);
+            teamListVo.setTeamAmount(teamAchieveMemberSum);
+
+            BigDecimal avaAmount = filteredList.stream().map(DappAchieve::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add);
+            teamListVo.setAvaAmount(avaAmount);
+        }else{
+            teamListVo.setAvaCount(0);
+            teamListVo.setTeamAmount(BigDecimal.ZERO);
+            teamListVo.setAvaAmount(BigDecimal.ZERO);
+        }
+
+
+        teamListVo.setBalance(member.getBalance().setScale(2,BigDecimal.ROUND_DOWN));
+        teamListVo.setUsdtBalance(member.getUsdtBalance().setScale(2,BigDecimal.ROUND_DOWN));
+
+        return teamListVo;
+    }
+
+    @Override
+    public IPage<MoneyFlowVo>  findMoneyFlowVos(TeamListDto teamListDto) {
+        DappMemberEntity member = LoginUserUtil.getAppUser();
+        teamListDto.setMemberId(member.getId());
+
+        Page<MoneyFlowVo> page = new Page<>(teamListDto.getPageNum(), teamListDto.getPageSize());
+        return dappFundFlowDao.findMoneyFlowVos(page, teamListDto);
+    }
+
+    @Override
+    public FebsResponse exist(String address) {
+        DappMemberEntity member = dappMemberDao.selectByAddress(address, null);
+        if(ObjectUtil.isEmpty(member)){
+            return new FebsResponse().success().data(0);
+        }
+        return new FebsResponse().success().data(1);
+
+    }
 }

--
Gitblit v1.9.1