公安安全监管

linwei

2024-01-08 ad2ec97fe0c60eee4bb7c8cdacff7c876f280ea7

 src/main/java/org/springblade/common/utils/NodeTreeUtil.java

@@ -1,7 +1,8 @@
package org.springblade.common.utils;

import org.springblade.common.node.TreeIntegerNode;
import org.springblade.common.node.TreeNode;
import org.springblade.common.node.TreeLongNode;
import org.springblade.common.node.TreeStringNode;
import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
import org.springblade.modules.house.vo.HouseTree;

@@ -40,6 +41,50 @@
    * @param treeMap
    * @return
    */
   public static List<TreeLongNode> getLongNodeTree(Map<Long, TreeLongNode> treeMap){
      List<TreeLongNode> tree = new ArrayList<>();
      if (treeMap.size() > 1) {
         treeMap.forEach((id, treeNode) -> {
            if (treeMap.containsKey(treeNode.getParentId())) {
               treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
            } else {
               tree.add(treeNode);
            }
         });
      }
      return tree;
   }

   /**
    * 树转换
    * @param treeMap
    * @return
    */
   public static List<TreeStringNode> getStringNodeTree(Map<String, TreeStringNode> treeMap){
      List<TreeStringNode> tree = new ArrayList<>();
      if (treeMap.size() > 0) {
         if (treeMap.size()==1){
            treeMap.forEach((id, treeNode) -> {
               tree.add(treeNode);
            });
         }else {
            treeMap.forEach((id, treeNode) -> {
               if (treeMap.containsKey(treeNode.getParentId())) {
                  treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
               } else {
                  tree.add(treeNode);
               }
            });
         }
      }
      return tree;
   }

   /**
    * 树转换
    * @param treeMap
    * @return
    */
   public static List<DoorplateAddressVOTree> getAddressNodeTree(Map<String, DoorplateAddressVOTree> treeMap){
      List<DoorplateAddressVOTree> tree = new ArrayList<>();
      if (treeMap.size() > 1) {