yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/mysql/wm/stocktaking/task/MesWmStockTakingTaskMapper.java
@@ -5,12 +5,15 @@
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.mes.controller.admin.wm.stocktaking.task.vo.MesWmStockTakingTaskPageReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.stocktaking.task.MesWmStockTakingTaskDO;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.apache.ibatis.annotations.Mapper;
import java.time.LocalDateTime;
/**
 * MES 盘点任务 Mapper
 *
 * @author 芋道源码
 * @author 超级管理员
 */
@Mapper
public interface MesWmStockTakingTaskMapper extends BaseMapperX<MesWmStockTakingTaskDO> {
@@ -19,6 +22,82 @@
        return selectOne(MesWmStockTakingTaskDO::getCode, code);
    }
    /**
     * 开始盘点:置为盘点中,并清空上一轮审核结果
     *
     * 用 UpdateWrapper 而非 updateById:MyBatis-Plus 默认更新策略忽略 null,
     * 直接 updateById 无法把审核人 / 审核时间 / 审核意见真正清空。
     *
     * @param id     盘点任务编号
     * @param status 目标状态(盘点中)
     * @return 影响行数
     */
    default int submitAndResetAuditInfo(Long id, Integer status) {
        return update(null, new LambdaUpdateWrapper<MesWmStockTakingTaskDO>()
                .eq(MesWmStockTakingTaskDO::getId, id)
                .set(MesWmStockTakingTaskDO::getStatus, status)
                .set(MesWmStockTakingTaskDO::getReviewerId, null)
                .set(MesWmStockTakingTaskDO::getReviewerName, null)
                .set(MesWmStockTakingTaskDO::getReviewTime, null)
                .set(MesWmStockTakingTaskDO::getReviewRemark, null));
    }
    /**
     * 提交结果审批:写入提交时间并清空上一轮的审核人与审核意见
     *
     * 任务状态保持「盘点中」不变(该状态同时承载盘点中与结果审批中两个阶段),
     * 以 reviewTime 非空且 reviewerId 为空表示「结果审批已提交、等待审核」。
     * 不修改 status 字段,故用 UpdateWrapper 精确写入。
     *
     * @param id 盘点任务编号
     * @return 影响行数
     */
    default int markResultApprovalSubmitted(Long id) {
        return update(null, new LambdaUpdateWrapper<MesWmStockTakingTaskDO>()
                .eq(MesWmStockTakingTaskDO::getId, id)
                .set(MesWmStockTakingTaskDO::getReviewTime, LocalDateTime.now())
                .set(MesWmStockTakingTaskDO::getReviewerId, null)
                .set(MesWmStockTakingTaskDO::getReviewerName, null)
                .set(MesWmStockTakingTaskDO::getReviewRemark, null));
    }
    /**
     * 审核结果:写入审核结果、审核人与审核意见
     *
     * @param id           盘点任务编号
     * @param status       目标状态(已完成 / 盘点中)
     * @param reviewerId   审核人编号
     * @param reviewerName 审核人姓名
     * @param reviewRemark 审核意见
     * @return 影响行数
     */
    default int auditStockTakingTask(Long id, Integer status, Long reviewerId,
                                     String reviewerName, String reviewRemark) {
        return update(null, new LambdaUpdateWrapper<MesWmStockTakingTaskDO>()
                .eq(MesWmStockTakingTaskDO::getId, id)
                .set(MesWmStockTakingTaskDO::getStatus, status)
                .set(MesWmStockTakingTaskDO::getReviewerId, reviewerId)
                .set(MesWmStockTakingTaskDO::getReviewerName, reviewerName)
                .set(MesWmStockTakingTaskDO::getReviewTime, LocalDateTime.now())
                .set(MesWmStockTakingTaskDO::getReviewRemark, reviewRemark));
    }
    /**
     * 清空审核标记
     *
     * 用于任务取消后清除结果审批痕迹
     *
     * @param id 盘点任务编号
     */
    default void clearAuditInfo(Long id) {
        update(null, new LambdaUpdateWrapper<MesWmStockTakingTaskDO>()
                .eq(MesWmStockTakingTaskDO::getId, id)
                .set(MesWmStockTakingTaskDO::getReviewerId, null)
                .set(MesWmStockTakingTaskDO::getReviewerName, null)
                .set(MesWmStockTakingTaskDO::getReviewTime, null)
                .set(MesWmStockTakingTaskDO::getReviewRemark, null));
    }
    default PageResult<MesWmStockTakingTaskDO> selectPage(MesWmStockTakingTaskPageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<MesWmStockTakingTaskDO>()
                .likeIfPresent(MesWmStockTakingTaskDO::getCode, reqVO.getCode())