From 9241ecc52d1f33a608d8cc501aa930122293105f Mon Sep 17 00:00:00 2001
From: liding <756868258@qq.com>
Date: 星期二, 12 五月 2026 16:46:33 +0800
Subject: [PATCH] fix:1.BOM导出

---
 src/main/java/com/ruoyi/technology/bean/vo/TechnologyBomStructureVo.java      |    3 +
 src/main/resources/mapper/technology/TechnologyBomStructureMapper.xml         |    3 
 src/main/java/com/ruoyi/basic/pojo/ProductModel.java                          |    1 
 src/main/java/com/ruoyi/technology/service/TechnologyBomService.java          |    2 
 src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java |  123 ++++++++++++++++++++--------------------
 src/main/java/com/ruoyi/technology/controller/TechnologyBomController.java    |    2 
 6 files changed, 69 insertions(+), 65 deletions(-)

diff --git a/src/main/java/com/ruoyi/basic/pojo/ProductModel.java b/src/main/java/com/ruoyi/basic/pojo/ProductModel.java
index f0e9470..8c0dc07 100644
--- a/src/main/java/com/ruoyi/basic/pojo/ProductModel.java
+++ b/src/main/java/com/ruoyi/basic/pojo/ProductModel.java
@@ -38,7 +38,6 @@
     private String model;
 
     @Excel(name = "浜у搧缂栫爜")
-    @TableField("product_code")
     private String productCode;
 
     /**
diff --git a/src/main/java/com/ruoyi/technology/bean/vo/TechnologyBomStructureVo.java b/src/main/java/com/ruoyi/technology/bean/vo/TechnologyBomStructureVo.java
index e41540d..d3719d6 100644
--- a/src/main/java/com/ruoyi/technology/bean/vo/TechnologyBomStructureVo.java
+++ b/src/main/java/com/ruoyi/technology/bean/vo/TechnologyBomStructureVo.java
@@ -23,5 +23,8 @@
     @Schema(description = "瑙勬牸鍨嬪彿")
     private String model;
 
+    @Schema(description = "浜у搧缂栫爜")
+    private String productCode;
+
     private List<TechnologyBomStructureVo> children;
 }
diff --git a/src/main/java/com/ruoyi/technology/controller/TechnologyBomController.java b/src/main/java/com/ruoyi/technology/controller/TechnologyBomController.java
index 226973f..9e35143 100644
--- a/src/main/java/com/ruoyi/technology/controller/TechnologyBomController.java
+++ b/src/main/java/com/ruoyi/technology/controller/TechnologyBomController.java
@@ -76,7 +76,7 @@
     @PreAuthorize("@ss.hasPermi('product:bom:export')")
     @Operation(summary = "瀵煎嚭BOM鏂囦欢")
     @Log(title = "瀵煎嚭BOM鏂囦欢", businessType = BusinessType.EXPORT)
-    public void exportBom(HttpServletResponse response, @RequestParam Integer bomId) {
+    public void exportBom(HttpServletResponse response, @RequestParam Long bomId) {
         technologyBomService.exportBom(response, bomId);
     }
 
diff --git a/src/main/java/com/ruoyi/technology/service/TechnologyBomService.java b/src/main/java/com/ruoyi/technology/service/TechnologyBomService.java
index 31fd335..d4aa3d3 100644
--- a/src/main/java/com/ruoyi/technology/service/TechnologyBomService.java
+++ b/src/main/java/com/ruoyi/technology/service/TechnologyBomService.java
@@ -26,7 +26,7 @@
 
     R uploadBom(MultipartFile file);
 
-    void exportBom(HttpServletResponse response, Integer bomId);
+    void exportBom(HttpServletResponse response, Long bomId);
 
     R copy(TechnologyBom technologyBom);
 }
diff --git a/src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java b/src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java
index edac5ac..8894288 100644
--- a/src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java
+++ b/src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java
@@ -15,7 +15,6 @@
 import com.ruoyi.common.utils.bean.BeanUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.framework.web.domain.R;
-import com.ruoyi.production.bean.dto.ProductStructureDto;
 import com.ruoyi.technology.bean.dto.BomImportDto;
 import com.ruoyi.technology.bean.dto.TechnologyBomDto;
 import com.ruoyi.technology.bean.dto.TechnologyBomStructureDto;
@@ -263,70 +262,72 @@
 
 
     @Override
-    public void exportBom(HttpServletResponse response, Integer bomId) {
+    public void exportBom(HttpServletResponse response, Long bomId) {
         if (bomId == null) {
+            throw new ServiceException("BOM ID涓嶈兘涓虹┖");
+        }
+
+        List<TechnologyBomStructureVo> treeData = technologyBomStructureService.listByBomId(bomId);
+        if (treeData == null || treeData.isEmpty()) {
             return;
         }
 
-//        List<ProductStructureDto> treeData = productStructureService.listBybomId(bomId);
-//        if (treeData == null || treeData.isEmpty()) {
-//            return;
-//        }
-//
-//        //  灏嗘爲褰㈢粨鏋勬墎骞冲寲 浣跨敤 BFS绠楁硶 瀵煎嚭,鎸夊眰绾ч『搴�
-//        List<BomImportDto> exportList = new ArrayList<>();
-//
-//        // Map<ID, Node> idMap 鐢ㄤ簬鏌ユ壘鐖惰妭鐐�
-//        Map<Long, ProductStructureDto> idMap = new HashMap<>();
-//        populateMap(treeData, idMap);
-//
-//        //  treeData 鐨勭涓�涓槸鏍硅妭鐐�
-//        for (ProductStructureDto root : treeData) {
-//            //  娣诲姞鏍硅妭鐐�
-//            BomImportDto rootRow = new BomImportDto();
-//            rootRow.setParentName(root.getProductName());
-//            rootRow.setParentSpec(root.getModel());
-//            rootRow.setUnitQty(root.getUnitQuantity());
-//            rootRow.setRemark("");
-//            exportList.add(rootRow);
-//
-//            //  BFS 閬嶅巻-闃熷垪
-//            Queue<ProductStructureDto> queue = new LinkedList<>();
-//            if (root.getChildren() != null) {
-//                queue.addAll(root.getChildren());
-//            }
-//
-//            while (!queue.isEmpty()) {
-//                ProductStructureDto child = queue.poll();
-//
-//                // 鏌ユ壘鐖惰妭鐐�
-//                ProductStructureDto parent = idMap.get(child.getParentId());
-//                if (parent == null) {
-//                    // 闄や簡鏈�澶栧眰鑺傜偣,鍏朵粬鑺傜偣鐨勭埗绫昏偗瀹氭槸涓嶄細涓虹┖鐨�
-//                    continue;
-//                }
-//
-//                BomImportDto row = new BomImportDto();
-//                // 鐖剁被淇℃伅
-//                row.setParentName(parent.getProductName());
-//                row.setParentSpec(parent.getModel());
-//                // 瀛愮被淇℃伅
-//                row.setChildName(child.getProductName());
-//                row.setChildSpec(child.getModel());
-//                row.setUnitQty(child.getUnitQuantity());
-//                row.setProcess(child.getProcessName());
-//
-//                exportList.add(row);
-//
-//                //  灏嗗瓙鑺傜偣鐨勫瓙鑺傜偣鍔犲叆闃熷垪-涓嬩竴灞�
-//                if (child.getChildren() != null && !child.getChildren().isEmpty()) {
-//                    queue.addAll(child.getChildren());
-//                }
-//            }
-//        }
+        //  灏嗘爲褰㈢粨鏋勬墎骞冲寲 浣跨敤 BFS绠楁硶 瀵煎嚭,鎸夊眰绾ч『搴�
+        List<BomImportDto> exportList = new ArrayList<>();
+
+        // Map<ID, Node> idMap 鐢ㄤ簬鏌ユ壘鐖惰妭鐐�
+        Map<Long, TechnologyBomStructureVo> idMap = new HashMap<>();
+        populateMap(treeData, idMap);
+
+        //  treeData 鐨勭涓�涓槸鏍硅妭鐐�
+        for (TechnologyBomStructureVo root : treeData) {
+            //  娣诲姞鏍硅妭鐐�
+            BomImportDto rootRow = new BomImportDto();
+            rootRow.setParentName(root.getProductName());
+            rootRow.setParentSpec(root.getModel());
+            rootRow.setUnitQty(root.getUnitQuantity());
+            rootRow.setParentCode(root.getProductCode());
+            rootRow.setRemark("");
+            exportList.add(rootRow);
+
+            //  BFS 閬嶅巻-闃熷垪
+            Queue<TechnologyBomStructureVo> queue = new LinkedList<>();
+            if (root.getChildren() != null) {
+                queue.addAll(root.getChildren());
+            }
+
+            while (!queue.isEmpty()) {
+                TechnologyBomStructureVo child = queue.poll();
+
+                // 鏌ユ壘鐖惰妭鐐�
+                TechnologyBomStructureVo parent = idMap.get(child.getParentId());
+                if (parent == null) {
+                    // 闄や簡鏈�澶栧眰鑺傜偣,鍏朵粬鑺傜偣鐨勭埗绫昏偗瀹氭槸涓嶄細涓虹┖鐨�
+                    continue;
+                }
+
+                BomImportDto row = new BomImportDto();
+                // 鐖剁被淇℃伅
+                row.setParentName(parent.getProductName());
+                row.setParentSpec(parent.getModel());
+                // 瀛愮被淇℃伅
+                row.setChildName(child.getProductName());
+                row.setChildSpec(child.getModel());
+                row.setUnitQty(child.getUnitQuantity());
+                row.setProcess(child.getOperationName());
+                row.setChildCode(child.getProductCode());
+
+                exportList.add(row);
+
+                //  灏嗗瓙鑺傜偣鐨勫瓙鑺傜偣鍔犲叆闃熷垪-涓嬩竴灞�
+                if (child.getChildren() != null && !child.getChildren().isEmpty()) {
+                    queue.addAll(child.getChildren());
+                }
+            }
+        }
 
         ExcelUtil<BomImportDto> util = new ExcelUtil<>(BomImportDto.class);
-//        util.exportExcel(response, exportList, "BOM缁撴瀯瀵煎嚭");
+        util.exportExcel(response, exportList, "BOM缁撴瀯瀵煎嚭");
     }
 
     @Override
@@ -408,11 +409,11 @@
         return s.replaceAll("[\\u00A0\\u3000]", "").trim();
     }
 
-    private void populateMap(List<ProductStructureDto> nodes, Map<Long, ProductStructureDto> map) {
+    private void populateMap(List<TechnologyBomStructureVo> nodes, Map<Long, TechnologyBomStructureVo> map) {
         if (nodes == null || nodes.isEmpty()) {
             return;
         }
-        for (ProductStructureDto node : nodes) {
+        for (TechnologyBomStructureVo node : nodes) {
             map.put(node.getId(), node);
             populateMap(node.getChildren(), map);
         }
diff --git a/src/main/resources/mapper/technology/TechnologyBomStructureMapper.xml b/src/main/resources/mapper/technology/TechnologyBomStructureMapper.xml
index 2030715..536cdba 100644
--- a/src/main/resources/mapper/technology/TechnologyBomStructureMapper.xml
+++ b/src/main/resources/mapper/technology/TechnologyBomStructureMapper.xml
@@ -21,7 +21,8 @@
                p.product_name as productName,
                pm.product_id as productId,
                pm.model,
-               top1.name as operationName
+               top1.name as operationName,
+               pm.product_code as productCode
         from technology_bom_structure tbs
         left join product_model pm on tbs.product_model_id = pm.id
         left join product p on pm.product_id = p.id

--
Gitblit v1.9.3