liu
2026-10-19 152cd504b3c6dff0cf6f98ea2111c455264d92de
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/mps/MesProMpsServiceImpl.java
@@ -3,8 +3,11 @@
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.mes.controller.admin.pro.mps.vo.MesProMpsImportExcelVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.mps.vo.MesProMpsImportRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.mps.vo.MesProMpsPageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.mps.vo.MesProMpsSaveReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.pro.workorder.vo.MesProWorkOrderSaveReqVO;
@@ -17,6 +20,10 @@
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.feedback.MesProFeedbackDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.route.MesProRouteProductBomDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.route.MesProRouteProductDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemTypeDO;
import cn.iocoder.yudao.module.mes.enums.md.MesMdItemTypeEnum;
import cn.iocoder.yudao.module.mes.service.md.item.MesMdItemTypeService;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.route.MesProRouteProcessDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workorder.MesProWorkOrderDO;
import cn.iocoder.yudao.module.mes.enums.pro.MesProMpsStatusEnum;
@@ -30,6 +37,7 @@
import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderProcessService;
import cn.iocoder.yudao.module.mes.controller.admin.pro.workorder.vo.process.MesProWorkOrderProcessSaveReqVO;
import cn.iocoder.yudao.module.crm.api.customer.CrmCustomerApi;
import cn.iocoder.yudao.module.crm.api.customer.dto.CrmCustomerRespDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
@@ -43,8 +51,10 @@
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
@@ -53,7 +63,7 @@
/**
 * MES 生产主计划(MPS)Service 实现类
 *
 * @author 芋道源码
 * @author 超级管理员
 */
@Service
@Validated
@@ -73,6 +83,8 @@
    private MesProWorkOrderService workOrderService;
    @Resource
    private MesMdItemService itemService;
    @Resource
    private MesMdItemTypeService itemTypeService;
    @Resource
    private MesMdAutoCodeRecordService autoCodeRecordService;
    @Resource
@@ -94,11 +106,12 @@
                          Long mdmItemId, String productName, BigDecimal quantity, Long clientId,
                          String requestDate, String remark) {
        // 1. 根据 MDM 物料 ID 查找 MES 产品
        cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemDO mesItem = itemService.getItemByMdmItemId(mdmItemId);
        MesMdItemDO mesItem = itemService.getItemByMdmItemId(mdmItemId);
        if (mesItem == null) {
            // MES 产品不存在,抛出异常并包含产品名称提示
            throw exception(PRO_MPS_MES_PRODUCT_NOT_EXISTS, productName != null ? productName : mdmItemId);
        }
        validateProductItem(mesItem.getId());
        // 2. 生成 MPS 编码
        String code = generateMpsCode();
@@ -138,8 +151,8 @@
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Long createMps(MesProMpsSaveReqVO createReqVO) {
        // 校验产品存在
        itemService.validateItemExists(createReqVO.getProductId());
        // 校验产品存在、启用且为产品类型
        validateProductItem(createReqVO.getProductId());
        // 如果未填写编码,自动生成
        if (StrUtil.isBlank(createReqVO.getCode())) {
@@ -165,6 +178,8 @@
    public void updateMps(MesProMpsSaveReqVO updateReqVO) {
        // 校验存在 + 草稿状态
        validateMpsExistsAndDraft(updateReqVO.getId());
        // 校验产品存在、启用且为产品类型
        validateProductItem(updateReqVO.getProductId());
        // 校验编码唯一
        validateMpsCodeUnique(updateReqVO.getId(), updateReqVO.getCode());
@@ -420,6 +435,73 @@
        return false;
    }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public MesProMpsImportRespVO importMpsList(List<MesProMpsImportExcelVO> importList, boolean updateSupport) {
        if (CollUtil.isEmpty(importList)) {
            throw exception(MES_PRO_MPS_IMPORT_LIST_IS_EMPTY);
        }
        MesProMpsImportRespVO respVO = MesProMpsImportRespVO.builder()
                .createCodes(new ArrayList<>()).updateCodes(new ArrayList<>())
                .failureCodes(new LinkedHashMap<>()).build();
        AtomicInteger index = new AtomicInteger(1);
        importList.forEach(importItem -> {
            int currentIndex = index.getAndIncrement();
            String key = StrUtil.blankToDefault(importItem.getCode(), "第 " + currentIndex + " 行");
            // 1. 解析产品
            if (StrUtil.isBlank(importItem.getProductCode())) {
                respVO.getFailureCodes().put(key, "产品编码不能为空");
                return;
            }
            MesMdItemDO item = itemService.getItemByCode(importItem.getProductCode().trim());
            if (item == null) {
                respVO.getFailureCodes().put(key, "产品编码[" + importItem.getProductCode() + "]不存在");
                return;
            }
            // 2. 解析客户
            Long clientId = null;
            if (StrUtil.isNotBlank(importItem.getClientName())) {
                CrmCustomerRespDTO customer = customerApi.getCustomerByName(importItem.getClientName().trim());
                if (customer == null) {
                    respVO.getFailureCodes().put(key, "客户名称[" + importItem.getClientName() + "]不存在");
                    return;
                }
                clientId = customer.getId();
            }
            // 3. 构建请求并创建/更新(复用现有 create/update 服务)
            MesProMpsSaveReqVO reqVO = BeanUtils.toBean(importItem, MesProMpsSaveReqVO.class);
            reqVO.setProductId(item.getId());
            reqVO.setUnitMeasureId(item.getUnitMeasureId());
            reqVO.setClientId(clientId);
            try {
                if (StrUtil.isBlank(importItem.getCode())) {
                    // 编码为空:仅创建(自动生成编码)
                    reqVO.setId(null);
                    reqVO.setCode(null);
                    createMps(reqVO);
                    respVO.getCreateCodes().add(StrUtil.isBlank(reqVO.getCode())
                            ? String.valueOf(currentIndex) : reqVO.getCode());
                } else {
                    MesProMpsDO existMps = mpsMapper.selectByCode(importItem.getCode());
                    if (existMps == null) {
                        reqVO.setId(null);
                        createMps(reqVO);
                        respVO.getCreateCodes().add(importItem.getCode());
                    } else if (updateSupport) {
                        reqVO.setId(existMps.getId());
                        updateMps(reqVO);
                        respVO.getUpdateCodes().add(importItem.getCode());
                    } else {
                        respVO.getFailureCodes().put(key, "MPS 编码已存在");
                    }
                }
            } catch (ServiceException e) {
                respVO.getFailureCodes().put(key, e.getMessage());
            }
        });
        return respVO;
    }
    /**
     * 判断单个 MPS 是否完成
     */
@@ -509,6 +591,17 @@
    }
    /**
     * 校验产品存在、启用且为产品类型
     */
    private void validateProductItem(Long itemId) {
        MesMdItemDO item = itemService.validateItemExistsAndEnable(itemId);
        MesMdItemTypeDO itemType = itemTypeService.getItemType(item.getItemTypeId());
        if (itemType == null || !MesMdItemTypeEnum.isProduct(itemType.getItemOrProduct())) {
            throw exception(PRO_MPS_PRODUCT_TYPE_INVALID);
        }
    }
    /**
     * 根据 MPS 创建生产工单
     */
    private Long createWorkOrderFromMps(MesProMpsDO mps, BigDecimal quantity) {