优化结构树查询接口

qomo-kn-cost-manage/src/main/java/com/zzsmart/qomo/kn/cost/manage/service/impl/BomTreeServiceImpl.java

 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
-import java.util.*;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.TreeMap;
 import java.util.stream.Collectors;
 
 @Service
+@Transactional
 public class BomTreeServiceImpl implements BomTreeService {
     @Autowired
     CostStandardDetailMapper costStandardDetailMapper;
             // 版本号
             wrapper.like("c2.version_number", version_number);
         }
-        List<CostStandardDetailVO> costStandardDetailVOS = costStandardDetailMapper.queryListResult(wrapper);
         //查询所有满足条件的父节点
+        List<CostStandardDetailVO> parentNodeList = costStandardDetailMapper.queryListResult(wrapper);
+        //所有子类节点
+        List<CostStandardDetailVO> childrenNodeList = new ArrayList<>();
         Map<String, CostStandardDetailVO> parentDataMap = new TreeMap<>();
-        if ((StrUtil.isNotEmpty(materialNumber) || StrUtil.isNotEmpty(version_number)) && costStandardDetailVOS.size() > 0) {
-            parentDataMap = costStandardDetailVOS.stream().collect(Collectors.toMap(CostStandardDetailVO::getMaterialNumber, vo -> vo));
-            //查询所有父节点下的子节点
-            QueryWrapper<CostStandardDetail> wrapper1 = new QueryWrapper<>();
-            for (String materialNumber1 : parentDataMap.keySet()) {
-                wrapper1.like("parent_material_number", materialNumber1);
+        if (parentNodeList.size() > 0) {
+            //父类结果只有1个
+            if (parentNodeList.size() == 1) {
+                parentDataMap = parentNodeList.stream().collect(Collectors.toMap(CostStandardDetailVO::getMaterialNumber, vo -> vo));
+            } else {
+                //父类结果有多个（找出最上级节点）
+                Map<Integer, List<CostStandardDetailVO>> levelGroupMap = parentNodeList.stream().collect(Collectors.groupingBy(CostStandardDetailVO::getLevel));
+                TreeMap<Integer, List<CostStandardDetailVO>> levelTreeMap = new TreeMap<>(levelGroupMap);
+                parentDataMap = levelTreeMap.firstEntry().getValue().stream().collect(Collectors.toMap(CostStandardDetailVO::getMaterialNumber, vo -> vo));
+            }
+            //查询所有父节点下的子节点(即第二层叶子节点)
+            List<CostStandardDetail> costStandardDetails = new ArrayList<>();
+            if (parentDataMap.size() > 0) {
+                QueryWrapper<CostStandardDetail> wrapper1 = new QueryWrapper<>();
+                int num = 0;
+                for (String materialNumber1 : parentDataMap.keySet()) {
+                    num = num + 1;
+                    if (num > 1) {
+                        wrapper1.or().eq("parent_material_number", materialNumber1);
+                    } else {
+                        wrapper1.eq("parent_material_number", materialNumber1);
+                    }
+                }
+                costStandardDetails = costStandardDetailMapper.selectList(wrapper1);
             }
-            List<CostStandardDetail> costStandardDetails = costStandardDetailMapper.selectList(wrapper1);
             //第三层叶子节点
-            List<CostStandardDetail> threeNodes=new ArrayList<>();
-            if(costStandardDetails!=null&&costStandardDetails.size()>0){
-               List<String> materialNoList = costStandardDetails.stream().map(CostStandardDetail::getMaterialNumber).collect(Collectors.toList());
-               QueryWrapper<CostStandardDetail> wrapper2 = new QueryWrapper<>();
-                   wrapper2.in("parent_material_number", materialNoList);
-
-               threeNodes = costStandardDetailMapper.selectList(wrapper2);
-           }
+            List<CostStandardDetail> threeNodes = new ArrayList<>();
+            if (costStandardDetails != null && costStandardDetails.size() > 0) {
+                List<String> materialNoList = costStandardDetails.stream().map(CostStandardDetail::getMaterialNumber).collect(Collectors.toList());
+                QueryWrapper<CostStandardDetail> wrapper2 = new QueryWrapper<>();
+                wrapper2.in("parent_material_number", materialNoList);
+                threeNodes = costStandardDetailMapper.selectList(wrapper2);
+            }
             //把第二层叶子节点转化为bomtreeinfoVO对象集合
-            costStandardDetailVOS=new ArrayList<>();
-            for (int i=0;i<costStandardDetails.size();i++){
+            for (int i = 0; i < costStandardDetails.size(); i++) {
                 CostStandardDetail costStandardDetail = costStandardDetails.get(i);
-                CostStandardDetailVO vo=new CostStandardDetailVO();
-                BeanUtils.copyProperties(costStandardDetail,vo);
-                costStandardDetailVOS.add(vo);
+                CostStandardDetailVO vo = new CostStandardDetailVO();
+                BeanUtils.copyProperties(costStandardDetail, vo);
+                childrenNodeList.add(vo);
             }
             //把第三层叶子节点转化为bomtreeinfoVO对象集合
-            for (int i=0;i<threeNodes.size();i++){
+            for (int i = 0; i < threeNodes.size(); i++) {
                 CostStandardDetail costStandardDetail = threeNodes.get(i);
-                CostStandardDetailVO vo=new CostStandardDetailVO();
-                BeanUtils.copyProperties(costStandardDetail,vo);
-                costStandardDetailVOS.add(vo);
+                CostStandardDetailVO vo = new CostStandardDetailVO();
+                BeanUtils.copyProperties(costStandardDetail, vo);
+                childrenNodeList.add(vo);
             }
         }
-        List<BomTreeInfoVO> bomTreeInfoVOS = BomTreeBuilder.buildBomTree(costStandardDetailVOS, parentDataMap);
+        List<BomTreeInfoVO> bomTreeInfoVOS = BomTreeBuilder.buildBomTree(childrenNodeList, parentDataMap);
         return bomTreeInfoVOS;
     }
 }

qomo-kn-cost-manage/src/main/java/com/zzsmart/qomo/kn/cost/manage/vo/CostStandardDetailSearch.java

 
 import io.swagger.annotations.ApiModel;
 import lombok.AllArgsConstructor;
-import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
 import java.util.List;
 
 @Data
-@Builder
 @NoArgsConstructor
 @AllArgsConstructor
-@ApiModel(value="bom结构树", description="bom结构树")
+@ApiModel(value = "bom结构树", description = "bom结构树")
 public class CostStandardDetailSearch extends CostStandardDetailVO {
     /**
      * 选择行的id