Java树形结构数据最简单的遍历方式——JDK8stream流

于 2021-07-20 10:11:54 发布

阅读量2k

点赞数

CC 4.0 BY-SA版权

分类专栏： Java初体验文章标签： java 集合 stream

本文链接：https://blog.youkuaiyun.com/zhangyongbaba/article/details/118928193

Java初体验专栏收录该内容

10 篇文章

订阅专栏

这篇博客介绍了如何利用Java Stream API对List<StudentResponse>集合进行过滤和分组，根据父级编码将学生响应对象进行归类。通过创建一个Map，以父级编码为键，对应的子菜单列表为值，然后遍历原始列表，将子菜单添加到对应父级编码的对象中。最后，筛选出父级编码为空的元素作为顶级菜单，完成整个数据结构的构建。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：使用stream流对list集合进行分组，得到一个key为父级编码，value为list的一个map，然后遍历list，用编码和map中的key作比较，如果相同则加入到自己的子菜单下。

最后留下父级编码为空的数据就完成了遍历。

List<StudentResponse> dataList;
Map<String, List<StudentResponse>> collect = dataList.stream()
    .filter(Objects::nonNull)
    .filter(o -> StringUtils.isNotBlank(o.getParentCode()))
    .collect(Collectors.groupingBy(StudentResponse::getParentCode));
for (StudentResponse data : dataList) {
    String code = data.getCode();
    List<StudentResponse> list = collect.get(code);
    data.setChildren(list);
}
dataList = dataList.stream()
    .filter(e -> StringUtils.isBlank(e.getParentCode()))
    .collect(Collectors.toList());