java 培训 Stream 遍历树形结构案例分享
可能平常会遇到一些需求，比如构建菜单，构建树形结构，数据库一般就使用父 id 来表示，为了降低数据库的查询压力，我们可以使用 Java8 中的 Stream 流一次性把数据查出来，然后通过流式处理，我们一起来看看，代码实现为了实现简单，就模拟查看数据库所有数据到 List 里面_java培训。
实体类：Menu.java
/**
* Menu
*
* @author lcry
*/
@Data
@Builder
public class Menu {
/**
* id
*/
public Integer id;
/**
* 名称
*/
public String name;
/**
* 父 id ，根节点为 0
*/
public Integer parentId;
/**
* 子节点信息
*/
public List<Menu> childList;
public Menu(Integer id, String name, Integer parentId) {
this.id = id;
this.name = name;
this.parentId = parentId;
}
public Menu(Integer id, String name, Integer parentId, List<Menu> childList) {
this.id = id;
this.name = name;
this.parentId = parentId;
this.childList = childList;
}
}
递归组装树形结构：
@Test
public void testtree(){
//模拟从数据库查询出来，公众号 Java 精选，有惊喜！
List<Menu> menus = Arrays.asList(
new Menu(1,"根节点",0),
new Menu(2,"子节点 1",1),
new Menu(3,"子节点 1.1",2),
new Menu(4,"子节点 1.2",2),
new Menu(5,"根节点 1.3",2),
new Menu(6,"根节点 2",1),
new Menu(7,"根节点 2.1",6),
new Menu(8,"根节点 2.2",6),
new Menu(9,"根节点 2.2.1",7),
new Menu(10,"根节点 2.2.2",7),
new Menu(11,"根节点 3",1),
new Menu(12,"根节点 3.1",11)
);
//获取父节点
List<Menu> collect = menus.stream().filter(m -> m.getParentId() == 0).map(
(m) -> {
m.setChildList(getChildrens(m, menus));
return m;
}
).collect(Collectors.toList());
System.out.println("-------转 json 输出结果-------");
System.out.println(JSON.toJSON(collect));
}
/**
* 递归查询子节点
* @param root 根节点
* @param all 所有节点
* @return 根节点信息
*/
private List<Menu> getChildrens(Menu root, List<Menu> all) {
List<Menu> children = all.stream().filter(m -> {
return Objects.equals(m.getParentId(), root.getId());
}).map(
(m) -> {
m.setChildList(getChildrens(m, all));
return m;
}
).collect(Collectors.toList());
return children;
}
格式化打印结果：
文章转载来源于 Java 编程
关注尚硅谷，轻松学IT 2021.11.23 加入
IT培训 www.atguigu.com
评论