Java8新特性Stream——优雅的递归遍历树形结构, 递归删除节点

copy战士

已于 2022-04-10 10:06:50 修改

阅读量8.9k

点赞数 5

分类专栏： Java 文章标签： java 后端

于 2021-04-02 15:29:46 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_45996984/article/details/111762134

版权

Java 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了如何在Java中处理树形结构数据，包括构建层级菜单、递归遍历以及根据ID删除节点及其所有子节点。示例中使用了Lombok注解简化代码，并展示了结合MybatisPlus框架进行递归删除的实现。

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

日常开发中会碰需要树形结构数据,层级菜单显示,或者是根据指定id删除该id和其包含的所有子节点,下面简单分享一下,如有错误,还请大佬指点

菜单实体类

/**
 * <p>菜单实体类</p>
 * @author : Zyy
 * @date : 2020-12-26 17:41
 **/
 
@Data //lombok注解,自动生成set和get方法
public class Menu {

    //id
    private String id;

    //菜单名
    private String name;

    //父节点id
    private String pid;

    //子节点信息
    private List<Menu> children;

	//无参构造
    public Menu() {}
    
    //带参构造
    public Menu(String id, String name, String pid) {
        this.id = id;
        this.name = name;
        this.pid= pid;
    }

}

递归遍历树形结构

    public void getTreeMenu() {
        //模拟数据库查出所有数据
        List<Menu> menus = Arrays.asList(
                new Menu(1, "根节点", 0),
                new Menu(2, "子节点1", 1),
                new Menu(3, "子子节点1.1", 2),
                new Menu(6, "子子子节点1.1.1", 3),
                new Menu(4, "子节点2", 1),
                new Menu(5, "子子节点2.1", 4)
        );

        //筛选出根节点
        List<Menu> list = menus.stream().filter(menu -> "0".equals(menu.getPid())).peek(
            //设置子节点信息
            menu -> menu.setChildren(getChildrens(menu, menus))
        ).collect(Collectors.toList());
    }
	
	private List<Menu> getChildrens(Menu root, List<Menu> menus) {
     	List<Menu> list = menus.stream().filter(menu ->
     		//筛选出下一节点元素
   			Objects.equals(menu.getPid(), root.getId())).map(menu -> {
   				//递归set子节点
				menu.setChildren(this.getChildrens(menu, menus));
                return menu;
          	}).collect(Collectors.toList());	
        return list;
    }

递归删除节点及其子节点(结合MybatisPlus框架)

	public void deleteMenus(String id) {
		//存储所有要删除的id
		List<String> idList = new ArrayList<>();
		//获得所有需要删除的id
		this.selectChildListById(id, idList);
		//添加删除的节点id
		idList.add(id);
		//MybatisPlus批量删除方法
		baseMapper.deleteBatchIds(idList);
	}
	
	private String selectChildListById(String id, List<String> idList) {
		//查询数据库pid等于id的数据
		List<Menu> childList = baseMapper.selectList(new QueryWrapper<Menu>().eq("pid", id).select("id"));
		//递归查询下一级id,同时将上一次查询结果添加到list集合
        childList.forEach(menu-> {
            idList.add(menu.getId());
            this.selectChildListById(menu.getId(), idList);
        });
	}