zhongrj
2024-05-29 a5fac95408a43ad43de9d88c30d06c0918c7bc8f
src/main/java/org/springblade/common/utils/NodeTreeUtil.java
@@ -4,12 +4,8 @@
import org.springblade.common.node.TreeLongNode;
import org.springblade.common.node.TreeStringNode;
import org.springblade.core.tool.node.TreeNode;
import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
import org.springblade.modules.house.vo.HouseTree;
import org.springblade.modules.system.node.DeptUserTreeNode;
import java.util.*;
import java.util.stream.Collectors;
/**
 * node tree 工具类
@@ -77,44 +73,6 @@
               }
            });
         }
      }
      return tree;
   }
   /**
    * 树转换
    * @param treeMap
    * @return
    */
   public static List<DoorplateAddressVOTree> getAddressNodeTree(Map<String, DoorplateAddressVOTree> treeMap){
      List<DoorplateAddressVOTree> tree = new ArrayList<>();
      if (treeMap.size() > 1) {
         treeMap.forEach((code, treeNode) -> {
            if (treeMap.containsKey(treeNode.getParentCode())) {
               treeMap.get(treeNode.getParentCode()).getChildren().add(treeNode);
            } else {
               tree.add(treeNode);
            }
         });
      }
      return tree;
   }
   /**
    * 树转换
    * @param treeMap
    * @return
    */
   public static List<HouseTree> getHouseTree(Map<String, HouseTree> treeMap){
      List<HouseTree> tree = new ArrayList<>();
      if (treeMap.size() > 1) {
         treeMap.forEach((code, treeNode) -> {
            if (treeMap.containsKey(treeNode.getParentCode())) {
               treeMap.get(treeNode.getParentCode()).getChildren().add(treeNode);
            } else {
               tree.add(treeNode);
            }
         });
      }
      return tree;
   }