liu
5 天以前 4bc513be11e671caf20ec6013a51dee42c2ed96e
yudao-module-crm/src/main/java/cn/iocoder/yudao/module/crm/dal/mysql/contract/CrmContractMapper.java
@@ -1,5 +1,6 @@
package cn.iocoder.yudao.module.crm.dal.mysql.contract;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.LocalDateTimeUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
@@ -12,6 +13,7 @@
import cn.iocoder.yudao.module.crm.enums.common.CrmBizTypeEnum;
import cn.iocoder.yudao.module.crm.enums.common.CrmSceneTypeEnum;
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;
@@ -26,8 +28,81 @@
@Mapper
public interface CrmContractMapper extends BaseMapperX<CrmContractDO> {
    default CrmContractDO selectByNoForUpdate(String no) {
        return selectOne(new LambdaQueryWrapperX<CrmContractDO>().eq(CrmContractDO::getNo, no).last("FOR UPDATE"));
    }
    default CrmContractDO selectByIdForUpdate(Long id) {
        return selectOne(new LambdaQueryWrapperX<CrmContractDO>().eq(CrmContractDO::getId, id).last("FOR UPDATE"));
    }
    default CrmContractDO selectByNo(String no) {
        return selectOne(CrmContractDO::getNo, no);
    }
    default CrmContractDO selectByNameAndCustomerId(String name, Long customerId) {
        return selectOne(new LambdaQueryWrapperX<CrmContractDO>()
                .eq(CrmContractDO::getName, name)
                .eq(CrmContractDO::getCustomerId, customerId));
    }
    /**
     * 提交审批:置为审批中,并清空上一轮审核结果
     *
     * 之所以用 UpdateWrapper 而非 updateById:MyBatis-Plus 默认的字段更新策略会忽略 null,
     * 直接 updateById 无法把审核不通过后残留的审核人、审核时间、审核意见真正清空。
     *
     * @param id     合同编号
     * @param status 目标状态(审批中)
     * @return 影响行数
     */
    default int submitAndResetAuditInfo(Long id, Integer status) {
        return update(null, new LambdaUpdateWrapper<CrmContractDO>()
                .eq(CrmContractDO::getId, id)
                .in(CrmContractDO::getAuditStatus, CrmAuditStatusEnum.DRAFT.getStatus(),
                        CrmAuditStatusEnum.REJECT.getStatus())
                .set(CrmContractDO::getAuditStatus, status)
                .set(CrmContractDO::getReviewerId, null)
                .set(CrmContractDO::getReviewerName, null)
                .set(CrmContractDO::getReviewTime, null)
                .set(CrmContractDO::getReviewRemark, null));
    }
    /**
     * 审核:写入审核结果、审核人与审核意见
     *
     * reviewRemark 允许为 null(审核通过时意见可选),故同样使用 UpdateWrapper 直写。
     *
     * @param id           合同编号
     * @param status       目标状态(审核通过 / 审核不通过)
     * @param reviewerId   审核人编号
     * @param reviewerName 审核人姓名
     * @param reviewRemark 审核意见
     * @return 影响行数
     */
    default int auditContract(Long id, Integer status, Long reviewerId,
                              String reviewerName, String reviewRemark) {
        return update(null, new LambdaUpdateWrapper<CrmContractDO>()
                .eq(CrmContractDO::getId, id)
                .eq(CrmContractDO::getAuditStatus, CrmAuditStatusEnum.PROCESS.getStatus())
                .set(CrmContractDO::getAuditStatus, status)
                .set(CrmContractDO::getReviewerId, reviewerId)
                .set(CrmContractDO::getReviewerName, reviewerName)
                .set(CrmContractDO::getReviewTime, LocalDateTime.now())
                .set(CrmContractDO::getReviewRemark, reviewRemark));
    }
    /**
     * BPM 审批结束后回写合同状态
     *
     * 仅当合同仍处于「审批中」时更新,避免与旧审批路径重复覆盖。
     */
    default int updateAuditStatusByBpm(Long id, Integer status, String reviewRemark) {
        return update(null, new LambdaUpdateWrapper<CrmContractDO>()
                .eq(CrmContractDO::getId, id)
                .eq(CrmContractDO::getAuditStatus, CrmAuditStatusEnum.PROCESS.getStatus())
                .set(CrmContractDO::getAuditStatus, status)
                .set(CrmContractDO::getReviewTime, LocalDateTime.now())
                .set(CrmContractDO::getReviewRemark, reviewRemark));
    }
    default PageResult<CrmContractDO> selectPageByCustomerId(CrmContractPageReqVO pageReqVO) {
@@ -51,6 +126,18 @@
    }
    default PageResult<CrmContractDO> selectPage(CrmContractPageReqVO pageReqVO, Long userId, CrmContractConfigDO config) {
        return selectPage(pageReqVO, userId, config, null);
    }
    /**
     * 分页查询(带客户权限过滤)
     *
     * @param pageReqVO 查询条件
     * @param userId 用户ID
     * @param config 合同配置
     * @param permittedCustomerIds 有权限的客户ID列表(null表示不限制)
     */
    default PageResult<CrmContractDO> selectPage(CrmContractPageReqVO pageReqVO, Long userId, CrmContractConfigDO config, List<Long> permittedCustomerIds) {
        MPJLambdaWrapperX<CrmContractDO> query = new MPJLambdaWrapperX<>();
        // 拼接数据权限的查询条件
        CrmPermissionUtils.appendPermissionCondition(query, CrmBizTypeEnum.CRM_CONTRACT.getType(),
@@ -64,12 +151,22 @@
                .eqIfPresent(CrmContractDO::getAuditStatus, pageReqVO.getAuditStatus())
                .orderByDesc(CrmContractDO::getId);
        // 客户权限过滤
        if (permittedCustomerIds != null) {
            if (CollUtil.isEmpty(permittedCustomerIds)) {
                return PageResult.empty();
            }
            query.in(CrmContractDO::getCustomerId, permittedCustomerIds);
        }
        // Backlog: 即将到期的合同
        LocalDateTime beginOfToday = LocalDateTimeUtil.beginOfDay(LocalDateTime.now());
        LocalDateTime endOfToday = LocalDateTimeUtil.endOfDay(LocalDateTime.now());
        if (CrmContractPageReqVO.EXPIRY_TYPE_ABOUT_TO_EXPIRE.equals(pageReqVO.getExpiryType())) { // 即将到期
            query.eq(CrmContractDO::getAuditStatus, CrmAuditStatusEnum.APPROVE.getStatus())
                    .between(CrmContractDO::getEndTime, beginOfToday, endOfToday.plusDays(config.getNotifyDays()));
                    // notifyDays 可能为 null(历史配置未填),为 null 时按 0 天处理,避免拆箱 NPE
                .between(CrmContractDO::getEndTime, beginOfToday,
                        endOfToday.plusDays(config.getNotifyDays() != null ? config.getNotifyDays() : 0));
        } else if (CrmContractPageReqVO.EXPIRY_TYPE_EXPIRED.equals(pageReqVO.getExpiryType())) { // 已到期
            query.eq(CrmContractDO::getAuditStatus, CrmAuditStatusEnum.APPROVE.getStatus())
                    .lt(CrmContractDO::getEndTime, endOfToday);
@@ -104,7 +201,9 @@
        LocalDateTime beginOfToday = LocalDateTimeUtil.beginOfDay(LocalDateTime.now());
        LocalDateTime endOfToday = LocalDateTimeUtil.endOfDay(LocalDateTime.now());
        query.eq(CrmContractDO::getAuditStatus, CrmAuditStatusEnum.APPROVE.getStatus()) // 必须审批通过!
                .between(CrmContractDO::getEndTime, beginOfToday, endOfToday.plusDays(config.getNotifyDays()));
                // notifyDays 可能为 null(历史配置未填),为 null 时按 0 天处理,避免拆箱 NPE
                .between(CrmContractDO::getEndTime, beginOfToday,
                        endOfToday.plusDays(config.getNotifyDays() != null ? config.getNotifyDays() : 0));
        return selectCount(query);
    }
@@ -118,6 +217,10 @@
        return selectOne(CrmContractDO::getBusinessId, businessId);
    }
    default CrmContractDO selectByOrderId(Long orderId) {
        return selectOne(CrmContractDO::getOrderId, orderId);
    }
    default List<CrmContractDO> selectListByCustomerId(Long customerId) {
        return selectList(CrmContractDO::getCustomerId, customerId);
    }