二级菜单 多级菜单 递归获取

二级菜单的获取

private List<Menu> markMenuTree(List<Menu> menuTree) {

        //保存树状菜单数据
        List<Menu> list=new ArrayList<>();
        //创建map集合暂时存储菜单数据,方便整合多级菜单
        Map<Integer,Menu> map=new HashMap<>();
        for (Menu menu : menuTree) {
            //key为主键  value为对象数据
            map.put(menu.getId(),menu);
            //没有父级id即为1级菜单
            if (menu.getParentId()==null){
                //把数据加入到1级菜单中
                list.add(menu);
            }
        }
        for (Menu menu : menuTree) {
            if (menu.getType()==1){
                //为子菜单  并且map有他的数据
                if (menu.getParentId()!=null&&map.containsKey(menu.getParentId())){
                    //使用父id来获取父级菜单对象
                    map.get(menu.getParentId()).getChildren().add(menu);
                }
            }
        }
        return list;
    }

实体类

import lombok.Data;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

@Data
public class Menu implements Serializable {
    private Integer id;
    private String name;
    private Integer parentId;
    private String url;
    private String icon;
    private String perms;
    private Short type;
    private Integer sort;
    private static final long serialVersionUID = 1L;
    //处理多级菜单   new出来后会有内存地址唯一
    private List<Menu> children=new ArrayList<>();

    public Menu(Integer id, String name, Integer parentId, String url, String icon, String perms, Short type, Integer sort) {
        this.id = id;
        this.name = name;
        this.parentId = parentId;
        this.url = url;
        this.icon = icon;
        this.perms = perms;
        this.type = type;
        this.sort = sort;
    }
}

多级菜单的获取

@Override
 public List<Menu> menuTree() {
   List<Menu> menus=menuMapper.getAllMenu();
    // 保存父级节点数据
    List<Menu> firstList=new ArrayList<>();
    Map<Integer,Menu> map=new HashMap<>();
    //循环
    for (Menu menu : menus) {
        map.put(menu.getId(),menu);
        //父节点
        if (menu.getParentId()==null){
            firstList.add(menu);
        }
    }
    //循环子节点
    for (Menu menu : menus) {
        if (menu.getParentId()!=null&&map.containsKey(menu.getParentId())){
            map.get(menu.getParentId()).getChildren().add(menu);
        }
    }
    return firstList;
}

递归获取

public TableResult getList() {
        //找所有数据
        List<Menu> allMenu = menuMapper.getAllMenu();
        //先找到父级目录
        //保存父级数据的集合
        List<Menu> firstList = new ArrayList<>();
        //使用map帮助我们简单修改数据
        Map<Integer, Menu> map = new HashMap<>();
        //找父级目录
        for (Menu menu : allMenu) {
            // 把数据保存到map中,利用内存地址传递来修改数据
            map.put(menu.getId(), menu);
            //找父级目录,放到父级数据的集合中
            if (menu.getParentId() == null) {
                //保存数据到父级集合
                firstList.add(menu);
            }
        }

        //找子级目录
        for (Menu menu : allMenu) {
            if (menu.getParentId() != null && map.containsKey(menu.getParentId())) {
                //找到父级
                map.get(menu.getParentId()).getChildren().add(menu);
            }
        }
        //firstList保存了父级目录数据,父级目录数据中有子级目录的数据getChildren(list集合)
        //target把整理好顺序的数据存放到target集合当中,返回展示
        //使用引用数据类型数据的内存地址,来进行数据修改
        List<Menu> target = new ArrayList<>();
        //调用递归方法进行处理排序
        getAllMenuTree(firstList, target);
        return TableResult.getResultBySuccess((long) target.size(), target);
    }

    private void getAllMenuTree(List<Menu> firstList, List<Menu> target) {
        for (Menu menu : firstList) {
            //先把父级目录加入目标集合当中保存
            target.add(menu);
            //如果子级集合长度大于0,就代表还有子级目录存在
            if (menu.getChildren().size() > 0) {
                //进行递归循环处理
                getAllMenuTree(menu.getChildren(), target);
            }
        }
    }