From bc1b16a708ca7bec89d2e1a435c023477f107a59 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期日, 25 十月 2026 12:42:04 +0800
Subject: [PATCH] feat(mes): 添加MES系统自动编码规则支持

---
 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/maintenanceplan/MesProMaintenancePlanServiceImpl.java |  107 +++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 102 insertions(+), 5 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/maintenanceplan/MesProMaintenancePlanServiceImpl.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/maintenanceplan/MesProMaintenancePlanServiceImpl.java
index ef5b5a1..d0b58fb 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/maintenanceplan/MesProMaintenancePlanServiceImpl.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/maintenanceplan/MesProMaintenancePlanServiceImpl.java
@@ -2,10 +2,14 @@
 
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.date.DateUtil;
+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.erp.api.purchase.ErpPurchasePlanApi;
+import cn.iocoder.yudao.module.erp.api.purchase.PurchasePlanApi;
 import cn.iocoder.yudao.module.erp.api.purchase.dto.ErpPurchasePlanCreateReqDTO;
+import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan.vo.MesProMaintenancePlanImportExcelVO;
+import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan.vo.MesProMaintenancePlanImportRespVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan.vo.MesProMaintenancePlanPageReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenanceplan.vo.MesProMaintenancePlanSaveReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.workorder.vo.MesProWorkOrderSaveReqVO;
@@ -15,9 +19,11 @@
 import cn.iocoder.yudao.module.mes.dal.dataobject.pro.maintenanceplan.MesProMaintenancePlanItemDO;
 import cn.iocoder.yudao.module.mes.dal.mysql.pro.maintenanceplan.MesProMaintenancePlanItemMapper;
 import cn.iocoder.yudao.module.mes.dal.mysql.pro.maintenanceplan.MesProMaintenancePlanMapper;
+import cn.iocoder.yudao.module.mes.enums.md.autocode.MesMdAutoCodeRuleCodeEnum;
 import cn.iocoder.yudao.module.mes.enums.pro.MesProWorkOrderSourceTypeEnum;
 import cn.iocoder.yudao.module.mes.enums.pro.MesProWorkOrderTypeEnum;
 import cn.iocoder.yudao.module.mes.service.cal.team.MesCalTeamService;
+import cn.iocoder.yudao.module.mes.service.md.autocode.MesMdAutoCodeRecordService;
 import cn.iocoder.yudao.module.mes.service.md.item.MesMdItemService;
 import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderService;
 import jakarta.annotation.Resource;
@@ -27,7 +33,9 @@
 import org.springframework.validation.annotation.Validated;
 
 import java.math.BigDecimal;
+import java.util.ArrayList;
 import java.util.Collection;
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
@@ -74,9 +82,11 @@
     private MesMdItemService itemService;
 
     @Resource
-    private ErpPurchasePlanApi erpPurchasePlanApi;
+    private PurchasePlanApi purchasePlanApi;
     @Resource
     private MesProWorkOrderService workOrderService;
+    @Resource
+    private MesMdAutoCodeRecordService autoCodeRecordService;
 
     @Override
     @Transactional(rollbackFor = Exception.class)
@@ -199,7 +209,7 @@
         createReqDTO.setPlanDate(plan.getPlanDate());
         createReqDTO.setRemark("鏉ユ簮璁″垝: " + plan.getCode());
         createReqDTO.setItems(items);
-        return erpPurchasePlanApi.createPurchasePlan(createReqDTO);
+        return purchasePlanApi.createPurchasePlan(createReqDTO);
     }
 
     @Override
@@ -216,6 +226,7 @@
         }
         // 3. 缁勮宸ュ崟
         MesProWorkOrderSaveReqVO workOrderReq = new MesProWorkOrderSaveReqVO();
+        workOrderReq.setCode(generateWorkOrderCode());
         workOrderReq.setName(plan.getName());
         workOrderReq.setType(convertPlanTypeToWorkOrderType(plan.getPlanType()));
         workOrderReq.setOrderSourceType(MesProWorkOrderSourceTypeEnum.MAINTENANCE_PLAN.getType());
@@ -230,6 +241,70 @@
         updateObj.setWorkOrderId(workOrderId);
         planMapper.updateById(updateObj);
         return workOrderId;
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public MesProMaintenancePlanImportRespVO importPlanList(List<MesProMaintenancePlanImportExcelVO> importList,
+                                                            boolean updateSupport) {
+        if (CollUtil.isEmpty(importList)) {
+            throw exception(MES_PRO_MAINTENANCE_PLAN_IMPORT_LIST_IS_EMPTY);
+        }
+        MesProMaintenancePlanImportRespVO respVO = MesProMaintenancePlanImportRespVO.builder()
+                .createNames(new ArrayList<>()).updateNames(new ArrayList<>())
+                .failureNames(new LinkedHashMap<>()).build();
+        // 1. 鎸夎鍒掑悕绉板垎缁勶紙鍚屼竴璁″垝鍚嶇О鐨勫琛屽悎骞朵负涓�涓鍒掞紝姣忚涓�鏉″浠�/鐗╄祫鏄庣粏锛�
+        int rowIndex = 1;
+        Map<String, List<MesProMaintenancePlanImportExcelVO>> groupMap = new LinkedHashMap<>();
+        for (MesProMaintenancePlanImportExcelVO item : importList) {
+            if (StrUtil.isBlank(item.getName())) {
+                respVO.getFailureNames().put("绗� " + rowIndex + " 琛�", "璁″垝鍚嶇О涓嶈兘涓虹┖");
+                rowIndex++;
+                continue;
+            }
+            groupMap.computeIfAbsent(item.getName(), k -> new ArrayList<>()).add(item);
+            rowIndex++;
+        }
+        // 2. 閫愮粍澶勭悊
+        groupMap.forEach((name, rows) -> {
+            MesProMaintenancePlanImportExcelVO main = rows.get(0);
+            List<MesProMaintenancePlanSaveReqVO.Item> items = new ArrayList<>();
+            for (MesProMaintenancePlanImportExcelVO row : rows) {
+                if (StrUtil.isBlank(row.getItemCode())) {
+                    respVO.getFailureNames().put(name, "鐗╂枡/澶囦欢缂栫爜涓嶈兘涓虹┖");
+                    return;
+                }
+                MesMdItemDO item = itemService.getItemByCode(row.getItemCode().trim());
+                if (item == null) {
+                    respVO.getFailureNames().put(name, "鐗╂枡/澶囦欢缂栫爜[" + row.getItemCode() + "]涓嶅瓨鍦�");
+                    return;
+                }
+                MesProMaintenancePlanSaveReqVO.Item it = new MesProMaintenancePlanSaveReqVO.Item();
+                it.setItemId(item.getId());
+                it.setQuantity(row.getItemQuantity());
+                it.setRemark(row.getItemRemark());
+                items.add(it);
+            }
+            MesProMaintenancePlanSaveReqVO reqVO = BeanUtils.toBean(main, MesProMaintenancePlanSaveReqVO.class);
+            reqVO.setItems(items);
+            MesProMaintenancePlanDO existPlan = planMapper.selectByName(name);
+            try {
+                if (existPlan == null) {
+                    reqVO.setId(null);
+                    createPlan(reqVO);
+                    respVO.getCreateNames().add(name);
+                } else if (updateSupport) {
+                    reqVO.setId(existPlan.getId());
+                    updatePlan(reqVO);
+                    respVO.getUpdateNames().add(name);
+                } else {
+                    respVO.getFailureNames().put(name, "璁″垝鍚嶇О宸插瓨鍦�");
+                }
+            } catch (ServiceException e) {
+                respVO.getFailureNames().put(name, e.getMessage());
+            }
+        });
+        return respVO;
     }
 
     /**
@@ -265,8 +340,30 @@
     }
 
     private String generatePlanCode() {
-        return "MP" + DateUtil.format(java.time.LocalDateTime.now(), "yyyyMMddHHmmss")
-                + String.format("%03d", (int) (Math.random() * 1000));
+        try {
+            return autoCodeRecordService.generateAutoCode(
+                    MesMdAutoCodeRuleCodeEnum.PRO_MAINTENANCE_PLAN_CODE.getCode());
+        } catch (Exception e) {
+            log.warn("[generatePlanCode] 缂栫爜瑙勫垯 {} 涓嶅瓨鍦紝浣跨敤鏃堕棿鎴崇敓鎴�",
+                    MesMdAutoCodeRuleCodeEnum.PRO_MAINTENANCE_PLAN_CODE.getCode(), e);
+            return "MP" + DateUtil.format(java.time.LocalDateTime.now(), "yyyyMMddHHmmssSSS");
+        }
+    }
+
+    /**
+     * 鐢熸垚宸ュ崟缂栫爜
+     *
+     * 浼樺厛浣跨敤缂栫爜瑙勫垯锛涜嫢鐜涓湭閰嶇疆 PRO_WORK_ORDER_CODE 瑙勫垯锛屽垯闄嶇骇涓烘椂闂存埑缂栫爜锛岄伩鍏嶇敓鎴愬伐鍗曡闃绘柇
+     */
+    private String generateWorkOrderCode() {
+        try {
+            return autoCodeRecordService.generateAutoCode(
+                    MesMdAutoCodeRuleCodeEnum.PRO_WORK_ORDER_CODE.getCode());
+        } catch (Exception e) {
+            log.warn("[generateWorkOrderCode] 缂栫爜瑙勫垯 {} 涓嶅瓨鍦紝浣跨敤鏃堕棿鎴崇敓鎴�",
+                    MesMdAutoCodeRuleCodeEnum.PRO_WORK_ORDER_CODE.getCode(), e);
+            return "WO" + DateUtil.format(java.time.LocalDateTime.now(), "yyyyMMddHHmmssSSS");
+        }
     }
 
     private MesProMaintenancePlanDO validatePlanExists(Long id) {

--
Gitblit v1.9.3