refactor(mall): 优化 VIP 信息获取逻辑

KKSU

2024-12-20 4c0f0fe5fd8ca17e90114d9473a7fb0bec494578

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

@@ -1,31 +1,27 @@
package cc.mrbird.febs.mall.service.impl;

import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.enumerates.FlowTypeEnum;
import cc.mrbird.febs.common.enumerates.RunVipDataDictionaryEnum;
import cc.mrbird.febs.common.enumerates.RunVipMoneyFlowTypeEnum;
import cc.mrbird.febs.common.enumerates.YesOrNoEnum;
import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.LoginUserUtil;
import cc.mrbird.febs.common.utils.MallUtils;
import cc.mrbird.febs.mall.conversion.RunVipConversion;
import cc.mrbird.febs.mall.dto.ApiChargeListDto;
import cc.mrbird.febs.mall.dto.ApiGoChargeDto;
import cc.mrbird.febs.mall.dto.ApiGoChargeInfoDto;
import cc.mrbird.febs.mall.dto.ApiMemberChargeFailDto;
import cc.mrbird.febs.mall.dto.*;
import cc.mrbird.febs.mall.entity.*;
import cc.mrbird.febs.mall.mapper.DataDictionaryCustomMapper;
import cc.mrbird.febs.mall.mapper.MallChargeMapper;
import cc.mrbird.febs.mall.mapper.MallMemberPaymentMapper;
import cc.mrbird.febs.mall.mapper.RunVipMapper;
import cc.mrbird.febs.mall.mapper.*;
import cc.mrbird.febs.mall.service.IRunVipService;
import cc.mrbird.febs.mall.vo.ApiChargeInfoVo;
import cc.mrbird.febs.mall.vo.ApiChargeVo;
import cc.mrbird.febs.mall.vo.ApiGoChargeVo;
import cc.mrbird.febs.mall.vo.ApiRunVipVo;
import cc.mrbird.febs.mall.vo.*;
import cc.mrbird.febs.rabbit.producter.AgentProducer;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUnit;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -36,7 +32,9 @@
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

@Slf4j
@Service
@@ -47,20 +45,24 @@
    private final MallMemberPaymentMapper mallMemberPaymentMapper;
    private final MallChargeMapper mallChargeMapper;
    private final AgentProducer agentProducer;
    private final MallMoneyFlowMapper mallMoneyFlowMapper;
    private final MallMemberWalletMapper mallMemberWalletMapper;
    private final MallMemberMapper mallMemberMapper;
    @Override
    public List<ApiRunVipVo> vipInfo() {
        MallMember member = LoginUserUtil.getLoginUser();
        Long memberId = LoginUserUtil.getLoginUser().getId();
        List<ApiRunVipVo> apiRunVipVos = new ArrayList<>();
        if (StrUtil.isEmpty(member.getLevel())) {
        MallMember mallMember = mallMemberMapper.selectById(memberId);
        if (StrUtil.isEmpty(mallMember.getLevel())) {
            return apiRunVipVos;
        }
        String level = member.getLevel();
        String level = mallMember.getLevel();
        //获取当前用户的VIP等级
        RunVip runVip = this.baseMapper.selectOne(new LambdaQueryWrapper<RunVip>().eq(RunVip::getVipCode, level));
        //获取大于当前用户等级序号的所有的VIP等级
        LambdaQueryWrapper<RunVip> runVipLambdaQueryWrapper = new LambdaQueryWrapper<>();
        runVipLambdaQueryWrapper.eq(RunVip::getState, YesOrNoEnum.YES.getValue());
        runVipLambdaQueryWrapper.gt(RunVip::getOrderNumber, runVip.getOrderNumber());
        runVipLambdaQueryWrapper.ge(RunVip::getOrderNumber, runVip.getOrderNumber());
        List<RunVip> runVips = this.baseMapper.selectList(runVipLambdaQueryWrapper);
        apiRunVipVos = RunVipConversion.INSTANCE.entityToVos(runVips);

@@ -71,7 +73,14 @@
    public FebsResponse goCharge(ApiGoChargeDto apiGoChargeDto) {

        Long memberId = LoginUserUtil.getLoginUser().getId();
        MallMember mallMember = mallMemberMapper.selectById(memberId);
        if (StrUtil.isBlank(mallMember.getTradePassword())) {
            throw new FebsException("未设置资金密码");
        }

        if (!mallMember.getTradePassword().equals(SecureUtil.md5(apiGoChargeDto.getTradeWord()))) {
            throw new FebsException("资金密码错误");
        }
        Long runVipId = apiGoChargeDto.getRunVipId();
        RunVip runVip = this.baseMapper.selectById(runVipId);
        if(runVip.getState() != YesOrNoEnum.YES.getValue()){
@@ -86,10 +95,23 @@
            return new FebsResponse().fail().message("请先绑定你的地址");
        }

        String sysAddress = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS.getType(),
                RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS.getCode()
        //判断系统的充值地址
        String trcType = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                RunVipDataDictionaryEnum.CHARGE_TYPE_TRC.getType(),
                RunVipDataDictionaryEnum.CHARGE_TYPE_TRC.getCode()
        ).getValue();
        String sysAddress = "";
        if(trcType.equals(mallMemberPayment.getBankNo())){
            sysAddress = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                    RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS_TRC.getType(),
                    RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS_TRC.getCode()
            ).getValue();
        }else{
            sysAddress = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                    RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS_BSC.getType(),
                    RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS_BSC.getCode()
            ).getValue();
        }
        String failMinutes = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                RunVipDataDictionaryEnum.CHARGE_SYS_FAIL_TIME.getType(),
                RunVipDataDictionaryEnum.CHARGE_SYS_FAIL_TIME.getCode()
@@ -115,6 +137,7 @@
        apiGoChargeVo.setAddress(mallCharge.getAddress());
        apiGoChargeVo.setAmount(mallCharge.getAmount());
        apiGoChargeVo.setSysAddress(mallCharge.getSysAddress());
        apiGoChargeVo.setSysAddressType(mallCharge.getType());

        /**
         * 充值接口调用后，发送一个延时队列
@@ -143,23 +166,8 @@
        apiGoChargeVo.setAddress(mallCharge.getAddress());
        apiGoChargeVo.setAmount(mallCharge.getAmount());
        apiGoChargeVo.setSysAddress(mallCharge.getSysAddress());
        apiGoChargeVo.setSysAddressType(mallCharge.getType());
        return new FebsResponse().success().data(apiGoChargeVo);
    }

    @Override
    public FebsResponse getChargeInfo() {
        Long memberId = LoginUserUtil.getLoginUser().getId();
        ApiChargeInfoVo apiChargeInfoVo = new ApiChargeInfoVo();

        String sysAddress = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS.getType(),
                RunVipDataDictionaryEnum.CHARGE_SYS_ADDRESS.getCode()
        ).getValue();
        apiChargeInfoVo.setChargeAddress(sysAddress);
        List<DataDictionaryCustom> dataDictionaryCustoms = dataDictionaryCustomMapper.selectDicByType(RunVipDataDictionaryEnum.CHARGE_TYPE_TRC.getType());
        List<String> coinTypeList = dataDictionaryCustoms.stream().map(DataDictionaryCustom::getValue).collect(Collectors.toList());
        apiChargeInfoVo.setCoinType(coinTypeList);
        return new FebsResponse().success().data(apiChargeInfoVo);
    }

    @Override
@@ -169,4 +177,148 @@
        Page<ApiChargeVo> page = new Page<>(apiChargeListDto.getPageNum(), apiChargeListDto.getPageSize());
        return mallChargeMapper.selectChargeListInPage(apiChargeListDto, page);
    }

    @Override
    public List<ApiGetRunDateVo> getRunDate(ApiGetRunDateDto queryDto) {

        Long memberId = LoginUserUtil.getLoginUser().getId();
        DateTime dateTime = DateUtil.parseDate(queryDto.getDateStr());

        DateTime startTime = DateUtil.beginOfMonth(dateTime);
        DateTime endTime = DateUtil.endOfMonth(dateTime);
        long between = DateUtil.between(startTime, endTime, DateUnit.DAY);
        List<ApiGetRunDateVo> apiGetRunDateVos = generateDateList(startTime, between);

        List<MallMoneyFlow> mallMoneyFlows = mallMoneyFlowMapper.selectList(
                new LambdaQueryWrapper<MallMoneyFlow>()
                        .eq(MallMoneyFlow::getMemberId, memberId)
                        .eq(MallMoneyFlow::getFlowType, FlowTypeEnum.SCORE.getValue())
                        .eq(MallMoneyFlow::getType, RunVipMoneyFlowTypeEnum.SCORE_OUT_BALANCE.getValue())
                        .ge(MallMoneyFlow::getCreatedTime, startTime)
                        .le(MallMoneyFlow::getCreatedTime, endTime)
        );
        if(CollUtil.isNotEmpty(mallMoneyFlows)){
            calculateProfits(apiGetRunDateVos,mallMoneyFlows);
        }

        return apiGetRunDateVos;
    }

    /**
     * 生成日期列表
     *
     * @param startDate 开始日期
     * @param days      从开始日期算起的天数
     * @return 一个包含日期信息的ApiGetRunDateVo对象列表
     */
    private List<ApiGetRunDateVo> generateDateList(DateTime startDate, long days) {
        // 使用IntStream生成从0到days（包含days）的整数流，表示每一天
        return IntStream.range(0, (int) days + 1)
                // 将每个整数映射为一个ApiGetRunDateVo对象
                .mapToObj(i -> {
                    // 创建ApiGetRunDateVo对象
                    ApiGetRunDateVo apiGetRunDateVo = new ApiGetRunDateVo();
                    // 计算日期：从起始日期开始偏移i天
                    DateTime dateTime = DateUtil.offsetDay(startDate, i);
                    // 格式化日期为"yyyy-MM-dd"字符串，并设置到对象中
                    apiGetRunDateVo.setDateStr(DateUtil.format(dateTime, "yyyy-MM-dd"));
                    // 初始化碳积分为0
                    apiGetRunDateVo.setScore(BigDecimal.ZERO);
                    // 返回构建好的对象
                    return apiGetRunDateVo;
                })
                // 将流中的对象收集到一个列表中
                .collect(Collectors.toList());
    }
    /**
     * 计算每日累计碳积分
     *
     * @param dateList 日期列表，包含了每日的碳积分信息
     * @param mallMoneyFlows 商城资金流列表，包含了每笔碳积分信息
     */
    private void calculateProfits(List<ApiGetRunDateVo> dateList, List<MallMoneyFlow> mallMoneyFlows) {
        // 将日期列表转换为映射，以便通过日期快速访问每日碳积分信息
        Map<String, ApiGetRunDateVo> dateMap = dateList.stream()
                .collect(Collectors.toMap(ApiGetRunDateVo::getDateStr, vo -> vo));

        // 遍历mallMoneyFlows
        for (MallMoneyFlow item : mallMoneyFlows) {
            // 格式化当前项的创建时间，提取日期
            String date = DateUtil.format(item.getCreatedTime(), "yyyy-MM-dd");
            // 从日期映射中获取对应的每日碳积分信息
            ApiGetRunDateVo apiGetRunDateVo = dateMap.get(date);
            // 如果找到了对应的每日碳积分信息
            if (apiGetRunDateVo != null) {
                // 获取当前日期的碳积分
                BigDecimal score = apiGetRunDateVo.getScore();
                // 计算新的碳积分，将碳积分与碳积分相加，并向下取整到个位数
                BigDecimal profit = score.add(item.getAmount().abs()).setScale(0,BigDecimal.ROUND_DOWN);
                // 更新每日碳积分信息的利润值
                apiGetRunDateVo.setScore(profit);
            }
        }
    }

    @Override
    public ApiRunDataVo runData() {
        Long memberId = LoginUserUtil.getLoginUser().getId();
        ApiRunDataVo apiRunDataVo = new ApiRunDataVo();
        MallMember mallMember = mallMemberMapper.selectById(memberId);
        String level = mallMember.getLevel();
        RunVip runVip = this.baseMapper.selectOne(new LambdaQueryWrapper<RunVip>().eq(RunVip::getVipCode, level));
        apiRunDataVo.setAimScore(runVip.getGrowthCnt());

        List<MallMoneyFlow> mallMoneyFlows = mallMoneyFlowMapper.selectList(
                new LambdaQueryWrapper<MallMoneyFlow>()
                        .eq(MallMoneyFlow::getMemberId, memberId)
                        .eq(MallMoneyFlow::getFlowType, FlowTypeEnum.SCORE.getValue())
                        .in(MallMoneyFlow::getType, RunVipMoneyFlowTypeEnum.GET_SCORE.getValue(),RunVipMoneyFlowTypeEnum.SYS_SCORE.getValue())
                        .ge(MallMoneyFlow::getCreatedTime, DateUtil.beginOfDay(DateUtil.date()))
                        .le(MallMoneyFlow::getCreatedTime, DateUtil.endOfDay(DateUtil.date()))
        );
        if(CollUtil.isNotEmpty(mallMoneyFlows)){
            BigDecimal reduce = mallMoneyFlows.stream().map(MallMoneyFlow::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add);
            apiRunDataVo.setScore(reduce.intValue());
        }

        MallMemberWallet mallMemberWallet = mallMemberWalletMapper.selectWalletByMemberId(memberId);
        apiRunDataVo.setBalance(mallMemberWallet.getBalance());
        apiRunDataVo.setRealScore(mallMemberWallet.getScore().intValue());
        return apiRunDataVo;
    }

    @Override
    public ApiRunHealthVo runHealth() {
        Long memberId = LoginUserUtil.getLoginUser().getId();
        ApiRunHealthVo apiRunHealthVo = new ApiRunHealthVo();

        MallMemberWallet mallMemberWallet = mallMemberWalletMapper.selectWalletByMemberId(memberId);
        int realScore = mallMemberWallet.getScore().intValue();
        apiRunHealthVo.setScore(realScore);
        List<MallMoneyFlow> mallMoneyFlows = mallMoneyFlowMapper.selectList(
                new LambdaQueryWrapper<MallMoneyFlow>()
                        .eq(MallMoneyFlow::getMemberId, memberId)
                        .eq(MallMoneyFlow::getFlowType, FlowTypeEnum.SCORE.getValue())
                        .in(MallMoneyFlow::getType, RunVipMoneyFlowTypeEnum.GET_SCORE.getValue(),RunVipMoneyFlowTypeEnum.SYS_SCORE.getValue())
                        .ge(MallMoneyFlow::getCreatedTime, DateUtil.beginOfDay(DateUtil.date()))
                        .le(MallMoneyFlow::getCreatedTime, DateUtil.endOfDay(DateUtil.date()))
        );
        if(CollUtil.isNotEmpty(mallMoneyFlows)){
            BigDecimal reduce = mallMoneyFlows.stream().map(MallMoneyFlow::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add);
            apiRunHealthVo.setScoreTotal(reduce.intValue());
        }
        List<MallMoneyFlow> mallMoneyFlowScoreOuts = mallMoneyFlowMapper.selectList(
                new LambdaQueryWrapper<MallMoneyFlow>()
                        .eq(MallMoneyFlow::getMemberId, memberId)
                        .eq(MallMoneyFlow::getFlowType, FlowTypeEnum.SCORE.getValue())
                        .eq(MallMoneyFlow::getType, RunVipMoneyFlowTypeEnum.SCORE_OUT_BALANCE.getValue())
                        .ge(MallMoneyFlow::getCreatedTime, DateUtil.beginOfDay(DateUtil.date()))
                        .le(MallMoneyFlow::getCreatedTime, DateUtil.endOfDay(DateUtil.date()))
        );
        if(CollUtil.isNotEmpty(mallMoneyFlowScoreOuts)){
            BigDecimal reduce = mallMoneyFlowScoreOuts.stream().map(MallMoneyFlow::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add).abs();
            apiRunHealthVo.setScoreUsed(reduce.intValue());
        }
        return apiRunHealthVo;
    }
}