From 0a1f000d5031c090059d5ae960019825196d4ba9 Mon Sep 17 00:00:00 2001
From: gongchunyi <deslre0381@gmail.com>
Date: 星期五, 10 四月 2026 17:46:19 +0800
Subject: [PATCH] fix: BOM导入验证产品、工序是否存在

---
 src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java |  255 +++++++++++++++++++++++++++++++++++---------------
 1 files changed, 177 insertions(+), 78 deletions(-)

diff --git a/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java b/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java
index b5c9562..1315d5c 100644
--- a/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java
+++ b/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java
@@ -4,6 +4,8 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.production.pojo.ProductMaterial;
+import com.ruoyi.production.pojo.ProductMaterialSku;
 import com.ruoyi.common.exception.ServiceException;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
@@ -42,12 +44,17 @@
     @Autowired
     private ProductBomMapper productBomMapper;
 
-
     @Autowired
     private ProductStructureService productStructureService;
 
     @Autowired
     private ProductProcessService productProcessService;
+
+    @Autowired
+    private ProductMaterialService productMaterialService;
+
+    @Autowired
+    private ProductMaterialSkuService productMaterialSkuService;
 
     @Autowired
     private SysDictDataMapper sysDictDataMapper;
@@ -61,12 +68,12 @@
     @Transactional(rollbackFor = Exception.class)
     public AjaxResult add(ProductBom productBom) {
         if (productBom == null || productBom.getDictCode() == null) {
-            throw new ServiceException("鏂板澶辫触,浜у搧绫诲瀷涓嶈兘涓虹┖");
+            throw new ServiceException("鏂板BOM澶辫触,浜у搧绫诲瀷涓嶈兘涓虹┖");
         }
 
         SysDictData sysDictData = sysDictDataMapper.selectDictDataById(productBom.getDictCode());
         if (sysDictData == null) {
-            throw new ServiceException("鏂板澶辫触,浜у搧绫诲瀷涓嶅瓨鍦�");
+            throw new ServiceException("鏂板BOM澶辫触,浜у搧绫诲瀷涓嶅瓨鍦�");
         }
 
         boolean save = productBomMapper.insert(productBom) > 0;
@@ -90,7 +97,6 @@
         if (sysDictData == null) {
             return AjaxResult.error("瀵煎叆澶辫触,浜у搧绫诲瀷涓嶅瓨鍦�");
         }
-
         ExcelUtil<BomImportDto> util = new ExcelUtil<>(BomImportDto.class);
         List<BomImportDto> list;
         try {
@@ -99,71 +105,128 @@
             return AjaxResult.error("Excel瑙f瀽澶辫触");
         }
 
-        if (list == null || list.isEmpty()) return AjaxResult.error("鏁版嵁涓虹┖");
+        if (list == null || list.isEmpty()) {
+            return AjaxResult.error("鏁版嵁涓虹┖");
+        }
 
-        //  澶勭悊瀛楁娓呯悊
         list.forEach(dto -> {
-            dto.setParentCode(clean(dto.getParentCode()));
-            dto.setCode(clean(dto.getCode()));
+            dto.setParentName(clean(dto.getParentName()));
+            dto.setParentSpec(clean(dto.getParentSpec()));
+            dto.setChildName(clean(dto.getChildName()));
+            dto.setChildSpec(clean(dto.getChildSpec()));
+            dto.setProcess(clean(dto.getProcess()));
         });
-        handleProcess(list);
-        Map<String, Long> processMap = productProcessService.list().stream()
-                .collect(Collectors.toMap(ProductProcess::getName, ProductProcess::getId, (k1, k2) -> k1));
 
-        //  鍒涘缓 BOM 鏁版嵁
+        Set<String> processNames = list.stream()
+                .map(BomImportDto::getProcess)
+                .filter(StringUtils::isNotBlank)
+                .collect(Collectors.toSet());
+
+        Map<String, Long> processMap = new HashMap<>();
+        if (!processNames.isEmpty()) {
+            List<ProductProcess> exists = productProcessService.list(
+                    new LambdaQueryWrapper<ProductProcess>().in(ProductProcess::getName, processNames)
+            );
+            processMap = exists.stream().collect(Collectors.toMap(ProductProcess::getName, ProductProcess::getId, (k1, k2) -> k1));
+
+            Map<String, Long> finalProcessMap = processMap;
+            List<String> missingProcesses = processNames.stream()
+                    .filter(n -> !finalProcessMap.containsKey(n))
+                    .collect(Collectors.toList());
+
+            if (!missingProcesses.isEmpty()) {
+                throw new ServiceException("瀵煎叆澶辫触锛屼互涓嬪伐搴忎笉瀛樺湪锛�" + String.join(", ", missingProcesses));
+            }
+        }
+
+        Set<String> materialKeys = new HashSet<>();
+        for (BomImportDto dto : list) {
+            if (StringUtils.isNotBlank(dto.getParentName())) {
+                materialKeys.add(dto.getParentName() + "|" + (dto.getParentSpec() == null ? "" : dto.getParentSpec()));
+            }
+            if (StringUtils.isNotBlank(dto.getChildName())) {
+                materialKeys.add(dto.getChildName() + "|" + (dto.getChildSpec() == null ? "" : dto.getChildSpec()));
+            }
+        }
+
+        Map<String, ProductMaterialSku> skuMap = validateAndGetSkuMap(materialKeys);
+
         ProductBom bom = new ProductBom();
-        bom.setVersion("1.0");
         bom.setDictCode(dictCode);
+        bom.setVersion("1.0");
         productBomMapper.insert(bom);
         bom.setBomNo("BM." + String.format("%05d", bom.getId()));
         productBomMapper.updateById(bom);
 
-        // 璁板綍宸茬粡鎻掑叆缁撴瀯鐨勮妭鐐癸細Key = "缂栫爜", Value = structure_id
+        //  璁板綍宸茬粡鎻掑叆缁撴瀯鐨勮妭鐐癸細Key = "鍚嶇О+瑙勬牸", Value = structure_id
         Map<String, Long> treePathMap = new HashMap<>();
 
-        for (int i = 0; i < list.size(); i++) {
-            BomImportDto dto = list.get(i);
-            String currentCode = dto.getCode();
-            String parentCode = dto.getParentCode();
+        for (BomImportDto dto : list) {
+            String parentName = dto.getParentName();
+            String parentSpec = dto.getParentSpec();
+            String childName = dto.getChildName();
+            String childSpec = dto.getChildSpec();
 
-            // 澶勭悊鏍硅妭鐐癸細涓�鑸寚绗竴琛屼笖娌℃湁鐖堕」缂栧彿
-            if (i == 0 && StringUtils.isBlank(parentCode)) {
-                ProductStructure rootNode = new ProductStructure();
-                rootNode.setBomId(bom.getId());
-                rootNode.setParentId(null); // 椤跺眰娌℃湁鐖惰妭鐐�
-                if (processMap.containsKey(dto.getProcess())) {
-                    rootNode.setProcessId(processMap.get(dto.getProcess()));
+            String parentKey = parentName + "|" + (parentSpec == null ? "" : parentSpec);
+            String childKey = childName + "|" + (childSpec == null ? "" : childSpec);
+
+            //  澶勭悊椤剁骇鑺傜偣 (瀛愰」鍙婂叾瑙勬牸鍧囦负绌哄綋浣滈《绾ц妭鐐瑰畾涔�)
+            if (StringUtils.isBlank(childName) && StringUtils.isBlank(childSpec)) {
+                if (!treePathMap.containsKey(parentKey)) {
+                    ProductMaterialSku model = skuMap.get(parentKey);
+                    ProductStructure node = saveStructure(bom.getId(), null, model, dto.getUnitQty(), null);
+                    treePathMap.put(parentKey, node.getId());
+                } else {
+                    // 濡傛灉宸茬粡瀛樺湪锛屼粎鍦ㄥ畠鏄《绾ц妭鐐规椂鏇存柊
+                    Long structureId = treePathMap.get(parentKey);
+                    ProductStructure node = productStructureService.getById(structureId);
+                    if (node != null && node.getParentId() == null && dto.getUnitQty() != null) {
+                        node.setUnitQuantity(dto.getUnitQty());
+                        productStructureService.updateById(node);
+                    }
                 }
-                rootNode.setUnitQuantity(BigDecimal.ONE);
-                productStructureService.save(rootNode);
-
-                treePathMap.put(currentCode, rootNode.getId());
                 continue;
             }
 
-            //  澶勭悊瀛愬眰绾ц妭鐐�
-            //  鎵惧埌鐖惰妭鐐瑰湪鏁版嵁搴撻噷鐨� ID
-            Long parentStructureId = treePathMap.get(parentCode);
+            //  澶勭悊鐖跺瓙灞傜骇鍏崇郴
+            Long parentStructureId = treePathMap.get(parentKey);
             if (parentStructureId == null) {
-                // 濡傛灉 Map 閲屾壘涓嶅埌锛岃鏄� Excel 椤哄簭涔变簡鎴栬�呮暟鎹湁璇�
-                throw new ServiceException("瀵煎叆澶辫触: 鐖堕」[" + parentCode + "]蹇呴』鍦ㄥ叾瀛愰」涔嬪墠瀹氫箟");
+                // 濡傛灉 Map 閲屾壘涓嶅埌鐖惰妭鐐癸紝瑙嗕负椤剁骇鐖惰妭鐐�
+                ProductMaterialSku parentModel = skuMap.get(parentKey);
+                ProductStructure parentNode = saveStructure(bom.getId(), null, parentModel, BigDecimal.ONE, null);
+                parentStructureId = parentNode.getId();
+                treePathMap.put(parentKey, parentStructureId);
             }
 
-            //  鎻掑叆缁撴瀯琛�
-            ProductStructure node = new ProductStructure();
-            node.setBomId(bom.getId());
-            node.setParentId(parentStructureId); // 鐖惰妭鐐笽D
-            node.setUnitQuantity(dto.getUnitQty());
-            if (processMap.containsKey(dto.getProcess())) {
-                node.setProcessId(processMap.get(dto.getProcess()));
+            //  鑾峰彇瀛愰」妯″瀷淇℃伅
+            ProductMaterialSku childModel = skuMap.get(childKey);
+            Long processId = null;
+            if (StringUtils.isNotBlank(dto.getProcess())) {
+                processId = processMap.get(dto.getProcess());
             }
-            productStructureService.save(node);
-
-            //  鎶婂綋鍓嶉」璁板綍鍒� Map, 浣滀负浠ュ悗鏇存繁灞傜骇鐨勭埗椤规煡鎵句緷鎹�
-            treePathMap.put(currentCode, node.getId());
+            // 鎻掑叆瀛愯妭鐐圭粨鏋勮〃
+            ProductStructure childNode = saveStructure(bom.getId(), parentStructureId, childModel, dto.getUnitQty(), processId);
+            // 鎶婂綋鍓嶅瓙椤硅褰曞埌 Map,浣滀负浠ュ悗鏇存繁灞傜骇鐨勭埗椤规煡鎵句緷鎹�
+            treePathMap.put(childKey, childNode.getId());
         }
 
         return AjaxResult.success("BOM瀵煎叆鎴愬姛");
+    }
+
+    /**
+     * 淇濆瓨缁撴瀯鑺傜偣
+     */
+    private ProductStructure saveStructure(Integer bomId, Long parentId, ProductMaterialSku model, BigDecimal qty, Long processId) {
+        ProductStructure node = new ProductStructure();
+        node.setBomId(bomId);
+        node.setParentId(parentId);
+        node.setProductModelId(model.getId());
+        node.setUnitQuantity(qty != null ? qty : BigDecimal.ONE);
+        ProductMaterial material = productMaterialService.getById(model.getProductId());
+        node.setUnit(material != null ? material.getUnit() : null);
+        node.setProcessId(processId);
+        productStructureService.save(node);
+        return node;
     }
 
 
@@ -189,7 +252,8 @@
         for (ProductStructureDto root : treeData) {
             //  娣诲姞鏍硅妭鐐�
             BomImportDto rootRow = new BomImportDto();
-            rootRow.setCode(root.getId().toString());
+            rootRow.setParentName(root.getProductName());
+            rootRow.setParentSpec(root.getModel());
             rootRow.setUnitQty(root.getUnitQuantity());
             rootRow.setRemark("");
             exportList.add(rootRow);
@@ -211,14 +275,14 @@
                 }
 
                 BomImportDto row = new BomImportDto();
-                // 鐖剁被缂栧彿
-                row.setParentCode(parent.getId().toString());
-                // 鏈韩缂栧彿
-                row.setCode(child.getId().toString());
+                // 鐖剁被淇℃伅
+                row.setParentName(parent.getProductName());
+                row.setParentSpec(parent.getModel());
+                // 瀛愮被淇℃伅
+                row.setChildName(child.getProductName());
+                row.setChildSpec(child.getModel());
                 row.setUnitQty(child.getUnitQuantity());
                 row.setProcess(child.getProcessName());
-//                row.setProcess();
-                row.setRemark("");
 
                 exportList.add(row);
 
@@ -233,6 +297,14 @@
         util.exportExcel(response, exportList, "BOM缁撴瀯瀵煎嚭");
     }
 
+    @Override
+    public String strengthById(Long bomId) {
+        if (bomId == null) {
+            return null;
+        }
+        return baseMapper.selectStrengthById(bomId);
+    }
+
     private void populateMap(List<ProductStructureDto> nodes, Map<Long, ProductStructureDto> map) {
         if (nodes == null || nodes.isEmpty()) {
             return;
@@ -243,39 +315,68 @@
         }
     }
 
-    private void handleProcess(List<BomImportDto> list) {
 
-        Set<String> processNames = list.stream()
-                .map(BomImportDto::getProcess)
-                .filter(StringUtils::isNotBlank)
-                .collect(Collectors.toSet());
-
-        if (processNames.isEmpty()) {
-            return;
+    /**
+     * 鎵归噺楠岃瘉骞惰幏鍙栦骇鍝丼KU淇℃伅
+     */
+    private Map<String, ProductMaterialSku> validateAndGetSkuMap(Set<String> materialKeys) {
+        if (materialKeys == null || materialKeys.isEmpty()) {
+            return Collections.emptyMap();
         }
 
-        List<ProductProcess> exists = productProcessService.list(
-                new LambdaQueryWrapper<ProductProcess>().in(ProductProcess::getName, processNames)
-        );
-
-        Set<String> existNames = exists.stream()
-                .map(ProductProcess::getName)
+        // 1. 鑾峰彇鎵�鏈変骇鍝佸悕绉板苟鏌ヨ
+        Set<String> productNames = materialKeys.stream()
+                .map(k -> k.split("\\|")[0])
                 .collect(Collectors.toSet());
 
-        List<ProductProcess> needSave = processNames.stream()
-                .filter(n -> !existNames.contains(n))
-                .map(n -> {
-                    ProductProcess p = new ProductProcess();
-                    p.setName(n);
-                    return p;
+        List<ProductMaterial> products = productMaterialService.list(
+                new LambdaQueryWrapper<ProductMaterial>().in(ProductMaterial::getProductName, productNames)
+        );
+        Map<String, ProductMaterial> productMap = products.stream()
+                .collect(Collectors.toMap(ProductMaterial::getProductName, p -> p, (k1, k2) -> k1));
+
+        // 妫�鏌ョ己澶辩殑浜у搧
+        List<String> missingProducts = productNames.stream()
+                .filter(n -> !productMap.containsKey(n))
+                .collect(Collectors.toList());
+        if (!missingProducts.isEmpty()) {
+            throw new ServiceException("瀵煎叆澶辫触锛屼互涓嬩骇鍝佹湭缁存姢锛�" + String.join(", ", missingProducts));
+        }
+
+        // 2. 鏌ヨ鎵�鏈夌浉鍏崇殑 SKU
+        // 鏋勯�犳煡璇細(productId = ? AND model = ?) OR (productId = ? AND model = ?) ...
+        // 鐢变簬 MyBatis Plus 鐨� Or 宓屽鍙兘杈冩參涓旀澶勬暟閲忛�氬父鍙帶锛屾垜浠彲浠ュ厛鏌ュ嚭鎵�鏈夎繖浜� Product 鐨勬墍鏈� SKU
+        List<Long> productIds = products.stream().map(ProductMaterial::getId).collect(Collectors.toList());
+        List<ProductMaterialSku> skus = productMaterialSkuService.list(
+                new LambdaQueryWrapper<ProductMaterialSku>().in(ProductMaterialSku::getProductId, productIds)
+        );
+
+        // 鏋勫缓 Map: "浜у搧鍚嶇О|瑙勬牸" -> Sku
+        Map<String, ProductMaterialSku> skuMap = new HashMap<>();
+        for (ProductMaterialSku sku : skus) {
+            ProductMaterial p = productMap.values().stream()
+                    .filter(pm -> pm.getId().equals(sku.getProductId()))
+                    .findFirst().orElse(null);
+            if (p != null) {
+                String key = p.getProductName() + "|" + (sku.getModel() == null ? "" : sku.getModel());
+                skuMap.put(key, sku);
+            }
+        }
+
+        // 妫�鏌ョ己澶辩殑瑙勬牸
+        List<String> missingSkus = materialKeys.stream()
+                .filter(k -> !skuMap.containsKey(k))
+                .map(k -> {
+                    String[] parts = k.split("\\|");
+                    return parts[0] + "[" + (parts.length > 1 ? parts[1] : "") + "]";
                 })
                 .collect(Collectors.toList());
 
-        if (!needSave.isEmpty()) {
-            productProcessService.saveBatch(needSave);
-            needSave.forEach(p -> p.setNo("GX" + String.format("%08d", p.getId())));
-            productProcessService.updateBatchById(needSave);
+        if (!missingSkus.isEmpty()) {
+            throw new ServiceException("瀵煎叆澶辫触锛屼互涓嬩骇鍝佽鏍兼湭缁存姢锛�" + String.join(", ", missingSkus));
         }
+
+        return skuMap;
     }
 
     private String clean(String s) {
@@ -283,5 +384,3 @@
         return s.replaceAll("[\\u00A0\\u3000]", "").trim();
     }
 }
-
-

--
Gitblit v1.9.3