liyong
5 天以前 be5783c8a7e13bbc76d33c95643d75779cce21db
yudao-module-crm/src/main/java/cn/iocoder/yudao/module/crm/dal/mysql/business/CrmBusinessMapper.java
@@ -1,5 +1,6 @@
package cn.iocoder.yudao.module.crm.dal.mysql.business;
import cn.hutool.core.date.LocalDateTimeUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
@@ -7,11 +8,14 @@
import cn.iocoder.yudao.module.crm.controller.admin.business.vo.business.CrmBusinessPageReqVO;
import cn.iocoder.yudao.module.crm.controller.admin.statistics.vo.funnel.CrmStatisticsFunnelReqVO;
import cn.iocoder.yudao.module.crm.dal.dataobject.business.CrmBusinessDO;
import cn.iocoder.yudao.module.crm.enums.business.CrmBusinessEndStatusEnum;
import cn.iocoder.yudao.module.crm.enums.common.CrmBizTypeEnum;
import cn.iocoder.yudao.module.crm.util.CrmPermissionUtils;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.apache.ibatis.annotations.Mapper;
import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.Collection;
import java.util.List;
@@ -65,10 +69,80 @@
                .eq(CrmBusinessDO::getOwnerUserId, ownerUserId));
    }
    default List<CrmBusinessDO> selectListByCustomerId(Long customerId) {
        return selectList(CrmBusinessDO::getCustomerId, customerId);
    }
    default PageResult<CrmBusinessDO> selectPage(CrmStatisticsFunnelReqVO pageVO) {
        return selectPage(pageVO, new LambdaQueryWrapperX<CrmBusinessDO>()
                .in(CrmBusinessDO::getOwnerUserId, pageVO.getUserIds())
                .betweenIfPresent(CrmBusinessDO::getCreateTime, pageVO.getTimes()));
    }
    // ========== 统计方法 ==========
    default Long selectTodayCount() {
        LocalDateTime beginOfToday = LocalDateTimeUtil.beginOfDay(LocalDateTime.now());
        LocalDateTime endOfToday = LocalDateTimeUtil.endOfDay(LocalDateTime.now());
        return selectCount(new LambdaQueryWrapperX<CrmBusinessDO>()
                .between(CrmBusinessDO::getCreateTime, beginOfToday, endOfToday));
    }
    default BigDecimal selectTotalPriceSum() {
        LambdaQueryWrapperX<CrmBusinessDO> query = new LambdaQueryWrapperX<>();
        query.select(CrmBusinessDO::getTotalPrice);
        List<CrmBusinessDO> list = selectList(query);
        return list.stream()
                .map(CrmBusinessDO::getTotalPrice)
                .filter(p -> p != null)
                .reduce(BigDecimal.ZERO, BigDecimal::add);
    }
    default Long selectSuccessCount() {
        return selectCount(new LambdaQueryWrapperX<CrmBusinessDO>()
                .eq(CrmBusinessDO::getEndStatus, CrmBusinessEndStatusEnum.WIN.getStatus()));
    }
    default BigDecimal selectSuccessPriceSum() {
        LambdaQueryWrapperX<CrmBusinessDO> query = new LambdaQueryWrapperX<>();
        query.eq(CrmBusinessDO::getEndStatus, CrmBusinessEndStatusEnum.WIN.getStatus())
                .select(CrmBusinessDO::getTotalPrice);
        List<CrmBusinessDO> list = selectList(query);
        return list.stream()
                .map(CrmBusinessDO::getTotalPrice)
                .filter(p -> p != null)
                .reduce(BigDecimal.ZERO, BigDecimal::add);
    }
    default Long selectProcessingCount() {
        return selectCount(new LambdaQueryWrapperX<CrmBusinessDO>()
                .isNull(CrmBusinessDO::getEndStatus));
    }
    default Long selectFailCount() {
        return selectCount(new LambdaQueryWrapperX<CrmBusinessDO>()
                .eq(CrmBusinessDO::getEndStatus, CrmBusinessEndStatusEnum.LOSE.getStatus()));
    }
    default List<Long> selectOwnerUserIds() {
        LambdaQueryWrapperX<CrmBusinessDO> query = new LambdaQueryWrapperX<>();
        query.select(CrmBusinessDO::getOwnerUserId).isNotNull(CrmBusinessDO::getOwnerUserId);
        List<CrmBusinessDO> list = selectList(query);
        return list.stream().map(CrmBusinessDO::getOwnerUserId).distinct().toList();
    }
    default Long selectCountByOwnerUserId(Long ownerUserId) {
        return selectCount(CrmBusinessDO::getOwnerUserId, ownerUserId);
    }
    default BigDecimal selectPriceSumByOwnerUserId(Long ownerUserId) {
        LambdaQueryWrapperX<CrmBusinessDO> query = new LambdaQueryWrapperX<>();
        query.eq(CrmBusinessDO::getOwnerUserId, ownerUserId).select(CrmBusinessDO::getTotalPrice);
        List<CrmBusinessDO> list = selectList(query);
        return list.stream()
                .map(CrmBusinessDO::getTotalPrice)
                .filter(p -> p != null)
                .reduce(BigDecimal.ZERO, BigDecimal::add);
    }
}