feat(mall): 添加会员列表直推、团队、业绩统计功能 - 在 MallMember 实体中添加直推数、团队数、业绩金额等字段 - 修改 get...

KKSU

2024-12-31 c5947256b22f2283214258827f61af09966cf490

 src/main/java/cc/mrbird/febs/mall/service/impl/AdminMallMemberServiceImpl.java

@@ -1,5 +1,6 @@
package cc.mrbird.febs.mall.service.impl;

import cc.mrbird.febs.common.configure.FebsConfigure;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.enumerates.*;
@@ -17,12 +18,14 @@
import cc.mrbird.febs.pay.model.MemberWithdrawalDto;
import cc.mrbird.febs.pay.service.IXcxPayService;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@@ -34,8 +37,9 @@

import javax.validation.constraints.NotNull;
import java.math.BigDecimal;
import java.util.Date;
import java.util.List;
import java.util.*;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Collectors;

/**
 * @author wzy
@@ -75,11 +79,61 @@
    private final SalemanCouponMapper salemanCouponMapper;
    private final MallMemberCouponMapper mallMemberCouponMapper;
    private final MallGoodsCouponMapper mallGoodsCouponMapper;
    private final FebsConfigure febsConfigure;
    private final MallChargeMapper mallChargeMapper;

    @Override
    public IPage<MallMember> getMallMemberList(MallMember mallMember, QueryRequest request) {
        Page<MallMember> page = new Page<>(request.getPageNum(), request.getPageSize());
        IPage<MallMember> mallMembers = this.baseMapper.selectMallMemberListInPage(page, mallMember);
        List<MallMember> records = mallMembers.getRecords();
        if(CollUtil.isNotEmpty(records)){
            List<CompletableFuture<Void>> futures = new ArrayList<>();
            DateTime endTime = DateUtil.date();
            records.forEach(item -> {
                CompletableFuture<Void> uCompletableFuture = CompletableFuture.runAsync(() -> {
                    //获取直推
                    Set<String> directInviteIds = mallMemberMapper.selectList(
                            new LambdaQueryWrapper<MallMember>()
                                    .eq(MallMember::getReferrerId, item.getInviteId())
                    ).stream()
                            .map(MallMember::getInviteId)
                            .collect(Collectors.toSet());
                    if(CollUtil.isNotEmpty(directInviteIds)){
                        item.setDirectCnt(directInviteIds.size());
                        //获取团队
                        List<MallMember> allMembers = mallMemberMapper.selectList(
                                new LambdaQueryWrapper<MallMember>()
                                        .eq(MallMember::getReferrerId, item.getInviteId())
                                        .or()
                                        .in(MallMember::getReferrerId, directInviteIds)
                        );


                        if(CollUtil.isNotEmpty(allMembers)){
                            item.setTeamCnt(allMembers.size());
                            // 获取团队业绩(不包含本人业绩)
                            List<MallCharge> mallCharges = mallChargeMapper.selectList(
                                    new LambdaQueryWrapper<MallCharge>()
                                            .in(MallCharge::getMemberId, allMembers.stream().map(MallMember::getId).collect(Collectors.toSet()))
                                            .eq(MallCharge::getState, YesOrNoEnum.YES.getValue())
                                            .ge(MallCharge::getCreatedTime, item.getDirectorTime())
                                            .lt(MallCharge::getCreatedTime, endTime)
                            );
                            if (CollUtil.isNotEmpty(mallCharges)) {
                                item.setAchieveCnt(mallCharges.stream()
                                        .map(MallCharge::getAmount)
                                        .reduce(BigDecimal.ZERO, BigDecimal::add));
                            }
                        }
                    }
                }, febsConfigure.asyncThreadPoolTaskExecutor());
                futures.add(uCompletableFuture);
            });
            // 等待所有任务完成
            CompletableFuture<Void> allOf = CompletableFuture.allOf(futures.toArray(new CompletableFuture[0]));
            allOf.join(); // 阻塞直到所有任务完成
        }
        return mallMembers;
    }