树节点父类:
package com.yw.framework.tree; import lombok.Data; import java.util.ArrayList; import java.util.LinkedList; import java.util.List; @Data public class Node { protected byte isCategory; protected String id; protected String elementId; protected String parentId; protected String name; protected List<Node> children = new ArrayList<>(); protected String description; protected int rank = 0; public Node addChild(Node node) { if (children == null) { this.children = new LinkedList<Node>(); } children.add(node); return this; } }
工具类:
package com.yw.framework.tree; import com.yw.module.core.element.base.BaseElement; import com.yw.module.synthesize.type.ElementTypeEnum; import java.util.ArrayList; import java.util.List; import java.util.Objects; import java.util.Optional; import java.util.stream.Collectors; /** * @Author zl * @Description 树形构造处理工具类 * * @Date 2021/6/1 9:12 */ public class BuildTree { /** * 功能描述: <生成树结构> * 〈〉 * * @Param: [T] * @Author: zl * @Date: 2021/6/1 9:10 */ public static List<Node> buildTree(List<? extends Node> nodeList) { List<Node> nodes = new ArrayList<>(); if (nodeList!=null){ for (Node node : nodeList) { String parentId = node.getParentId(); if (parentId == null || "".equals(parentId)) { nodes.add(node); continue; } for (Node parent : nodeList) { if (parent.getId().equals(parentId)) { List<Node> parentList = parent.getChildren(); if (parentList == null) { parentList = new ArrayList<>(); parentList.add(node); parent.setChildren(parentList); } else { parentList.add(node); } } } } } return nodes; } /** * 给树节点设置层级rank标识 * * @param nodes * @Author zl */ public static void setLevel(List<Node> nodes) { if (nodes!=null){ for (Node node : nodes) { String parentId = node.getParentId(); if (parentId == null || "".equals(parentId)) { node.setRank(1); dfsLevel(node); } } } } /** * 递归遍历节点数 * @param node */ public static void dfsLevel(Node node) { List<Node> children = node.getChildren(); if (children == null || children.isEmpty()) { return; } for (Node child : children) { child.setRank(node.getRank() + 1); dfsLevel(child); } } /** * 获取树的最大层级 * * @param nodes * @return */ public static int maxLevel(List<Node> nodes,int maxlevel) { if (nodes==null){ return 0; } for (Node node : nodes) { maxlevel = Math.max(node.getRank(), maxlevel); if (node.getChildren()!=null) { return maxLevel(node.getChildren(), maxlevel); } } return maxlevel; } /** * 获取树的某一层级所有节点 * * @param nodes * @param level * @return */ public static List<Node> getLevelNode(List<Node> nodes, int level, List<Node> nodeList) { if (nodes==null){ return nodeList; } for (Node node : nodes) { if (node.getRank() == level) { nodeList.add(node); } if (node.getChildren()!=null) { getLevelNode(node.getChildren(), level, nodeList); } } return nodeList; } /** * 获取树的所有叶子节点 * * @param nodes * @return */ public static List<Node> getLeafNode(List<? extends Node> nodes, List<Node> leafNode) { if (nodes != null) { for (Node node : nodes) { if (node.getChildren() == null || node.getChildren().size() == 0) { leafNode.add(node); } else { getLeafNode(node.getChildren(), leafNode); } } } return leafNode; } /** * 删除给定的叶子节点 * @param nodes * @param ids * @param deleteSize * @Author zl * @return */ public static int deleteLeafNode(List<Node> nodes, String[] ids, int deleteSize) { boolean b = nodes.removeIf(node -> { if ((node.getParentId() == null || "".equals(node.getParentId())) && (node.getChildren() == null || node.getChildren().size() == 0)) { boolean isDelete = true; for (int i = 0; i < ids.length; i++) { if (ids[i].equals(node.getId())) { isDelete = false; break; } } return isDelete; } return false; }); if (b) { deleteSize++; } for (Node node : nodes) { if (node.getChildren() == null || node.getChildren().size() == 0) { boolean isDelete = true; for (int i = 0; i < ids.length; i++) { if (ids[i].equals(node.getId())) { isDelete = false; } } if (isDelete) { deleteSize++; for (Node parent : nodes) { if (parent.getId().equals(node.getParentId())) { parent.getChildren().remove(node); break; } } } } else { return deleteLeafNode(node.getChildren(), ids, deleteSize); } } return deleteSize; } /** * 获取某个父节点下面的所有子节点id * 并过滤要素 (isCategory为0) * @param baseElements * @param id * @Author zl * @return */ public static List<String> getTreeElementSonEntity(List<? extends BaseElement> baseElements, String id, List<String> ids) { for (BaseElement baseElement : baseElements) { //遍历出父id等于参数的id,add进子节点集合 if (Objects.equals(id, baseElement.getParentId())) { //符合的具体要素 if (baseElement.getIsCategory() == 0){ ids.add(baseElement.getId()); } //递归遍历下一级 getTreeElementSonEntity(baseElements, baseElement.getId(), ids); } } return ids; } /** * 获取某个要素节点所有父节点(分类) * * @param elements * @param id * @Author zl * @return */ public static<T extends BaseElement> List<T> getTreeElementParentEntity(List<T> elements, String id, List<T> newList) { //找到父节点 Optional<T> first = elements.stream().filter( e -> Objects.equals(e.getId(), id) ).findFirst(); //已经找到最顶层 if (!first.isPresent()){ return newList; } T t = first.get(); newList.add(t); //递归遍历父级 return getTreeElementParentEntity(elements, t.getParentId(), newList); } }