yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/mysql/md/item/MesMdItemMapper.java
@@ -1,5 +1,6 @@
package cn.iocoder.yudao.module.mes.dal.mysql.md.item;
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,12 @@
import org.apache.ibatis.annotations.Mapper;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
/**
 * MES 物料产品 Mapper
@@ -43,4 +50,22 @@
        return selectCount(MesMdItemDO::getUnitMeasureId, unitMeasureId);
    }
    default MesMdItemDO selectByMdmItemId(Long mdmItemId) {
        return selectOne(MesMdItemDO::getMdmItemId, mdmItemId);
    }
    /**
     * 批量查询已存在的 MDM 物料 ID 集合
     *
     * @param mdmItemIds MDM 物料 ID 列表
     * @return 存在于 mes_md_item 中的 mdmItemId 集合
     */
    default Set<Long> selectMdmItemIdsByMdmItemIds(Collection<Long> mdmItemIds) {
        if (CollUtil.isEmpty(mdmItemIds)) {
            return Collections.emptySet();
        }
        List<MesMdItemDO> list = selectList(MesMdItemDO::getMdmItemId, mdmItemIds);
        return convertSet(list, MesMdItemDO::getMdmItemId);
    }
}