yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/mysql/pd/documentaudit/MesPdDocumentAuditMapper.java
@@ -1,5 +1,6 @@
package cn.iocoder.yudao.module.mes.dal.mysql.pd.documentaudit;
import cn.hutool.core.collection.CollUtil;
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;
@@ -8,6 +9,7 @@
import org.apache.ibatis.annotations.Mapper;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
@Mapper
@@ -18,6 +20,16 @@
                .eqIfPresent(MesPdDocumentAuditDO::getDocumentId, reqVO.getDocumentId())
                .likeIfPresent(MesPdDocumentAuditDO::getAuditNo, reqVO.getAuditNo())
                .eqIfPresent(MesPdDocumentAuditDO::getAuditStatus, reqVO.getAuditStatus())
                .orderByDesc(MesPdDocumentAuditDO::getId));
    }
    default PageResult<MesPdDocumentAuditDO> selectPage(MesPdDocumentAuditPageReqVO reqVO,
                                                         Collection<Long> documentIds) {
        return selectPage(reqVO, new LambdaQueryWrapperX<MesPdDocumentAuditDO>()
                .eqIfPresent(MesPdDocumentAuditDO::getDocumentId, reqVO.getDocumentId())
                .likeIfPresent(MesPdDocumentAuditDO::getAuditNo, reqVO.getAuditNo())
                .eqIfPresent(MesPdDocumentAuditDO::getAuditStatus, reqVO.getAuditStatus())
                .inIfPresent(MesPdDocumentAuditDO::getDocumentId, documentIds)
                .orderByDesc(MesPdDocumentAuditDO::getId));
    }
@@ -44,4 +56,19 @@
                .eq(MesPdDocumentAuditDO::getAuditStatus, 0)) > 0;
    }
    /**
     * 批量查询指定资料ID列表中所有待审核的记录
     *
     * @param documentIds 资料ID列表
     * @return 待审核记录列表
     */
    default List<MesPdDocumentAuditDO> selectPendingAuditListByDocumentIds(Collection<Long> documentIds) {
        if (CollUtil.isEmpty(documentIds)) {
            return Collections.emptyList();
        }
        return selectList(new LambdaQueryWrapperX<MesPdDocumentAuditDO>()
                .in(MesPdDocumentAuditDO::getDocumentId, documentIds)
                .eq(MesPdDocumentAuditDO::getAuditStatus, 0));
    }
}