JAVA--递归遍历树级菜单结构(多层嵌套菜单)

3,151 阅读1分钟

数据库存储的菜单结果: 在这里插入图片描述 parentid为0的都是根节点,也就是一级菜单,后面的子菜单的parentid为父菜单的ID。





MenumDTO类(菜单类)的结构:

public class MenumDTO {
    private Integer id;

    private String content;

    private Integer parentid;

    private Date createtime;

    private Integer num;

    private List<MenumDTO> childs;

  	//get和set方法省略
}

采用递归方法,遍历成树级结构菜单。

//获得树级结构菜单
 public List<MenumDTO> getMenumList() throws IOException {
        //拿到菜单的所有数据
        List<MenumDTO> list=menumMapper.getMenumList();
        //存储根节点的菜单,即一级菜单
        List<MenumDTO> rootlist=new ArrayList<>();
        //遍历所有数据,找到根节点菜单
        for (MenumDTO menumDTO:
             list) {
            if(menumDTO.getParentid().equals(0)){
                //找到根节点菜单的时候,寻找这个根节点菜单下的子节点菜单。
                findChilds(menumDTO,list);
                //添加到根节点的列表中
                rootlist.add(menumDTO);
            }
        }
        return rootlist;
    }

    private void findChilds(MenumDTO root,List<MenumDTO> list){
        List<MenumDTO> childlist=new ArrayList<>();
        //遍历所有数据,找到是入参父节点的子节点的数据,然后加到childlist集合中。
        for (MenumDTO menum :
                list) {
            if (root.getId().equals(menum.getParentid()))
                childlist.add(menum);
        }
        //若子节点不存在,那么就不必再遍历子节点中的子节点了 直接返回。
        if(childlist.size()==0)
            return;
        //设置父节点的子节点列表
        root.setChilds(childlist);
        //若子节点存在,接着递归调用该方法,寻找子节点的子节点。
        for (MenumDTO childs :
                childlist) {
            findChilds(childs, list);
        }
    }

结果:

在这里插入图片描述


总结: 先拿到所有的菜单数据,然后遍历菜单数据,找到根节点,找到根节点。然后调用找到子节点的方法,在子节点方法中递归调用自己,也就是说如果这个节点有子节点,那么递归调用方法找到子节点的子节点,直到某个节点下面没有子节点。