From f4d4d29368ccacb807f93e2033cd4a643a3ddade Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期一, 30 三月 2026 10:20:28 +0800
Subject: [PATCH] yys 1.修改bug

---
 src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java |  126 ++++++++++++++++++++++--------------------
 1 files changed, 66 insertions(+), 60 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 2554b5e..982b0c2 100644
--- a/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java
+++ b/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java
@@ -2,6 +2,7 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.basic.pojo.Product;
@@ -97,6 +98,53 @@
         return AjaxResult.error();
     }
 
+    /**
+     * 閫掑綊淇濆瓨 BOM锛氬厛淇濆瓨鐖堕」 鈫� 淇濆瓨瀛愰」 鈫� 寤虹珛鍏崇郴 鈫� 閫掑綊瀛愰」鐨勫瓙椤�
+     * @param children 褰撳墠鐖堕」鐨勫瓙椤瑰垪琛�
+     * @return 淇濆瓨鍚庣殑鐖堕」浜у搧ID
+     */
+    private void saveBomRecursive(List<BomImportDto> children,ProductBom bom,ProductModel rootModel,Map<String, Long> processMap) {
+        // 1. 鑾峰彇children涓瓙椤逛骇鍝佺紪鍙蜂负绌虹殑鏁版嵁
+        List<BomImportDto> parentChildren = children
+                .stream()
+                .filter(child -> StringUtils.isEmpty(child.getChildCode()))
+                .collect(Collectors.toList());
+        if(CollectionUtils.isEmpty(parentChildren)){
+            throw new ServiceException("璇烽�夋嫨鐖堕」浜у搧缂栧彿");
+        }
+        BomImportDto parentId = parentChildren.get(0); // 鐖剁骇鏁版嵁
+        ProductStructure rootNode = new ProductStructure();
+        rootNode.setBomId(bom.getId());
+        rootNode.setParentId(null); // 椤跺眰娌℃湁鐖惰妭鐐�
+        rootNode.setProductModelId(rootModel.getId());
+        rootNode.setUnitQuantity(BigDecimal.ONE);
+        rootNode.setUnit(rootModel.getUnit());
+        productStructureService.save(rootNode);
+
+        // 2. 閬嶅巻瀛愰」锛岄�愪釜澶勭悊
+        for (BomImportDto child : children) {
+            //  璺宠繃娌℃湁瀛愰」鐨勭埗椤�
+            if(StringUtils.isEmpty(child.getChildCode())){
+                continue;
+            }
+            //  鑾峰彇瀛愰」妯″瀷淇℃伅
+            ProductModel childModel = findModel(child.getChildName(), child.getChildSpec());
+
+            //  鎻掑叆缁撴瀯琛�
+            ProductStructure node = new ProductStructure();
+            node.setBomId(bom.getId());
+            node.setParentId(rootNode.getId()); // 鐖惰妭鐐笽D
+            node.setProductModelId(childModel.getId());
+            node.setUnitQuantity(child.getUnitQty());
+            node.setUnit(childModel.getUnit());
+            if (processMap.containsKey(child.getProcess())) {
+                node.setProcessId(processMap.get(child.getProcess()));
+            }
+            productStructureService.save(node);
+        }
+    }
+
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public AjaxResult uploadBom(MultipartFile file) {
@@ -120,67 +168,25 @@
         handleProcess(list);
         Map<String, Long> processMap = productProcessService.list().stream()
                 .collect(Collectors.toMap(ProductProcess::getName, ProductProcess::getId, (k1, k2) -> k1));
-
-        //  鍒涘缓 BOM 鏁版嵁
-        BomImportDto first = list.get(0);
-        ProductModel rootModel = findModel(first.getParentName(), first.getParentSpec());
-        ProductBom bom = new ProductBom();
-        bom.setProductModelId(rootModel.getId());
-        bom.setVersion("1.0");
-        productBomMapper.insert(bom);
-        bom.setBomNo("BM." + String.format("%05d", bom.getId()));
-        productBomMapper.updateById(bom);
-
-        // 璁板綍宸茬粡鎻掑叆缁撴瀯鐨勮妭鐐癸細Key = "鍚嶇О+瑙勬牸", Value = structure_id
-        Map<String, Long> treePathMap = new HashMap<>();
-
-        for (int i = 0; i < list.size(); i++) {
-            BomImportDto dto = list.get(i);
-            String parentKey = dto.getParentName() + "|" + dto.getParentSpec();
-            String childKey = dto.getChildName() + "|" + dto.getChildSpec();
-
-            //澶勭悊鏍硅妭鐐�,绗竴琛屼笖瀛愰」涓虹┖
-            if (i == 0 && StringUtils.isBlank(dto.getChildName())) {
-                ProductStructure rootNode = new ProductStructure();
-                rootNode.setBomId(bom.getId());
-                rootNode.setParentId(null); // 椤跺眰娌℃湁鐖惰妭鐐�
-                rootNode.setProductModelId(rootModel.getId());
-                rootNode.setUnitQuantity(BigDecimal.ONE);
-                rootNode.setUnit(rootModel.getUnit());
-                productStructureService.save(rootNode);
-
-                treePathMap.put(parentKey, rootNode.getId());
-                continue;
-            }
-
-            //  澶勭悊瀛愬眰绾ц妭鐐�
-            //  鎵惧埌鐖惰妭鐐瑰湪鏁版嵁搴撻噷鐨� ID
-            Long parentStructureId = treePathMap.get(parentKey);
-            if (parentStructureId == null) {
-                // 濡傛灉 Map 閲屾壘涓嶅埌锛岃鏄� Excel 椤哄簭涔变簡鎴栬�呮暟鎹湁璇�
-                throw new ServiceException("瀵煎叆澶辫触: 鐖堕」[" + dto.getParentName() + "]蹇呴』鍦ㄥ叾瀛愰」涔嬪墠瀹氫箟");
-            }
-
-            //  鑾峰彇瀛愰」妯″瀷淇℃伅
-            ProductModel childModel = findModel(dto.getChildName(), dto.getChildSpec());
-
-            //  鎻掑叆缁撴瀯琛�
-            ProductStructure node = new ProductStructure();
-            node.setBomId(bom.getId());
-            node.setParentId(parentStructureId); // 鐖惰妭鐐笽D
-            node.setProductModelId(childModel.getId());
-            node.setUnitQuantity(dto.getUnitQty());
-            node.setUnit(childModel.getUnit());
-            if (processMap.containsKey(dto.getProcess())) {
-                node.setProcessId(processMap.get(dto.getProcess()));
-            }
-            productStructureService.save(node);
-
-            //  鎶婂綋鍓嶅瓙椤硅褰曞埌 Map,浣滀负浠ュ悗鏇存繁灞傜骇鐨勭埗椤规煡鎵句緷鎹�
-            //  鍚屼竴鐖堕」涓嬬殑鍚屽悕瀛愰」涓嶉渶瑕侀噸澶嶈褰�
-            treePathMap.put(childKey, node.getId());
+        // 1. 鎸夌埗椤圭紪鐮佸垎缁�
+        Map<String, List<BomImportDto>> parentMap = list.stream()
+                .filter(bom -> StringUtils.isNotBlank(bom.getParentCode()))
+                .collect(Collectors.groupingBy(BomImportDto::getParentCode));
+        // 2. 閬嶅巻鎵�鏈夌埗椤癸紝閫掑綊淇濆瓨
+        for (Map.Entry<String, List<BomImportDto>> entry : parentMap.entrySet()) {
+            //  鍒涘缓 BOM 鏁版嵁
+            BomImportDto first = entry.getValue().get(0);
+            ProductModel rootModel = findModel(first.getParentName(), first.getParentSpec());
+            ProductBom bom = new ProductBom();
+            bom.setProductModelId(rootModel.getId());
+            bom.setVersion("1.0");
+            productBomMapper.insert(bom);
+            bom.setBomNo("BM." + String.format("%05d", bom.getId()));
+            productBomMapper.updateById(bom);
+            // 澶勭悊bom瀛愯〃鏁版嵁
+            List<BomImportDto> children = entry.getValue();
+            saveBomRecursive(children,bom,rootModel,processMap);
         }
-
         return AjaxResult.success("BOM瀵煎叆鎴愬姛");
     }
 

--
Gitblit v1.9.3