From 49e8cd7c6642a5da92172edc70cb914142e6c0ad Mon Sep 17 00:00:00 2001
From: chenhj <1263187585@qq.com>
Date: 星期五, 24 四月 2026 11:13:24 +0800
Subject: [PATCH] 文件工具类变更
---
src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java | 481 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 473 insertions(+), 8 deletions(-)
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 3c6cfca..2a4cd1d 100644
--- a/src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java
+++ b/src/main/java/com/ruoyi/technology/service/impl/TechnologyBomServiceImpl.java
@@ -1,20 +1,485 @@
package com.ruoyi.technology.service.impl;
+import cn.hutool.core.collection.CollUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+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.bean.BeanUtils;
+import com.ruoyi.common.utils.poi.ExcelUtil;
+import com.ruoyi.framework.web.domain.R;
+import com.ruoyi.production.bean.dto.BomImportDto;
+import com.ruoyi.production.bean.dto.ProductStructureDto;
+import com.ruoyi.technology.bean.dto.TechnologyBomDto;
+import com.ruoyi.technology.bean.dto.TechnologyBomStructureDto;
+import com.ruoyi.technology.bean.vo.TechnologyBomStructureVo;
+import com.ruoyi.technology.bean.vo.TechnologyBomVo;
import com.ruoyi.technology.mapper.TechnologyBomMapper;
+import com.ruoyi.technology.mapper.TechnologyBomStructureMapper;
+import com.ruoyi.technology.mapper.TechnologyRoutingMapper;
import com.ruoyi.technology.pojo.TechnologyBom;
+import com.ruoyi.technology.pojo.TechnologyBomStructure;
+import com.ruoyi.technology.pojo.TechnologyRouting;
import com.ruoyi.technology.service.TechnologyBomService;
+import com.ruoyi.technology.service.TechnologyBomStructureService;
+import jakarta.servlet.http.HttpServletResponse;
+import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.multipart.MultipartFile;
-/**
- * <p>
- * BOM琛� 鏈嶅姟瀹炵幇绫�
- * </p>
- *
- * @author 鑺杞欢锛堟睙鑻忥級鏈夐檺鍏徃
- * @since 2026-04-20 10:05:55
- */
+import java.math.BigDecimal;
+import java.util.*;
+import java.util.concurrent.ThreadLocalRandom;
+import java.util.stream.Collectors;
+
@Service
+@RequiredArgsConstructor
public class TechnologyBomServiceImpl extends ServiceImpl<TechnologyBomMapper, TechnologyBom> implements TechnologyBomService {
+ private final TechnologyBomMapper technologyBomMapper;
+ private final IProductModelService productModelService;
+ private final TechnologyBomStructureService technologyBomStructureService;
+ private final TechnologyRoutingMapper technologyRoutingMapper;
+ private final IProductService productService;
+ private final TechnologyBomStructureMapper technologyBomStructureMapper;
+
+ /**
+ * 鍒嗛〉鏌ヨBOM鍒楄〃銆�
+ */
+ @Override
+ public IPage<TechnologyBomVo> listPage(Page<TechnologyBomDto> page, TechnologyBomDto technologyBomDto) {
+ return technologyBomMapper.listPage(page, technologyBomDto);
+ }
+
+ /**
+ * 鏍规嵁瑙勬牸鏌ヨBOM骞惰浆鎹负杩斿洖瀵硅薄銆�
+ */
+ @Override
+ public List<TechnologyBomVo> listByModel(Long productModelId) {
+ List<TechnologyBom> list = this.list(Wrappers.<TechnologyBom>lambdaQuery()
+ .eq(TechnologyBom::getProductModelId, productModelId));
+ List<TechnologyBomVo> result = new ArrayList<>(list.size());
+ for (TechnologyBom item : list) {
+ TechnologyBomVo vo = new TechnologyBomVo();
+ vo.setId(item.getId());
+ vo.setProductModelId(item.getProductModelId());
+ vo.setRemark(item.getRemark());
+ vo.setVersion(item.getVersion());
+ vo.setCreateTime(item.getCreateTime());
+ vo.setUpdateTime(item.getUpdateTime());
+ vo.setCreateUser(item.getCreateUser());
+ vo.setUpdateUser(item.getUpdateUser());
+ vo.setBomNo(item.getBomNo());
+ vo.setDeptId(item.getDeptId());
+ result.add(vo);
+ }
+ return result;
+ }
+
+ /**
+ * 鏂板BOM骞跺垵濮嬪寲鏍圭粨鏋勮妭鐐广��
+ */
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public R add(TechnologyBom technologyBom) {
+ validateProductModel(technologyBom.getProductModelId());
+ boolean saved = technologyBomMapper.insert(technologyBom) > 0;
+ if (!saved) {
+ return R.fail("Add BOM failed");
+ }
+ technologyBom.setBomNo("BM." + String.format("%05d", technologyBom.getId()));
+ technologyBomMapper.updateById(technologyBom);
+ initRootStructure(technologyBom.getId().longValue(), technologyBom.getProductModelId());
+ return R.ok();
+ }
+
+ /**
+ * 淇敼BOM锛岃鏍煎彉鍖栨椂鍚屾鍒锋柊鍏宠仈缁撴瀯涓庤矾绾挎暟鎹��
+ */
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public R update(TechnologyBom technologyBom) {
+ if (technologyBom.getId() == null) {
+ throw new ServiceException("BOM id is required");
+ }
+ validateProductModel(technologyBom.getProductModelId());
+ TechnologyBom oldBom = technologyBomMapper.selectById(technologyBom.getId());
+ if (oldBom == null) {
+ throw new ServiceException("BOM not found");
+ }
+ if (oldBom.getProductModelId() != null && !oldBom.getProductModelId().equals(technologyBom.getProductModelId())) {
+ technologyRoutingMapper.updateProductModelByBomId(technologyBom.getProductModelId(), technologyBom.getId().longValue());
+ technologyBomStructureService.remove(Wrappers.<TechnologyBomStructure>lambdaQuery()
+ .eq(TechnologyBomStructure::getBomId, technologyBom.getId().longValue()));
+ initRootStructure(technologyBom.getId().longValue(), technologyBom.getProductModelId());
+ }
+ if (technologyBom.getBomNo() == null) {
+ technologyBom.setBomNo(oldBom.getBomNo());
+ }
+ technologyBomMapper.updateById(technologyBom);
+ return R.ok();
+ }
+
+ /**
+ * 鍒犻櫎BOM鍓嶆牎楠屾槸鍚﹀凡琚伐鑹鸿矾绾垮紩鐢ㄣ��
+ */
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public boolean batchDelete(List<Long> ids) {
+ if (ids == null || ids.isEmpty()) {
+ throw new ServiceException("Select at least one BOM");
+ }
+ List<TechnologyRouting> list = technologyRoutingMapper.selectList(Wrappers.<TechnologyRouting>lambdaQuery()
+ .in(TechnologyRouting::getBomId, ids));
+ if (!list.isEmpty()) {
+ throw new ServiceException("BOM is referenced by routing");
+ }
+ technologyBomStructureService.remove(Wrappers.<TechnologyBomStructure>lambdaQuery()
+ .in(TechnologyBomStructure::getBomId, ids));
+ return this.removeBatchByIds(ids);
+ }
+
+ /**
+ * 鏍￠獙浜у搧瑙勬牸鏄惁瀛樺湪銆�
+ */
+ private void validateProductModel(Long productModelId) {
+ if (productModelId == null) {
+ throw new ServiceException("Product model is required");
+ }
+ ProductModel productModel = productModelService.getById(productModelId);
+ if (productModel == null) {
+ throw new ServiceException("Product model not found");
+ }
+ }
+
+ /**
+ * 鍒濆鍖朆OM鏍硅妭鐐圭粨鏋勩��
+ */
+ private void initRootStructure(Long bomId, Long productModelId) {
+ ProductModel productModel = productModelService.getById(productModelId);
+ TechnologyBomStructure root = new TechnologyBomStructure();
+ root.setBomId(bomId);
+ root.setParentId(null);
+ root.setProductModelId(productModelId);
+ root.setUnit(productModel.getUnit());
+ root.setUnitQuantity(BigDecimal.ONE);
+ technologyBomStructureService.save(root);
+ }
+
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public R uploadBom(MultipartFile file) {
+ ExcelUtil<BomImportDto> util = new ExcelUtil<>(BomImportDto.class);
+ List<BomImportDto> list;
+ try {
+ list = util.importExcel(file.getInputStream());
+ } catch (Exception e) {
+ return R.fail("Excel瑙ｆ瀽澶辫触");
+ }
+
+ if (list == null || list.isEmpty()) return R.fail("鏁版嵁涓虹┖");
+
+ // 澶勭悊宸ュ簭
+ 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());
+ TechnologyBom bom = new TechnologyBom();
+ bom.setProductModelId(rootModel.getId());
+ bom.setVersion("1.0");
+ technologyBomMapper.insert(bom);
+ bom.setBomNo("BM." + String.format("%05d", bom.getId()));
+ technologyBomMapper.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())) {
+ TechnologyBomStructure rootNode = new TechnologyBomStructure();
+ rootNode.setBomId(bom.getId().longValue());
+ rootNode.setParentId(null); // 椤跺眰娌℃湁鐖惰妭鐐�
+ rootNode.setProductModelId(rootModel.getId());
+ rootNode.setUnitQuantity(BigDecimal.ONE);
+ rootNode.setUnit(rootModel.getUnit());
+ technologyBomStructureService.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());
+
+ // 鎻掑叆缁撴瀯琛�
+ TechnologyBomStructure node = new TechnologyBomStructure();
+ node.setBomId(bom.getId().longValue());
+ 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()));
+// }
+ technologyBomStructureService.save(node);
+
+ // 鎶婂綋鍓嶅瓙椤硅褰曞埌 Map,浣滀负浠ュ悗鏇存繁灞傜骇鐨勭埗椤规煡鎵句緷鎹�
+ // 鍚屼竴鐖堕」涓嬬殑鍚屽悕瀛愰」涓嶉渶瑕侀噸澶嶈褰�
+ treePathMap.put(childKey, node.getId());
+ }
+
+ return R.ok("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缁撴瀯瀵煎嚭");
+ }
+
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public R copy(TechnologyBom technologyBom) {
+ TechnologyBom oldTechnologyBom = technologyBomMapper.selectById(technologyBom.getId());
+ List<TechnologyBomStructureVo> oldTechnologyBomStructureVos = technologyBomStructureService.listByBomId(technologyBom.getId().longValue());
+ //鏍￠獙浜у搧瑙勬牸鏄惁瀛樺湪銆�
+ validateProductModel(oldTechnologyBom.getProductModelId());
+ TechnologyBom newTechnologyBom = new TechnologyBom();
+ newTechnologyBom.setProductModelId(oldTechnologyBom.getProductModelId());
+ newTechnologyBom.setVersion("FZ" + oldTechnologyBom.getVersion());
+ newTechnologyBom.setRemark(oldTechnologyBom.getRemark());
+ boolean saved = technologyBomMapper.insert(newTechnologyBom) > 0;
+ if (!saved) {
+ return R.fail("Copy BOM failed");
+ }
+ newTechnologyBom.setBomNo("BM." + String.format("%05d", newTechnologyBom.getId()));
+ technologyBomMapper.updateById(newTechnologyBom);
+ //鍒濆鍖朆OM鏍硅妭鐐圭粨鏋勩��
+ initRootStructure(newTechnologyBom.getId().longValue(), newTechnologyBom.getProductModelId());
+ //鎶婁骇鍝佺粨鏋勯噷闈㈢殑鏁版嵁涔熷叏閮ㄩ兘澶嶅埗
+ TechnologyBomStructureVo technologyBomStructureVo = oldTechnologyBomStructureVos.get(0);
+ TechnologyBomStructureDto technologyBomStructureDto = convertTree(technologyBomStructureVo);
+ technologyBomStructureDto.setBomId(newTechnologyBom.getId().longValue());
+ technologyBomStructureService.addTechnologyBomStructure(technologyBomStructureDto);
+ return R.ok();
+ }
+
+ 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();
+ }
+
+ 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);
+ }
+ }
+
+ /**
+ * 閫掑綊杞崲鏍戝舰缁撴瀯 VO -> DTO
+ * 鑷姩鐢熸垚铏氭嫙 tempId / parentTempId锛屼繚璇佷换鎰忓眰绾ф爲缁撴瀯姝ｇ‘
+ */
+ public static TechnologyBomStructureDto convertTree(TechnologyBomStructureVo vo) {
+ if (vo == null) {
+ return null;
+ }
+ TechnologyBomStructureDto realDto = convertNode(vo, "0"); // 鏍硅妭鐐圭埗ID=0锛堢函鏁板瓧锛�
+ TechnologyBomStructureDto rootDto = new TechnologyBomStructureDto();
+ rootDto.setTempId("0");
+ rootDto.setChildren(Collections.singletonList(realDto));
+
+ return rootDto;
+ }
+
+ /**
+ * 鏍稿績閫掑綊鏂规硶
+ * @param vo 鍘熷鑺傜偣
+ * @param parentTempId 鐖惰妭鐐� 绾暟瀛桰D
+ * @return 杞崲鍚嶥TO
+ */
+ private static TechnologyBomStructureDto convertNode(TechnologyBomStructureVo vo, String parentTempId) {
+ if (vo == null) {
+ return null;
+ }
+
+ TechnologyBomStructureDto dto = new TechnologyBomStructureDto();
+ BeanUtils.copyProperties(vo, dto);
+
+ String currentTempId = getNumberId();
+ dto.setTempId(currentTempId);
+ dto.setParentTempId(parentTempId);
+
+
+ dto.setId(null);
+ dto.setParentId(null);
+
+ // ===================== 閫掑綊瀛愯妭鐐� =====================
+ List<TechnologyBomStructureVo> voChildren = vo.getChildren();
+ if (CollUtil.isNotEmpty(voChildren)) {
+ List<TechnologyBomStructureDto> dtoChildren = new ArrayList<>();
+ for (TechnologyBomStructureVo childVo : voChildren) {
+ // 瀛愯妭鐐圭殑鐖禝D = 褰撳墠鑺傜偣鐨勬暟瀛桰D
+ dtoChildren.add(convertNode(childVo, currentTempId));
+ }
+ dto.setChildren(dtoChildren);
+ } else {
+ dto.setChildren(new ArrayList<>());
+ }
+
+ return dto;
+ }
+
+ /**
+ * 鐢熸垚 13浣� 绾暟瀛楅殢鏈篒D锛堝畨鍏ㄣ�佷笉閲嶅銆侀珮鎬ц兘锛�
+ */
+ private static String getNumberId() {
+ // 鐢熸垚 1000000000000 ~ 9999999999999 涔嬮棿鐨勬暟瀛�
+ long min = 1000000000000L;
+ long max = 9999999999999L;
+ long randomNum = ThreadLocalRandom.current().nextLong(min, max + 1);
+ return String.valueOf(randomNum);
+ }
}
--
Gitblit v1.9.3