java 两种方式生成递归菜单,支持树断裂情况下生成新的树

1、菜单类

@Data
public class Menu {
    private Long id;
    private String name;
    private Long pid;
    private List<Menu> children;

    public Menu(Long id, String name, Long pid) {
        this.id = id;
        this.name = name;
        this.pid = pid;
    }
}

2、父id确定情况下生成新的树

public static void test1(){
        List<Menu> menuList=new ArrayList<>();
        menuList.add(new Menu(1L,"菜单1",0L));
        menuList.add(new Menu(2L,"菜单2",1L));
        menuList.add(new Menu(3L,"菜单3",2L));
        menuList.add(new Menu(4L,"菜单4",5L));

        List<Menu> menus=menuList.stream().filter(m->m.getPid()==0).map(m->{
            m.setChildren(getChildrens(m,menuList));
            return m;
        }).collect(Collectors.toList());
        System.out.println(JSONUtil.toJsonPrettyStr(menus));
    }

3、递归生成子树 

private static List<Menu> getChildrens(Menu root, List<Menu> all) {
        List<Menu> children = all.stream().filter(m -> Objects.equals(m.getPid(), root.getId())).map(
                (m) -> {
                    m.setChildren(getChildrens(m, all));
                    return m;
                }
        ).collect(Collectors.toList());
        return children;
    }

4、有查询条件树断裂情况下根据id和pid生成新的树

public static void test2(){
        List<Menu> menuList=new ArrayList<>();
        menuList.add(new Menu(1L,"菜单1",0L));
        menuList.add(new Menu(2L,"菜单2",1L));
        menuList.add(new Menu(3L,"菜单3",2L));
        menuList.add(new Menu(4L,"菜单4",5L));

        List<Long> IdList=menuList.stream().map(s->s.getId()).collect(Collectors.toList());
        List<Long> PidList=menuList.stream().map(s->s.getPid()).collect(Collectors.toList());
        /**
         * 求两者的交集
         */
        List<Long> jiaoJiIdList=new ArrayList<>();
        for(Long id:IdList){
            if(PidList.contains(id)){
                jiaoJiIdList.add(id);
            }
        }

        /**
         * 过滤时所有pid在交集里面的数据,若pid在交集中则说明该元素有父类
         */
        List<Menu> menus=menuList.stream().filter(m->!jiaoJiIdList.contains(m.getPid())).map(m->{
            m.setChildren(getChildrens(m,menuList));
            return m;
        }).collect(Collectors.toList());
        System.out.println(JSONUtil.toJsonPrettyStr(menus));
    }

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值