From 79054eef527c991f9a84ab3bc3fd2603a5521eb1 Mon Sep 17 00:00:00 2001
From: liyong <18434998025@163.com>
Date: 星期一, 26 一月 2026 12:45:02 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/dev_New' into dev_New

---
 src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java |  291 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 280 insertions(+), 11 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 d519593..2554b5e 100644
--- a/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java
+++ b/src/main/java/com/ruoyi/production/service/impl/ProductBomServiceImpl.java
@@ -1,19 +1,36 @@
 package com.ruoyi.production.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
-import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import com.ruoyi.framework.web.domain.AjaxResult;
-import com.ruoyi.production.dto.ProductBomDto;
-import com.ruoyi.production.dto.ProductProcessDto;
-import com.ruoyi.production.pojo.ProductBom;
-import com.ruoyi.production.mapper.ProductBomMapper;
-import com.ruoyi.production.pojo.ProductProcess;
-import com.ruoyi.production.service.ProductBomService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import org.springframework.beans.BeanUtils;
+import com.ruoyi.basic.pojo.Product;
+import com.ruoyi.basic.pojo.ProductModel;
+import com.ruoyi.basic.service.IProductModelService;
+import com.ruoyi.basic.service.IProductService;
+import com.ruoyi.common.exception.ServiceException;
+import com.ruoyi.common.utils.StringUtils;
+import com.ruoyi.common.utils.poi.ExcelUtil;
+import com.ruoyi.framework.web.domain.AjaxResult;
+import com.ruoyi.production.dto.BomImportDto;
+import com.ruoyi.production.dto.ProductBomDto;
+import com.ruoyi.production.dto.ProductStructureDto;
+import com.ruoyi.production.mapper.ProductBomMapper;
+import com.ruoyi.production.pojo.ProductBom;
+import com.ruoyi.production.pojo.ProductProcess;
+import com.ruoyi.production.pojo.ProductStructure;
+import com.ruoyi.production.service.ProductBomService;
+import com.ruoyi.production.service.ProductProcessService;
+import com.ruoyi.production.service.ProductStructureService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.multipart.MultipartFile;
+
+import javax.servlet.http.HttpServletResponse;
+import java.math.BigDecimal;
+import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -27,23 +44,275 @@
 public class ProductBomServiceImpl extends ServiceImpl<ProductBomMapper, ProductBom> implements ProductBomService {
 
     @Autowired
+    private IProductService productService;
+
+    @Autowired
     private ProductBomMapper productBomMapper;
+
+
+    @Autowired
+    private IProductModelService productModelService;
+
+    @Autowired
+    private ProductStructureService productStructureService;
+
+    @Autowired
+    private ProductProcessService productProcessService;
 
     @Override
     public IPage<ProductBomDto> listPage(Page page, ProductBomDto productBomDto) {
-        return productBomMapper.listPage(page,productBomDto);
+        return productBomMapper.listPage(page, productBomDto);
     }
 
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public AjaxResult add(ProductBom productBom) {
         boolean save = productBomMapper.insert(productBom) > 0;
         if (save) {
-            // 鏍规嵁id鐢熸垚no瀛楁锛欸X + 8浣嶆暟瀛楋紙涓嶈冻8浣嶅墠闈㈣ˉ0锛�
             String no = "BM." + String.format("%05d", productBom.getId());
             productBom.setBomNo(no);
             productBomMapper.updateById(productBom);
+
+            //  鏌ヨ鍑轰骇鍝佹ā鍨嬩俊鎭�
+            if (productBom.getProductModelId() == null) {
+                throw new ServiceException("璇烽�夋嫨浜у搧妯″瀷");
+            }
+
+            ProductModel productModel = productModelService.getById(productBom.getProductModelId());
+            if (productModel == null) {
+                throw new ServiceException("閫夋嫨鐨勪骇鍝佹ā鍨嬩笉瀛樺湪");
+            }
+
+            //  娣诲姞鍒濆鐨勪骇鍝佺粨鏋�
+            ProductStructure productStructure = new ProductStructure();
+            productStructure.setProductModelId(productBom.getProductModelId());
+            productStructure.setUnit(productModel.getUnit());
+            productStructure.setUnitQuantity(BigDecimal.valueOf(1));
+            productStructure.setBomId(productBom.getId());
+
+            productStructureService.save(productStructure);
+
             return AjaxResult.success();
         }
         return AjaxResult.error();
     }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public AjaxResult uploadBom(MultipartFile file) {
+        ExcelUtil<BomImportDto> util = new ExcelUtil<>(BomImportDto.class);
+        List<BomImportDto> list;
+        try {
+            list = util.importExcel(file.getInputStream());
+        } catch (Exception e) {
+            return AjaxResult.error("Excel瑙f瀽澶辫触");
+        }
+
+        if (list == null || list.isEmpty()) return AjaxResult.error("鏁版嵁涓虹┖");
+
+        //  澶勭悊宸ュ簭
+        list.forEach(dto -> {
+            dto.setParentName(clean(dto.getParentName()));
+            dto.setParentSpec(clean(dto.getParentSpec()));
+            dto.setChildName(clean(dto.getChildName()));
+            dto.setChildSpec(clean(dto.getChildSpec()));
+        });
+        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());
+        }
+
+        return AjaxResult.success("BOM瀵煎叆鎴愬姛");
+    }
+
+
+    @Override
+    public void exportBom(HttpServletResponse response, Integer bomId) {
+        if (bomId == null) {
+            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());
+                }
+            }
+        }
+
+        ExcelUtil<BomImportDto> util = new ExcelUtil<>(BomImportDto.class);
+        util.exportExcel(response, exportList, "BOM缁撴瀯瀵煎嚭");
+    }
+
+    private void populateMap(List<ProductStructureDto> nodes, Map<Long, ProductStructureDto> map) {
+        if (nodes == null || nodes.isEmpty()) {
+            return;
+        }
+        for (ProductStructureDto node : nodes) {
+            map.put(node.getId(), node);
+            populateMap(node.getChildren(), map);
+        }
+    }
+
+    private ProductModel findModel(String name, String spec) {
+        Product product = productService.getOne(new LambdaQueryWrapper<Product>()
+                .eq(Product::getProductName, name).last("limit 1"));
+        if (product == null) throw new ServiceException("浜у搧鏈淮鎶わ細" + name);
+
+        ProductModel model = productModelService.getOne(new LambdaQueryWrapper<ProductModel>()
+                .eq(ProductModel::getProductId, product.getId())
+                .eq(ProductModel::getModel, spec).last("limit 1"));
+        if (model == null) throw new ServiceException("瑙勬牸鏈淮鎶わ細" + name + "[" + spec + "]");
+        return model;
+    }
+
+    private void handleProcess(List<BomImportDto> list) {
+
+        Set<String> processNames = list.stream()
+                .map(BomImportDto::getProcess)
+                .filter(StringUtils::isNotBlank)
+                .collect(Collectors.toSet());
+
+        if (processNames.isEmpty()) {
+            return;
+        }
+
+        List<ProductProcess> exists = productProcessService.list(
+                new LambdaQueryWrapper<ProductProcess>().in(ProductProcess::getName, processNames)
+        );
+
+        Set<String> existNames = exists.stream()
+                .map(ProductProcess::getName)
+                .collect(Collectors.toSet());
+
+        List<ProductProcess> needSave = processNames.stream()
+                .filter(n -> !existNames.contains(n))
+                .map(n -> {
+                    ProductProcess p = new ProductProcess();
+                    p.setName(n);
+                    return p;
+                })
+                .collect(Collectors.toList());
+
+        if (!needSave.isEmpty()) {
+            productProcessService.saveBatch(needSave);
+            needSave.forEach(p -> p.setNo("GX" + String.format("%08d", p.getId())));
+            productProcessService.updateBatchById(needSave);
+        }
+    }
+
+    private String clean(String s) {
+        if (s == null) return null;
+        return s.replaceAll("[\\u00A0\\u3000]", "").trim();
+    }
 }
+
+

--
Gitblit v1.9.3