使用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对菜单数据进行处理,首先筛选出顶级菜单项,并通过递归方式为其添加子菜单。最后,得到了一个完整的树形菜单结构。

相关推荐
繁依Fanyi1 分钟前
简易安卓句分器实现
java·服务器·开发语言·算法·eclipse
慧都小妮子12 分钟前
Spire.PDF for .NET【页面设置】演示:打开 PDF 时自动显示书签或缩略图
java·pdf·.net
m512717 分钟前
LinuxC语言
java·服务器·前端
IU宝21 分钟前
C/C++内存管理
java·c语言·c++
湫ccc21 分钟前
《Python基础》之pip换国内镜像源
开发语言·python·pip
瓜牛_gn22 分钟前
依赖注入注解
java·后端·spring
fhvyxyci22 分钟前
【C++之STL】摸清 string 的模拟实现(下)
开发语言·c++·string
hakesashou23 分钟前
Python中常用的函数介绍
java·网络·python
qq_4597300325 分钟前
C 语言面向对象
c语言·开发语言
佚先森32 分钟前
2024ARM网络验证 支持一键云注入引流弹窗注册机 一键脱壳APP加固搭建程序源码及教程
java·html