使用Java流API构建树形结构数据

简介:

在实际开发中,构建树状层次结构是常见需求,如组织架构、目录结构或菜单系统。本教案通过解析给定的Java代码,展示如何使用Java 8 Stream API将扁平化的菜单数据转换为具有层级关系的树形结构。

java 复制代码
@Data
@NoArgsConstructor
@AllArgsConstructor
static class Menu {
    private Long id;
    private String name;
    private Long parentId;
    private List<Menu> children;
}

提示:此处定义了一个名为Menu的类,它包含了菜单项的基本属性,包括ID、名称、父菜单ID以及一个用于存储子菜单项的列表。

2. main方法解析及实现功能

输入参数:

  • List<Menu> menus:包含所有菜单项的集合。

输出结果:

  • List<Menu>:一个仅包含顶级菜单项的列表,每个顶级菜单项已填充了其下级子菜单。
java 复制代码
public static void main(String[] args) {
        List<Menu> menus = menusData();
        /**
         * 从菜单列表中筛选出顶级菜单,并为其添加子菜单。
         *
         * @param menus 菜单列表,包含所有菜单项。
         * @return 包含所有顶级菜单的列表,其中每个顶级菜单均已包含其所有子菜单。
         */
        List<Menu> topLevelMenus = menus.stream() // 使用流处理menus集合
                        .filter(menu -> menu.getParentId() == 0 || menus.stream().noneMatch(other -> other.getId().equals(menu.getParentId()))) // 筛选条件:父菜单ID为0或不存在对应父菜单的菜单项
                        .peek(menu -> menu.setChildren(getChildren(menu, menus))) // 为每个顶级菜单设置子菜单
                        .collect(Collectors.toList()); // 将筛选后的顶级菜单集合转换为List(Menu)类型
    }

3. 辅助方法------获取指定菜单的所有子菜单

java 复制代码
 /**
     * 获取指定菜单的所有子菜单。
     *
     * @param menu 指定的菜单对象,我们要查找它的子菜单。
     * @param menus 所有菜单的列表,从中筛选出子菜单。
     * @return 返回一个包含指定菜单所有子菜单的列表。这个列表中的每个菜单对象都可能包含它们自己的子菜单列表。
     */
    private static List<Menu> getChildren(Menu menu, List<Menu> menus) {
        // 使用流对菜单列表进行处理,筛选出指定菜单的子菜单
        return menus.stream()
                .filter(child -> child.getParentId().equals(menu.getId())) // 筛选条件:菜单的父菜单ID与指定菜单ID匹配
                .peek(child -> child.setChildren(getChildren(child, menus))) // 递归设置每个子菜单的子菜单列表
                .collect(Collectors.toList()); // 收集结果,生成列表
    }
java 复制代码
private static List<Menu> menusData() {
    return Arrays.asList(
            new Menu(1L, "一级菜单1", 0L, null),
            new Menu(2L, "二级菜单1", 1L, null),
            new Menu(3L, "三级菜单1", 2L, null),
            new Menu(4L, "一级菜单2", 0L, null),
            new Menu(5L, "二级菜单2", 4L, null),
            new Menu(6L, "一级菜单3", 0L, null)
    );
}

5.完整代码,以及演示(TreeExample.java)

java 复制代码
package com.tenement.auto;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class TreeExample {

    @Data
    @NoArgsConstructor
    @AllArgsConstructor
    static
    class Menu {
        private Long id;
        private String name;
        private Long parentId;

        private List<Menu> children;
    }

    public static void main(String[] args) {
        List<Menu> menus = menusData();
        /**
         * 从菜单列表中筛选出顶级菜单,并为其添加子菜单。
         *
         * @param menus 菜单列表,包含所有菜单项。
         * @return 包含所有顶级菜单的列表,其中每个顶级菜单均已包含其所有子菜单。
         */
        List<Menu> topLevelMenus = menus.stream() // 使用流处理menus集合
                        .filter(menu -> menu.getParentId() == 0 || menus.stream().noneMatch(other -> other.getId().equals(menu.getParentId()))) // 筛选条件:父菜单ID为0或不存在对应父菜单的菜单项
                        .peek(menu -> menu.setChildren(getChildren(menu, menus))) // 为每个顶级菜单设置子菜单
                        .collect(Collectors.toList()); // 将筛选后的顶级菜单集合转换为List(Menu)类型
    }

    /**
     * 获取指定菜单的所有子菜单。
     *
     * @param menu 指定的菜单对象,我们要查找它的子菜单。
     * @param menus 所有菜单的列表,从中筛选出子菜单。
     * @return 返回一个包含指定菜单所有子菜单的列表。这个列表中的每个菜单对象都可能包含它们自己的子菜单列表。
     */
    private static List<Menu> getChildren(Menu menu, List<Menu> menus) {
        // 使用流对菜单列表进行处理,筛选出指定菜单的子菜单
        return menus.stream()
                .filter(child -> child.getParentId().equals(menu.getId())) // 筛选条件:菜单的父菜单ID与指定菜单ID匹配
                .peek(child -> child.setChildren(getChildren(child, menus))) // 递归设置每个子菜单的子菜单列表
                .collect(Collectors.toList()); // 收集结果,生成列表
    }

    private static List<Menu> menusData() {
        return Arrays.asList(
                new Menu(1L, "一级菜单1", 0L,null),
                new Menu(2L, "二级菜单1", 1L,null),
                new Menu(3L, "三级菜单1", 2L,null),
                new Menu(4L, "一级菜单2", 0L,null),
                new Menu(5L, "二级菜单2", 4L,null),
                new Menu(6L, "一级菜单3", 0L,null)
        );
    }

}

总结:该案例展示了如何利用Java 的Stream API对菜单数据进行处理,首先筛选出顶级菜单项,并通过递归方式为其添加子菜单。最后,得到了一个完整的树形菜单结构。

相关推荐
李永奉5 小时前
C语言-嵌入式开发C语言常用预处理指令合集
c语言·开发语言
Patrick在香港5 小时前
Python Docker镜像从1.2GB到89MB:多阶段构建的完整优化实录
java·python·docker·信息可视化·容器·数据分析·ai编程
上海安当技术5 小时前
敏感数据怎么防拖库?信封加密(DEK+KEK 二层密钥)架构设计与 Java 实战
java·开发语言·python
我是唐青枫5 小时前
Java JCommander 实战详解:用注解解析命令行参数
java·开发语言
大鹏说大话6 小时前
C++ 内存布局详解:类、虚函数、虚表底层原理
java·开发语言
键盘会跳舞6 小时前
C++:容器适配器 queue 的源码级拆解
开发语言·c++·queue
还是鼠鼠6 小时前
Spring AI连接DeepSeek与通义千问:application.yaml配置详解
java·通义千问·spring ai·deepseek
计算机小白一个7 小时前
蓝桥杯 Java B 组之哈希表应用(两数之和、重复元素判断)
java·数据结构·算法·蓝桥杯
Loge编程生活7 小时前
打卡信奥刷题(3449)用C++实现信奥题 P10429 [蓝桥杯 2024 省 B] 拔河
开发语言·数据结构·c++·算法·青少年编程
在世修行8 小时前
从零打造 C# 工业视觉检测系统(八):Modbus RTU 协议解析与 PLC 剔除控制实战
开发语言·c#