java treenode_java之TreeUtils生成一切对象树形结构案例
赞
踩
项目中经常会遇到各种需要以树形结构展示的功能，比较常见的，如菜单树，分类树，部门树等等，如果为每种类型都遍历递归生成树形结构返回给前端，显得有些冗余且麻烦，并且其实逻辑都是一致的，只是遍历的对象不同而已，故其实可以通过面向接口思维，来实现这种通用工具类的实现。
TreeNode用来表示每个树节点的抽象，即需要生成树的对象需要实现此接口。
/**
* 树节点父类，所有需要使用{@linkplain TreeUtils}工具类形成树形结构等操作的节点都需要实现该接口
*
* @param 节点id类型
*/
public interface TreeNode {
/**
* 获取节点id
*
* @return 树节点id
*/
T id();
/**
* 获取该节点的父节点id
*
* @return 父节点id
*/
T parentId();
/**
* 是否是根节点
*
* @return true：根节点
*/
boolean root();
/**
* 设置节点的子节点列表
*
* @param children 子节点
*/
void setChildren(List extends TreeNode> children);
/**
* 获取所有子节点
*
* @return 子节点列表
*/
List extends TreeNode> getChildren();
}
TreeUtils用来生成树形结构，以及获取所有叶子节点等操作
/**
* 树形结构工具类
*
* @author meilin.huang
* @version 1.0
* @date 2019-08-24 1:57 下午
*/
public class TreeUtils {
/**
* 根据所有树节点列表，生成含有所有树形结构的列表
*
* @param nodes 树形节点列表
* @param 节点类型
* @return 树形结构列表
*/
public static > List generateTrees(List nodes) {
List roots = new ArrayList<>();
for (Iterator ite = nodes.iterator(); ite.hasNext(); ) {
T node = ite.next();
if (node.root()) {
roots.add(node);
// 从所有节点列表中删除该节点，以免后续重复遍历该节点
ite.remove();
}
}
roots.forEach(r -> {
setChildren(r, nodes);
});
return roots;
}
/**
* 从所有节点列表
