Java8新特性-使用Stream流来实现递归遍历树形结构（案例）

Java8新特性-Stream流

可能平常会遇到一些需求，比如构建菜单，构建树形结构，数据库一般就使用父id来表示，为了降低数据库的查询压力，我们可以使用Java8中的Stream流一次性把数据查出来，然后通过流式处理。

我们一起来看看，代码实现为了实现简单，就模拟查看数据库所有数据到List里面。

实体类：Menu.java

package com.example.demo.model;

import lombok.Builder;
import lombok.Data;

import java.util.List;

/**
 * 树结构：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;
    }

}

递归组装树形结构：

package com.example.demo;

import com.alibaba.fastjson.JSON;
import com.example.demo.model.Menu;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

@SpringBootTest
class DemoApplicationTests {
    
    @Test
    void contextLoads() {
        //模拟从数据库查询出来
        List<Menu> menus = Arrays.asList(
                new Menu(1,"商品管理",0),
                new Menu(2,"会员管理",0),
                new Menu(3,"系统管理",0),
                new Menu(4,"系统工具",0),
                new Menu(5,"用户管理",3),
                new Menu(6,"角色管理",3),
                new Menu(7,"菜单栏管理",3),
                new Menu(8,"日志管理",4),
                new Menu(9,"商品列表",1),
                new Menu(10,"订单列表",1),
                new Menu(11,"会员等级",2),
                new Menu(12,"品牌列表",1)
        );

        //获取父节点
        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;
    }

}

格式化打印结果：

[{
	"name": "商品管理",
	"childList": [{
		"name": "商品列表",
		"childList": [],
		"id": 9,
		"parentId": 1
	}, {
		"name": "订单列表",
		"childList": [],
		"id": 10,
		"parentId": 1
	}, {
		"name": "品牌列表",
		"childList": [],
		"id": 12,
		"parentId": 1
	}],
	"id": 1,
	"parentId": 0
}, {
	"name": "会员管理",
	"childList": [{
		"name": "会员等级",
		"childList": [],
		"id": 11,
		"parentId": 2
	}],
	"id": 2,
	"parentId": 0
}, {
	"name": "系统管理",
	"childList": [{
		"name": "用户管理",
		"childList": [],
		"id": 5,
		"parentId": 3
	}, {
		"name": "角色管理",
		"childList": [],
		"id": 6,
		"parentId": 3
	}, {
		"name": "菜单栏管理",
		"childList": [],
		"id": 7,
		"parentId": 3
	}],
	"id": 3,
	"parentId": 0
}, {
	"name": "系统工具",
	"childList": [{
		"name": "日志管理",
		"childList": [],
		"id": 8,
		"parentId": 4
	}],
	"id": 4,
	"parentId": 0
}]

总结

Stream 是 Java8 中处理集合的关键抽象概念，它可以指定你希望对集合进行的操作，可以执行非常复杂的查找、过滤和映射数据等操作。使用Stream API 对集合数据进行操作，就类似于使用 SQL 执行的数据库查询。也可以使用 Stream API 来并行执行操作。简而言之，Stream API 提供了一种高效且易于使用的处理数据的方式。