逻辑

KKSU

2024-07-08 fbcc5ef9fec4a1b1581ea41b03d83f07da8514de

 src/test/java/cc/mrbird/febs/MemberTest.java

@@ -1,6 +1,24 @@
package cc.mrbird.febs;

import cc.mrbird.febs.dapp.chain.ChainEnum;
import cc.mrbird.febs.dapp.chain.ChainService;
import cc.mrbird.febs.dapp.dto.BatchTransferDto;
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 +35,132 @@
//    @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 withdraw() {//注册

        ArrayList<BatchTransferDto> objects = new ArrayList<>();
        BatchTransferDto batchTransferDto = new BatchTransferDto("0x677696d67c2b9e87aecf60bc17b116f5b25e2927",new BigDecimal("38.8"));
        objects.add(batchTransferDto);

        BatchTransferDto batchTransferDtoFee = new BatchTransferDto("0x3caf594503573341ae7a83fc6daed9bd5276b6bd",new BigDecimal("1.2"));
        objects.add(batchTransferDtoFee);
        ChainService.getInstance(ChainEnum.BSC_USDT.name()).transferList(objects);
    }
//
//    @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() {//存储