消防系统---保护对象接口优化

amos-boot-module/amos-boot-module-biz/amos-boot-module-equip-biz/src/main/java/com/yeejoin/equipmanage/service/impl/BuildingServiceImpl.java

@@ -710,25 +710,23 @@ public class BuildingServiceImpl extends ServiceImpl<BuildingMapper, Building> i
 
     @Override
     public List<PointTreeVo> getBuildTreeNot(String bizOrgCode) {
-        List<PointTreeVo> responses = this.baseMapper.getBuildList(bizOrgCode,null);
-        List<PointTreeVo> treeList = new ArrayList<>();
+        List<PointTreeVo> responses = this.baseMapper.getBuildList(bizOrgCode, null);
+        Map<Long, PointTreeVo> nodeMap = responses.parallelStream().collect(Collectors.toMap(PointTreeVo::getSequenceNbr, t -> t));
+        List<PointTreeVo> treeList = responses.parallelStream()
+                .filter(node -> node.getParentId() == null || node.getParentId() == -1)
+                .map(node -> buildTree(node, nodeMap))
+                .collect(Collectors.toList());
+        return treeList;
+    }
 
-        for (PointTreeVo tree : responses) {
-            if (tree.getParentId() == null || tree.getParentId() == -1) {
-                treeList.add(tree);
-            }
-            for (PointTreeVo treeNode : responses) {
-                if (tree.getSequenceNbr().equals(treeNode.getParentId())) {
-                    if (tree.getChildren() == null) {
-                        tree.setChildren(new ArrayList<>());
-                    }
-                    tree.getChildren().add(treeNode);
-                }
-            }
+    private PointTreeVo buildTree(PointTreeVo node, Map<Long, PointTreeVo> nodeMap) {
+        if (node.getChildren() == null) {
+            node.setChildren(new ArrayList<>());
         }
-
-
-        return treeList;
+        nodeMap.values().parallelStream()
+                .filter(childNode -> childNode.getParentId().equals(node.getSequenceNbr()))
+                .forEach(childNode -> node.getChildren().add(buildTree(childNode, nodeMap)));
+        return node;
     }
 
     @Override