JavaUtils-遞歸解析多級菜單(數據樹)

工具類

多級菜單數據:id、父級id、用於裝載子數據的集合
/**
 * 樹形數據實體接口
 * @param <E>
 * @author jianda
 * @date 2017年5月26日
 */
public interface TreeEntity<E> {
    public String getId();
    public String getParentId();
    public void setChildList(List<E> childList);
}
/**
 * 解析樹形數據工具類
 * 
 * @author jianda
 * @date 2017年5月29日
 */
public class TreeParser{
    /**
     * 解析樹形數據
     * @param topId
     * @param entityList
     * @return
     * @author jianda
     * @date 2017年5月29日
     */
    public static <E extends TreeEntity<E>> List<E> getTreeList(String topId, List<E> entityList) {
        List<E> resultList=new ArrayList<>();
        
        //獲取頂層元素集合
        String parentId;
        for (E entity : entityList) {
            parentId=entity.getParentId();
            if(parentId==null||topId.equals(parentId)){
                resultList.add(entity);
            }
        }
        
        //獲取每個頂層元素的子數據集合
        for (E entity : resultList) {
            entity.setChildList(getSubList(entity.getId(),entityList));
        }
        
        return resultList;
    }
    
    /**
     * 獲取子數據集合
     * @param id
     * @param entityList
     * @return
     * @author jianda
     * @date 2017年5月29日
     */
    private  static  <E extends TreeEntity<E>>  List<E> getSubList(String id, List<E> entityList) {
        List<E> childList=new ArrayList<>();
        String parentId;
        
        //子集的直接子對象
        for (E entity : entityList) {
            parentId=entity.getParentId();
            if(id.equals(parentId)){
                childList.add(entity);
            }
        }
        
        //子集的間接子對象
        for (E entity : childList) {
            entity.setChildList(getSubList(entity.getId(), entityList));
        }
        
        //遞歸退出條件
        if(childList.size()==0){
            return null;
        }
        
        return childList;
    }
}

測試

public class Menu implements TreeEntity<Menu>{
    public String id;
    public String name;
    public String parentId;
    public List<Menu> childList;
    //set、get方法...
public class Test {
    public static void main(String[] args) {
        List<Menu> list=new ArrayList<>();
        Menu menu1=new Menu();
        menu1.setId("1");
        menu1.setParentId("0");
        menu1.setName("菜單1");
        list.add(menu1);
        
        Menu menu2=new Menu();
        menu2.setId("2");
        menu2.setParentId("0");
        menu2.setName("菜單2");
        list.add(menu2);
        
        Menu menu3=new Menu();
        menu3.setId("3");
        menu3.setParentId("1");
        menu3.setName("菜單11");
        list.add(menu3);
        
        Menu menu4=new Menu();
        menu4.setId("4");
        menu4.setParentId("3");
        menu4.setName("菜單111");
        list.add(menu4);
        
        List<Menu>    menus=TreeParser.getTreeList("0",list);
        System.out.println(menus);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章