KKSU
2024-05-31 8484fc52b9434a933c55545dbe90e1a79d4ab919
src/test/java/cc/mrbird/febs/ProfitTest.java
@@ -5,10 +5,7 @@
import cc.mrbird.febs.common.enumerates.ProductEnum;
import cc.mrbird.febs.common.utils.MallUtils;
import cc.mrbird.febs.mall.dto.TeamPerkDto;
import cc.mrbird.febs.mall.entity.DataDictionaryCustom;
import cc.mrbird.febs.mall.entity.MallMember;
import cc.mrbird.febs.mall.entity.MallOrderItem;
import cc.mrbird.febs.mall.entity.MallProductBuy;
import cc.mrbird.febs.mall.entity.*;
import cc.mrbird.febs.mall.mapper.*;
import cc.mrbird.febs.mall.quartz.OrderSettlementJob;
import cc.mrbird.febs.mall.service.*;
@@ -18,6 +15,7 @@
import cc.mrbird.febs.pay.service.UnipayService;
import cc.mrbird.febs.rabbit.producter.AgentProducer;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
@@ -51,75 +49,80 @@
    private DataDictionaryCustomMapper dataDictionaryCustomMapper;
    @Autowired
    private MallProductBuyMapper mallProductBuyMapper;
    @Test
    public void dynamicProfit(){
        QueryWrapper<MallProductBuy> query = new QueryWrapper<>();
        query.eq("state", ProductEnum.PRODUCT_BUY_SUCCESS.getValue());
        query.eq("mate_state",ProductEnum.PRODUCT_BUY_MATE_STATE_FAIL.getValue());
        List<MallProductBuy> mallProductBuys = mallProductBuyMapper.selectList(query);
        if(CollUtil.isEmpty(mallProductBuys)){
            return;
        }
        Set<Long> collect = mallProductBuys.stream().map(MallProductBuy::getMemberId).collect(Collectors.toSet());
        if(CollUtil.isEmpty(collect)){
            return;
        }
        for(Long memberId : collect){
            MallMember mallMember = memberMapper.selectById(memberId);
            if(ObjectUtil.isEmpty(mallMember)){
                continue;
            }
            //上级
            String referrerId = mallMember.getReferrerId();
            MallMember mallMemberRef = memberMapper.selectInfoByInviteId(referrerId);
            if(ObjectUtil.isEmpty(mallMemberRef)){
                continue;
            }
            List<MallMember> mallMembersAll = memberMapper.selectAllChildAgentListByInviteId(referrerId);
            List<MallMember> mallMembers = memberMapper.selectByRefererId(referrerId);
            Set<Long> mallMembersSet = mallMembers.stream().map(MallMember::getId).collect(Collectors.toSet());
            mallMembersSet.retainAll(collect);
            List<Long> intersection = new ArrayList<>(mallMembersSet);//上级直推有效人数
            if(CollUtil.isEmpty(intersection)){
                mallMemberRef.setLevel(MemberLevelNewEnum.ZERO_LEVEL.getType());
                memberMapper.updateById(mallMemberRef);
                continue;
            }
            /**
             * 从会员等级最低到最高,比较intersection与会员设置的直推人数directCntLevel比较
             * directCntLevel >= intersection.size(),则满足升级会员等级条件,跳出循环
             * teamCnt >= mallMembersAll.size(),则满足升级会员等级条件,跳出循环
             * 达到最高级别后,升级会员等级,跳出循环
             */
            String level = MemberLevelNewEnum.ZERO_LEVEL.getType();
            boolean flag = false;
            while (!flag){
                DataDictionaryCustom levelDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                        level, level
                );
                TeamPerkDto teamPerkDto = JSONObject.parseObject(levelDic.getValue(), TeamPerkDto.class);
                Integer directCntLevel = teamPerkDto.getDirectCnt();
                Integer teamCnt = teamPerkDto.getTeamCnt();
                if(directCntLevel >= intersection.size() &&
                        teamCnt >= mallMembersAll.size()){
                    mallMemberRef.setLevel(level);
                    memberMapper.updateById(mallMemberRef);
                    flag = true;
                }
                String nextLevel = MemberLevelNewEnum.ZERO_LEVEL.getNextLevel(level);
                level = nextLevel;
                if(level.equals(MemberLevelNewEnum.WU_LEVEL.getType()) &&
                        nextLevel.equals(MemberLevelNewEnum.WU_LEVEL.getType()) ){
                    mallMemberRef.setLevel(level);
                    memberMapper.updateById(mallMemberRef);
                    flag = true;
                }
            }
        }
    }
//    @Test
//    public void dynamicProfit(){
//
//        QueryWrapper<MallProductBuy> query = new QueryWrapper<>();
//        query.eq("state", ProductEnum.PRODUCT_BUY_SUCCESS.getValue());
//        query.eq("mate_state",ProductEnum.PRODUCT_BUY_MATE_STATE_FAIL.getValue());
//        List<MallProductBuy> mallProductBuys = mallProductBuyMapper.selectList(query);
//        if(CollUtil.isEmpty(mallProductBuys)){
//            return;
//        }
//
//        Set<Long> collect = mallProductBuys.stream().map(MallProductBuy::getMemberId).collect(Collectors.toSet());
//        if(CollUtil.isEmpty(collect)){
//            return;
//        }
//        for(Long memberId : collect){
//            MallMember mallMember = memberMapper.selectById(memberId);
//            if(ObjectUtil.isEmpty(mallMember)){
//                continue;
//            }
//            //上级
//            String referrerId = mallMember.getReferrerId();
//            MallMember mallMemberRef = memberMapper.selectInfoByInviteId(referrerId);
//            if(ObjectUtil.isEmpty(mallMemberRef)){
//                continue;
//            }
//            List<MallMember> mallMembersAll = memberMapper.selectAllChildAgentListByInviteId(referrerId);
//
//            List<MallMember> mallMembers = memberMapper.selectByRefererId(referrerId);
//            Set<Long> mallMembersSet = mallMembers.stream().map(MallMember::getId).collect(Collectors.toSet());
//            mallMembersSet.retainAll(collect);
//            List<Long> intersection = new ArrayList<>(mallMembersSet);//上级直推有效人数
//            if(CollUtil.isEmpty(intersection)){
//                mallMemberRef.setLevel(MemberLevelNewEnum.ZERO_LEVEL.getType());
//                memberMapper.updateById(mallMemberRef);
//                continue;
//            }
//            /**
//             * 从会员等级最低到最高,比较intersection与会员设置的直推人数directCntLevel比较
//             * directCntLevel >= intersection.size(),则满足升级会员等级条件,跳出循环
//             * teamCnt >= mallMembersAll.size(),则满足升级会员等级条件,跳出循环
//             * 达到最高级别后,升级会员等级,跳出循环
//             */
//            String level = MemberLevelNewEnum.ZERO_LEVEL.getType();
//            boolean flag = false;
//            while (!flag){
//                DataDictionaryCustom levelDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
//                        level, level
//                );
//                TeamPerkDto teamPerkDto = JSONObject.parseObject(levelDic.getValue(), TeamPerkDto.class);
//                Integer directCntLevel = teamPerkDto.getDirectCnt();
//                Integer teamCnt = teamPerkDto.getTeamCnt();
//                if(directCntLevel <= intersection.size()){
//                    if(teamCnt <= mallMembersAll.size()){
//                        mallMemberRef.setLevel(level);
//                        memberMapper.updateById(mallMemberRef);
//
//                        String nextLevel = MemberLevelNewEnum.ZERO_LEVEL.getNextLevel(level);
//                        level = nextLevel;
//                        if(level.equals(MemberLevelNewEnum.WU_LEVEL.getType()) &&
//                                nextLevel.equals(MemberLevelNewEnum.WU_LEVEL.getType()) ){
//                            mallMemberRef.setLevel(level);
//                            memberMapper.updateById(mallMemberRef);
//                            flag = true;
//                        }
//                    }else{
//                        flag = true;
//                    }
//                }else{
//                    flag = true;
//                }
//            }
//        }
//    }
//    @Autowired
//    private AgentConsumer agentConsumer;
//
@@ -175,25 +178,26 @@
    private NBYHService nbyhService;
    @Autowired
    private IAdminMallMemberService mallMemberService;
    @Autowired
    private MallProductNftMapper mallProductNftMapper;
    @Test
    public void agentProfitTrade() {
//        agentService.timeGetOrderBuy();
//        agentService.BuyToSell();
//        try {
//            System.out.println(resourceLoader.getResource("classpath:lkl/api_private_key.pem").getInputStream());
//        } catch (IOException e) {
//            e.printStackTrace();
        boolean flag = true;
        Integer i = 0;
//        while (flag){
////            agentService.BuyToSellV2();
//            DateTime dateTime = DateUtil.offsetDay(new DateTime(), i);
//            agentService.BuyToSellV3(dateTime);
//            i = i+1;
//            List<MallProductBuy> mallProductBuys = mallProductBuyMapper.selectListByStateAndMateState(
//                    ProductEnum.PRODUCT_BUY_SUCCESS.getValue(),
//                    ProductEnum.PRODUCT_BUY_MATE_STATE_FAIL.getValue());
//            if(CollUtil.isEmpty(mallProductBuys)){
//                flag = false;
//            }
//        }
//        nbyhService.tradeSinge(306L);
//        nbyhService.tradeSingeOuterTransfer(306L);
//        nbyhService.accountInfoQuery(306L);
//        nbyhService.transferResultInfoQuery(315L);
//
//        mallMemberService.chargeAgree(307L);
//        mallMemberService.searchInfo(307L);
//        agentService.perkAgent();
        System.out.println(i);
    }
    @Test
    public void agentProfit() {