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;

}

/**

* 从所有节点列表