KKSU
2024-12-30 bd4d6ee6e9e303e6fdbf05fede63dc3cd50263c3
src/main/java/cc/mrbird/febs/mall/service/impl/MemberProfitServiceImpl.java
@@ -2,6 +2,7 @@
import cc.mrbird.febs.common.enumerates.*;
import cc.mrbird.febs.common.utils.AppContants;
import cc.mrbird.febs.common.utils.MallUtils;
import cc.mrbird.febs.mall.entity.*;
import cc.mrbird.febs.mall.mapper.*;
import cc.mrbird.febs.mall.service.AsyncService;
@@ -51,6 +52,9 @@
    private final MallMemberCouponMapper mallMemberCouponMapper;
    private final RunVipMapper runVipMapper;
    private final MallMemberWalletMapper mallMemberWalletMapper;
    private final RunNodeSetMapper runNodeSetMapper;
    private final MallChargeMapper mallChargeMapper;
    private final IMallMoneyFlowService mallMoneyFlowService;
    private final AsyncService asyncService;
@@ -646,6 +650,141 @@
        }
    }
    @Override
    public void updateMemberCoin() {
        try {
            // 操作时间
            DateTime operationDate = DateUtil.date();
            // 获取最低级别的会员等级
            RunVip minRunVip = runVipMapper.selectOne(new LambdaQueryWrapper<RunVip>()
                    .orderByAsc(RunVip::getOrderNumber)
                    .last("LIMIT 1")
            );
            if (minRunVip == null) {
                log.error("最低级别的会员等级未找到");
                return;
            }
            // 获取全部referrerId不为空的用户,并获取每个用户的所有上级id,返回一个set集合
            List<MallMember> mallMembers = mallMemberMapper.selectList(
                new LambdaQueryWrapper<MallMember>()
                    .isNotNull(MallMember::getReferrerId)
            );
            if (CollUtil.isEmpty(mallMembers)) {
                return;
            }
            Set<Long> memberIds = mallMembers.stream()
                .map(MallMember::getId)
                .collect(Collectors.toSet());
            if (CollUtil.isEmpty(memberIds)) {
                return;
            }
            // 构建成员ID到MallMember对象的映射
            Map<Long, MallMember> memberMap = mallMemberMapper.selectBatchIds(memberIds).stream()
                    .collect(Collectors.toMap(MallMember::getId, member -> member));
            for (Long memberId : memberIds) {
                MallMember mallMember = memberMap.get(memberId);
                if (mallMember == null || minRunVip.getVipCode().equals(mallMember.getLevel())) {
                    continue;
                }
                // 获取所有购买了会员等级的直推成员
                List<MallMember> directMembers = mallMemberMapper.selectList(
                    new LambdaQueryWrapper<MallMember>()
                        .eq(MallMember::getReferrerId, mallMember.getInviteId())
                        .ne(MallMember::getLevel, minRunVip.getVipCode())
                );
                if (CollUtil.isEmpty(directMembers)) {
                    continue;
                }
                for (MallMember item : directMembers) {
                    BigDecimal realScore = calculateDirectScore(mallMember, item, operationDate);
                    if(BigDecimal.ZERO.compareTo(realScore) >= 0){
                        continue;
                    }
                    walletService.addScore(realScore, memberId);
                    String orderNo = MallUtils.getOrderNum("ZLS");
                    mallMoneyFlowService.runVipMoneyFlowAdd(
                            memberId,
                            item.getId(),
                            orderNo,
                            FlowTypeEnum.SCORE.getValue(),
                            RunVipMoneyFlowTypeEnum.DIRECT_SCORE.getValue(),
                            realScore,
                            StrUtil.format(RunVipMoneyFlowTypeEnum.DIRECT_SCORE.getDescription(), realScore),
                            YesOrNoEnum.YES.getValue()
                    );
                }
            }
        } catch (Exception e) {
            log.error("更新会员积分时发生异常", e);
            throw new RuntimeException("更新会员积分时发生异常", e); // 根据业务需求选择是否抛出异常
        }
    }
    private BigDecimal calculateDirectScore(MallMember mallMember, MallMember directMember, Date operationDate) {
        // 实际助力
        BigDecimal realScore = BigDecimal.ZERO;
        try {
            String directLevel = directMember.getLevel();
            RunVip directRunVip = runVipMapper.selectOne(new LambdaQueryWrapper<RunVip>().eq(RunVip::getVipCode, directLevel));
            if (directRunVip == null) {
                return realScore;
            }
            // 获取每一个会员的前一日碳积分总和
            LambdaQueryWrapper<MallMoneyFlow> mallMoneyFlowLambdaQueryWrapper = new LambdaQueryWrapper<>();
            mallMoneyFlowLambdaQueryWrapper.ge(MallMoneyFlow::getCreatedTime, DateUtil.offsetDay(operationDate, -1));
            mallMoneyFlowLambdaQueryWrapper.eq(MallMoneyFlow::getFlowType, FlowTypeEnum.SCORE.getValue());
            mallMoneyFlowLambdaQueryWrapper.eq(MallMoneyFlow::getType, RunVipMoneyFlowTypeEnum.GET_SCORE.getValue());
            List<MallMoneyFlow> mallMoneyFlows = mallMoneyFlowMapper.selectList(mallMoneyFlowLambdaQueryWrapper);
            if (CollUtil.isEmpty(mallMoneyFlows)) {
                return realScore;
            }
            String memberLevel = mallMember.getLevel();
            RunVip memberRunVip = runVipMapper.selectOne(new LambdaQueryWrapper<RunVip>().eq(RunVip::getVipCode, memberLevel));
            if (memberRunVip == null) {
                return realScore;
            }
            // 如果直推小于会员本身的会员等级,则全部助力
            if (memberRunVip.getOrderNumber() >= directRunVip.getOrderNumber()) {
                BigDecimal rebatePercent = directRunVip.getRebatePercent();
                BigDecimal totalScore = mallMoneyFlows.stream()
                        .map(MallMoneyFlow::getAmount)
                        .reduce(BigDecimal.ZERO, BigDecimal::add);
                realScore = totalScore.multiply(rebatePercent).setScale(0, RoundingMode.DOWN);
            } else {
                Integer growthCnt = memberRunVip.getGrowthCnt();
                if (growthCnt == null || growthCnt <= 0) {
                    return realScore;
                }
                BigDecimal rebatePercent1 = memberRunVip.getRebatePercent();
                if (rebatePercent1 == null || rebatePercent1.compareTo(BigDecimal.ZERO) <= 0) {
                    return realScore;
                }
                realScore = new BigDecimal(growthCnt).multiply(rebatePercent1).setScale(0, RoundingMode.DOWN);
            }
        } catch (Exception e) {
            // 记录日志并返回默认值
            log.error("Error calculating direct score", e);
            return BigDecimal.ZERO;
        }
        return realScore;
    }
    public boolean isDivisibleByTwo(int number) {
        return number % 2 == 0;
@@ -688,4 +827,109 @@
            }
        }
    }
    @Override
    public void updateNodeScore() {
        /**
         * 获取全部的节点
         */
        List<MallMember> mallMembers = mallMemberMapper.selectList(
                new LambdaQueryWrapper<MallMember>()
                        .eq(MallMember::getDirector, YesOrNoEnum.YES.getValue())
        );
        if(CollUtil.isEmpty(mallMembers)){
            return;
        }
        //获取节点设置
        List<RunNodeSet> runNodeSets = runNodeSetMapper.selectList(null);
        BigDecimal balanceToCoin = new BigDecimal(
                dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                        RunVipDataDictionaryEnum.RUN_VIP_BALANCE_TO_COIN.getType(),
                        RunVipDataDictionaryEnum.RUN_VIP_BALANCE_TO_COIN.getCode()).getValue()
        ).setScale(2, BigDecimal.ROUND_DOWN);
        DateTime endTime = DateUtil.date();
        for (MallMember item : mallMembers) {
            //获取总业绩
            BigDecimal achieve = getDirectAchieve(item.getInviteId(),item.getDirectorTime(),endTime);
            if(BigDecimal.ZERO.compareTo(achieve) >= 0){
                continue;
            }
            BigDecimal nodePercent = getNodePercent(runNodeSets, achieve);
            if(BigDecimal.ZERO.compareTo(nodePercent) >= 0){
                continue;
            }
            asyncService.releaseNodeCoin(achieve,nodePercent,balanceToCoin,item.getId());
            item.setDirectorTime(endTime);
            mallMemberMapper.updateById(item);
        }
    }
    private BigDecimal getDirectAchieve(String inviteId,Date startTime,Date endTime) {
        try {
            // 获取直推成员和团队成员的ID集合
            Set<Long> memberIds = getTeamMemberIds(inviteId);
            if (CollUtil.isEmpty(memberIds)) {
                return BigDecimal.ZERO;
            }
            // 获取团队业绩(不包含本人业绩)
            List<MallCharge> mallCharges = mallChargeMapper.selectList(
                    new LambdaQueryWrapper<MallCharge>()
                            .in(MallCharge::getMemberId, memberIds)
                            .eq(MallCharge::getState, YesOrNoEnum.YES.getValue())
                            .ge(MallCharge::getCreatedTime, startTime)
                            .lt(MallCharge::getCreatedTime, endTime)
            );
            if (CollUtil.isEmpty(mallCharges)) {
                return BigDecimal.ZERO;
            }
            return mallCharges.stream()
                    .map(MallCharge::getAmount)
                    .reduce(BigDecimal.ZERO, BigDecimal::add);
        } catch (Exception e) {
            // 异常处理
            log.error("Error occurred while calculating direct achieve for member: {}", inviteId, e);
            return BigDecimal.ZERO;
        }
    }
    private Set<Long> getTeamMemberIds(String inviteId) {
        Set<Long> memberIds = new HashSet<>();
        // 获取直推成员
        List<MallMember> directMembers = mallMemberMapper.selectList(
                new LambdaQueryWrapper<MallMember>()
                        .eq(MallMember::getReferrerId, inviteId)
        );
        memberIds.addAll(directMembers.stream().map(MallMember::getId).collect(Collectors.toSet()));
        // 获取团队成员
        List<MallMember> teamMembers = mallMemberMapper.selectList(
                new LambdaQueryWrapper<MallMember>()
                        .in(MallMember::getReferrerId, directMembers.stream().map(MallMember::getInviteId).collect(Collectors.toSet()))
        );
        memberIds.addAll(teamMembers.stream().map(MallMember::getId).collect(Collectors.toSet()));
        return memberIds;
    }
    private BigDecimal getNodePercent(List<RunNodeSet> runNodeSets, BigDecimal directAchieve) {
        BigDecimal percent = BigDecimal.ZERO;
        //获取directAchieve在runNodeSets中哪一个位置
        for(RunNodeSet item : runNodeSets){
            BigDecimal minAchieve = item.getMinAchieve();
            BigDecimal maxAchieve = item.getMaxAchieve();
            BigDecimal nodePercent = new BigDecimal(item.getNodePercent());
            if(directAchieve.compareTo(minAchieve) >= 0 && directAchieve.compareTo(maxAchieve) < 0){
                percent = nodePercent;
                break;
            }
        }
        return percent;
    }
}