锋盈数科-知识库 Logo
首页
软件开发
计算机基础
Hello Halo
新手必读
关于本知识库
登录 →
锋盈数科-知识库 Logo
首页 软件开发 计算机基础 Hello Halo 新手必读 关于本知识库
登录
  1. 首页
  2. 软件开发
  3. JAVA
  4. Java Stream流轻松遍历树形结构

Java Stream流轻松遍历树形结构

0
  • JAVA
  • 发布于 2024-09-26
  • 0 次阅读
黄健
黄健

在日常开发中,经常会遇到返回树形及遍历树形结构的需求,比如构建菜单,项目-任务树形结构,角色权限树形结构等。数据库一般使用parent_id来表示上一层级,遍历的时候一般使用递归或多层for循环,增加了数据库的查询压力。

为了降低数据库的查询压力,我们可以使用Java8中的Stream流一次性把数据查出来,然后通过流式处理。

话不多说,直接上Demo。

首先,创建一个Menu 菜单的实体类

@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;
    }
}

接下来,模拟生成一颗树形结构List

集合。

private static List<Menu> createTree() {
        //模拟创建一棵树
        return  = Arrays.asList(
                new Menu(1, "目录", 0),
                new Menu(2, "第一章节", 1),
                new Menu(3, "1.1 大战光明顶", 2),
                new Menu(4, "1.2 偶遇赵敏", 2),
                new Menu(5, "1.3 迎娶周芷若", 2),
                new Menu(6, "第二章节", 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, "第三章节", 1),
                new Menu(12, "3.1 刺杀董卓", 11),
                new Menu(13, "3.1.1 貂蝉献艺", 12),
                new Menu(14, "结束", 0)
        );

    }

然后,使用根据上级 parentId 获取所有子节点的方法,这里我们可以使用递归的方式去实现

private static List<Menu> getAllChildrenByRoot(Menu root, List<Menu> all) {
        return all.stream().filter(m -> Objects.equals(m.getParentId(), root.getId()))
                .peek(
                        (m) -> m.setChildList(getAllChildrenByRoot(m, all))
                ).collect(Collectors.toList());
    }

最后,使用main方法验证一下

public static void main(String[] args) {
        List<Menu> menus = createTree();

        List<Menu> menuList = menus.stream().filter(m -> m.getParentId() == 0)
                .peek(
                (m) -> m.setChildList(getAllChildrenByRoot(m, menus))
        ).collect(Collectors.toList());
        System.out.println("/*****collect to json*****/");
        System.out.println(JSONObject.toJSON(menuList));
    }

打印一下结果

[
    {
        "name":"目录",
        "childList":[
            {
                "name":"第一章节",
                "childList":[
                    {
                        "name":"1.1 大战光明顶",
                        "childList":[

                        ],
                        "id":3,
                        "parentId":2
                    },
                    {
                        "name":"1.2 偶遇赵敏",
                        "childList":[

                        ],
                        "id":4,
                        "parentId":2
                    },
                    {
                        "name":"1.3 迎娶周芷若",
                        "childList":[

                        ],
                        "id":5,
                        "parentId":2
                    }
                ],
                "id":2,
                "parentId":1
            },
            {
                "name":"第二章节",
                "childList":[
                    {
                        "name":"2.1 挥泪斩狮王",
                        "childList":[
                            {
                                "name":"2.2.1 败走麦城",
                                "childList":[

                                ],
                                "id":9,
                                "parentId":7
                            },
                            {
                                "name":"2.2.2 火烧赤壁",
                                "childList":[

                                ],
                                "id":10,
                                "parentId":7
                            }
                        ],
                        "id":7,
                        "parentId":6
                    },
                    {
                        "name":"2.2 大意失街亭",
                        "childList":[

                        ],
                        "id":8,
                        "parentId":6
                    }
                ],
                "id":6,
                "parentId":1
            },
            {
                "name":"第三章节",
                "childList":[
                    {
                        "name":"3.1 刺杀董卓",
                        "childList":[
                            {
                                "name":"3.1.1 貂蝉献艺",
                                "childList":[

                                ],
                                "id":13,
                                "parentId":12
                            }
                        ],
                        "id":12,
                        "parentId":11
                    }
                ],
                "id":11,
                "parentId":1
            }
        ],
        "id":1,
        "parentId":0
    },
    {
        "name":"结束",
        "childList":[

        ],
        "id":14,
        "parentId":0
    }
]


原文链接: https://onlyou.blog.csdn.net//article/details/135277469

标签: #JAVA 991
相关文章

Spring 实现 3 种异步接口 2024-10-18 09:07

大家好,我是苏三~ 如何处理比较耗时的接口? 这题我熟,直接上异步接口,使用 Callable、WebAsyncTask 和 DeferredResult、CompletableFuture等均可实现。 但这些方法有局限性,处理结果仅返回单个值。在某些场景下,如果需要接口异步处理的同时,还持续不断地

重学SpringBoot3-集成Redis(五)之布隆过滤器 2024-10-08 11:24

更多SpringBoot3内容请关注我的专栏:《SpringBoot3》 期待您的点赞👍收藏⭐评论✍ 重学SpringBoot3-集成Redis(五)之布隆过滤器 1. 什么是布隆过滤器? * 基本概念 适用场景 2. 使用 Redis 实现布隆过滤器 * 项目依赖 Redis 配置

SpringBoot整合异步任务执行 2024-10-08 11:24

同步任务: 同步任务是在单线程中按顺序执行,每次只有一个任务在执行,不会引发线程安全和数据一致性等 并发问题 同步任务需要等待任务执行完成后才能执行下一个任务,无法同时处理多个任务,响应慢,影响用 户体验 异步任务: 异步任务是在多线程中同时执行,多个任务可以并发执行,同时处理多个请求,响应快,资源

springboot kafka多数据源,通过配置动态加载发送者和消费者 2024-10-08 11:24

前言 最近做项目,需要支持kafka多数据源,实际上我们也可以通过代码固定写死多套kafka集群逻辑,但是如果需要不修改代码扩展呢,因为kafka本身不处理额外逻辑,只是起到削峰,和数据的传递,那么就需要对架构做一定的设计了。 准备test kafka本身非常容易上手,如果我们需要单元测试,引入ja

SpringBoot 集成 Redis 2024-10-08 11:24

一:SpringBoot 集成 Redis ①Redis是一个 NoSQL(not only)数据库, 常作用缓存 Cache 使用。 ②Redis是一个中间件、是一个独立的服务器;常用的数据类型: string , hash ,set ,zset , list ③通过Redis客户端可以使用多种语

SpringBoot整合QQ邮箱 2024-10-08 11:24

SpringBoot可以通过导入依赖的方式集成多种技术,这当然少不了我们常用的邮箱,现在本章演示SpringBoot整合QQ邮箱发送邮件…. 下面按步骤进行: 1.获取QQ邮箱授权码 1.1 登录QQ邮箱 1.2 开启SMTP服务 找到下图中的SMTP服务区域,如果当前账号未开启的话自己手动开启。

目录

IT 外包服务商

  • 意见投递
  • zyf6619

软件开发应用

主菜单

  • 首页
  • 软件开发
  • 计算机基础
  • Hello Halo
  • 新手必读
  • 关于本知识库
Copyright © 2024 your company All Rights Reserved. Powered by Halo.