From 5744562b9e25774940b126111d1c442988e75765 Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Mon, 08 Jul 2024 15:40:46 +0800
Subject: [PATCH] 逻辑
---
src/test/java/cc/mrbird/febs/MemberTest.java | 135 +++++++++++++++++++++++++++++++++++++++++---
1 files changed, 124 insertions(+), 11 deletions(-)
diff --git a/src/test/java/cc/mrbird/febs/MemberTest.java b/src/test/java/cc/mrbird/febs/MemberTest.java
index e6da7cf..e233971 100644
--- a/src/test/java/cc/mrbird/febs/MemberTest.java
+++ b/src/test/java/cc/mrbird/febs/MemberTest.java
@@ -1,6 +1,21 @@
package cc.mrbird.febs;
+import cc.mrbird.febs.dapp.entity.DappMemberEntity;
+import cc.mrbird.febs.dapp.entity.DappStorage;
+import cc.mrbird.febs.dapp.enumerate.MemberLevelEnum;
+import cc.mrbird.febs.dapp.mapper.DappMemberDao;
+import cc.mrbird.febs.dapp.mapper.DappStorageMapper;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
+
+import javax.annotation.Resource;
+import java.math.BigDecimal;
+import java.util.*;
+import java.util.stream.Collectors;
/**
* @author wzy
@@ -17,22 +32,120 @@
// @Resource
// private DappSystemService dappSystemService;
//
-// @Resource
-// private DappStorageMapper dappStorageMapper;
+ @Resource
+ private DappStorageMapper dappStorageMapper;
//
// @Resource
// private DataDictionaryCustomMapper dataDictionaryCustomMapper;
//
-// @Resource
-// private DappMemberDao dappMemberDao;
+ @Resource
+ private DappMemberDao dappMemberDao;
//
-// @Test
-// public void register() {//注册
-// ConnectDto connectDto = new ConnectDto();
-// connectDto.setAddress("0x597e5F0EDB294e7d28EB10A95aa2EEcd169a2dd7");
-// connectDto.setInviteId("53609999");
-// dappMemberService.connect(connectDto);
-// }
+ @Test
+ public void register() {//注册
+ List<DappMemberEntity> dappMemberEntities = dappMemberDao.selectList(null);
+ for(DappMemberEntity memberEntityNow : dappMemberEntities){
+ DappMemberEntity dappMemberEntity = dappMemberDao.selectById(memberEntityNow.getId());
+ if(StrUtil.isEmpty(dappMemberEntity.getRefererIds())){
+ continue;
+ }
+ //获取全部上级
+ ArrayList<DappMemberEntity> activeMembers = new ArrayList<>();
+ List<String> refererIds = Arrays.asList(StrUtil.split(dappMemberEntity.getRefererIds(),","));
+ for(String inviteId : refererIds){
+ DappMemberEntity memberEntity = dappMemberDao.selectMemberInfoByInviteId(inviteId);
+ if(ObjectUtil.isNotEmpty(memberEntity)){
+ activeMembers.add(memberEntity);
+ }
+ }
+ if(CollUtil.isEmpty(activeMembers)){
+ continue;
+ }
+
+ for(DappMemberEntity activeMember : activeMembers){
+ //获取每一个上级的所有直推，分辨大小区是否满足业绩需求
+ List<DappMemberEntity> directMembers = dappMemberDao.selectChildMemberDirectOrNot(activeMember.getInviteId(), 1, null);
+ if(CollUtil.isEmpty(directMembers)){
+ continue;
+ }
+ if(directMembers.size() < 2){
+ continue;
+ }
+ //获取activeMember的每一个直推的团队有效业绩
+ HashMap<Long, BigDecimal> daXiaoQu = new HashMap<>();
+ for(DappMemberEntity directMember : directMembers){
+ //每一个直推的团队
+ List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(directMember.getInviteId(), 2, null);
+ direct.add(directMember);
+ if(CollUtil.isEmpty(direct)){
+ continue;
+ }
+ //获取直推的团队的全部有效业绩
+ List<Long> collect = direct.stream().map(DappMemberEntity::getId).collect(Collectors.toList());
+ QueryWrapper<DappStorage> storageQueryWrapper = new QueryWrapper<>();
+ storageQueryWrapper.in("member_id",collect);
+// storageQueryWrapper.eq("state",1);
+ List<DappStorage> dappStorages = dappStorageMapper.selectList(storageQueryWrapper);
+ if(CollUtil.isEmpty(dappStorages)){
+ continue;
+ }
+
+ BigDecimal directAchieve = dappStorages.stream()
+ .map(DappStorage::getAmount) // 映射amount到流中
+ .reduce(BigDecimal.ZERO, BigDecimal::add);
+ daXiaoQu.put(directMember.getId(),directAchieve);
+ }
+ if(daXiaoQu.isEmpty()){
+ continue;
+ }
+ //输出小区的总业绩
+ BigDecimal sum = sumExcludingMax(daXiaoQu);
+ BigDecimal dao_1_levelAchieve = new BigDecimal(MemberLevelEnum.DAO_1.getLevelAchieve());
+ BigDecimal dao_2_levelAchieve = new BigDecimal(MemberLevelEnum.DAO_2.getLevelAchieve());
+ BigDecimal dao_3_levelAchieve = new BigDecimal(MemberLevelEnum.DAO_3.getLevelAchieve());
+ if(sum.compareTo(dao_3_levelAchieve) >= 0){
+ activeMember.setLevel(MemberLevelEnum.DAO_3.getCode());
+ dappMemberDao.updateById(activeMember);
+ }else if(sum.compareTo(dao_2_levelAchieve) >= 0 && sum.compareTo(dao_3_levelAchieve) < 0){
+ activeMember.setLevel(MemberLevelEnum.DAO_2.getCode());
+ dappMemberDao.updateById(activeMember);
+ }else if(sum.compareTo(dao_1_levelAchieve) >= 0 && sum.compareTo(dao_2_levelAchieve) < 0){
+ activeMember.setLevel(MemberLevelEnum.DAO_1.getCode());
+ dappMemberDao.updateById(activeMember);
+ }else{
+ activeMember.setLevel(MemberLevelEnum.DAO_0.getCode());
+ dappMemberDao.updateById(activeMember);
+ }
+ }
+ }
+ }
+
+ public static BigDecimal sumExcludingMax(HashMap<Long, BigDecimal> map) {
+ if (map == null) {
+ throw new NullPointerException("HashMap cannot be null");
+ }
+
+ // 使用Stream API找到最大的BigDecimal值及其对应的key
+ Optional<Map.Entry<Long, BigDecimal>> maxEntry = map.entrySet().stream()
+ .max((e1, e2) -> e1.getValue().compareTo(e2.getValue()));
+
+ // 检查是否找到了最大值
+ if (maxEntry.isPresent()) {
+ // 移除最大值对应的键值对
+ map.remove(maxEntry.get().getKey());
+
+ // 计算剩余数据的累计和
+ BigDecimal sumOfRemaining = map.entrySet().stream()
+ .map(Map.Entry::getValue)
+ .reduce(BigDecimal.ZERO, BigDecimal::add);
+
+ // 返回累计和
+ return sumOfRemaining;
+ } else {
+ // 如果没有找到最大值（即HashMap为空），则返回BigDecimal.ZERO
+ return BigDecimal.ZERO;
+ }
+ }
//
// @Test
// public void registerV2() {//存储
--
Gitblit v1.9.1