java构建树形菜单递归工具类

1.设计菜单实体

import java.util.List;

public class Menu {
    //菜单id
    private Long id;
    //父节点id
    private Long parentId;
    //菜单名称
    private String name;
    //对应路径
    private String url;
    //图标
    private String icon;
    //是否选中
    private boolean selected;
    
    private List<Menu> childrens;

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public Long getParentId() {
        return parentId;
    }

    public void setParentId(Long parentId) {
        this.parentId = parentId;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getUrl() {
        return url;
    }

    public void setUrl(String url) {
        this.url = url;
    }

    public String getIcon() {
        return icon;
    }

    public void setIcon(String icon) {
        this.icon = icon;
    }

    public boolean isSelected() {
        return selected;
    }

    public void setSelected(boolean selected) {
        this.selected = selected;
    }

    public List<Menu> getChildrens() {
        return childrens;
    }

    public void setChildrens(List<Menu> childrens) {
        this.childrens = childrens;
    }
    
    
    

}

2.工具类如下：

import java.util.ArrayList;
import java.util.List;

import com.alibaba.fastjson.JSONObject;
/**
 * 构建树形菜单
 * MenuUtils
 * @author lishang
 * 2018年8月30日
 */
public class MenuUtils {
    /**
     * 针对二级菜单
     * treeMenuList:( ). <br/> 
     * @author lishang 
     * @param sourceList
     * @return
     */
    public static List<Menu> treeMenuList(List<Menu> sourceList){
        List<Menu> targetList=new ArrayList<>();
        if (sourceList==null) {
            return null;
        }
        List<Menu> pmenus=new ArrayList<>();
        for (Menu menu : sourceList) {
            if(menu.getParentId()==0l||menu==null){
                pmenus.add(menu);
            }
        }
        sourceList.removeAll(pmenus);
        for (int i = 0; i < pmenus.size(); i++) {
            Menu pmenu=pmenus.get(i);
            List<Menu> cmenus=new ArrayList<>();
            for (Menu menu : sourceList) {
                if(pmenu.getId().equals(menu.getParentId())){
                    cmenus.add(menu);
                }
            }
            pmenu.setChildrens(cmenus);
            sourceList.removeAll(cmenus);
        }
        return targetList;
    }
    
    public static List<Menu> treeRoot(List<Menu> sourceList){
        
        
        return sourceList;
    }
    
    
    /**
     * 递归获取菜单
     * treeRoot:( ). <br/> 
     * @author lishang 
     * @param sourceList
     * @param rootMenu
     * @return
     */
    public static Menu treeRoot(List<Menu> sourceList,Menu rootMenu)
    {
        if (sourceList == null)
        {
            return null;
        }
        List<Menu> childList=new ArrayList<>();
        for (Menu menu : sourceList) {
            if(rootMenu.getId().equals(menu.getParentId())){
                Menu menuChild = treeRoot(sourceList, menu);
                childList.add(menuChild);
            }
        }
        if(childList.size()==0){
            return rootMenu;
        }
        rootMenu.setChildrens(childList);
        return rootMenu;
    }
    
    
    public static void main(String[] args) {
        List<Menu> sourceList=new ArrayList<>();
        
        Menu menu=new Menu();
        menu.setParentId(0l);
        menu.setId(1l);
        menu.setName("菜单一级");
        sourceList.add(menu);
        
        Menu menu2=new Menu();
        menu2.setParentId(1l);
        menu2.setId(2l);
        menu2.setName("菜单二级1");
        sourceList.add(menu2);
        
        Menu menu3=new Menu();
        menu3.setParentId(2l);
        menu3.setId(3l);
        menu3.setName("菜单三级");
        sourceList.add(menu3);
        
        Menu menu4=new Menu();
        menu4.setParentId(3l);
        menu4.setId(4l);
        menu4.setName("菜单四级");
        sourceList.add(menu4);
        
        Menu menu5=new Menu();
        menu5.setParentId(1l);
        menu5.setId(6l);
        menu5.setName("菜单二级2");
        sourceList.add(menu5);
        
        Menu childrens = treeRoot(sourceList, menu);
        System.out.println(JSONObject.toJSON(childrens));
    }

}

 

 

 

 

 

import java.util.List;

public class Menu {
    //菜单id
    private Long id;
    //父节点id
    private Long parentId;
    //菜单名称
    private String name;
    //对应路径
    private String url;
    //图标
    private String icon;
    //是否选中
    private boolean selected;
    
    private List<Menu> childrens;

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public Long getParentId() {
        return parentId;
    }

    public void setParentId(Long parentId) {
        this.parentId = parentId;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getUrl() {
        return url;
    }

    public void setUrl(String url) {
        this.url = url;
    }

    public String getIcon() {
        return icon;
    }

    public void setIcon(String icon) {
        this.icon = icon;
    }

    public boolean isSelected() {
        return selected;
    }

    public void setSelected(boolean selected) {
        this.selected = selected;
    }

    public List<Menu> getChildrens() {
        return childrens;
    }

    public void setChildrens(List<Menu> childrens) {
        this.childrens = childrens;
    }
    
    
    

}