工具類
多級菜單數據: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);
}
}