實體類:
public class Menu {
// 菜單id
private String id;
// 菜單名稱
private String name;
// 父菜單id
private String parentId;
// 菜單url
private String url;
// 菜單圖標
private String icon;
// 菜單順序
private int order;
// 子菜單
private List<Menu> children;
// ... 省去getter和setter方法以及toString方法
}
菜單一般需要排序,我們根據Menu的order字段進行排序:
/*
* 排序,根據order排序
*/
public Comparator<Menu> order(){
Comparator<Menu> comparator = new Comparator<Menu>() {
@Override
public int compare(Menu o1, Menu o2) {
if(o1.getOrder() != o2.getOrder()){
return o1.getOrder() - o2.getOrder();
}
return 0;
}
};
return comparator;
}
生成樹的方法:
public Map<String,Object> findTree(){
Map<String,Object> data = new HashMap<String,Object>();
try {//查詢所有菜單
List<Menu> allMenu = menuDao.findTree();
//根節點
List<Menu> rootMenu = new ArrayList<Menu>();
for (Menu nav : allMenu) {
if(nav.getParentId().equals("0")){//父節點是0的,爲根節點。
rootMenu.add(nav);
}
}
/* 根據Menu類的order排序 */
Collections.sort(rootMenu, order());
//爲根菜單設置子菜單,getClild是遞歸調用的
for (Menu nav : rootMenu) {
/* 獲取根節點下的所有子節點 使用getChild方法*/
List<Menu> childList = getChild(nav.getId(), allMenu);
nav.setChildren(childList);//給根節點設置子節點
}
/**
* 輸出構建好的菜單數據。
*
*/
data.put("success", "true");
data.put("list", rootMenu);
return data;
} catch (Exception e) {
data.put("success", "false");
data.put("list", new ArrayList());
return data;
}
}
獲取子菜單:
/**
* 獲取子節點
* @param id 父節點id
* @param allMenu 所有菜單列表
* @return 每個根節點下,所有子菜單列表
*/
public List<Menu> getChild(String id,List<Menu> allMenu){
//子菜單
List<Menu> childList = new ArrayList<Menu>();
for (Menu nav : allMenu) {
// 遍歷所有節點,將所有菜單的父id與傳過來的根節點的id比較
//相等說明:爲該根節點的子節點。
if(nav.ParentId().equals(id)){
childList.add(nav);
}
}
//遞歸
for (Menu nav : childList) {
nav.setChildren(getChild(nav.getId(), allMenu));
}
Collections.sort(childList,order());//排序
//如果節點下沒有子節點,返回一個空List(遞歸退出)
if(childList.size() == 0){
return new ArrayList<Menu>();
}
return childList;
}
返回的JSON串如下:
{
"success": "true",
"list": [
{
"id": "1",
"name": "Java",
"parentid": "0",
"url": "http://www.aliouchen.com",
"order": 1,
"children": [
{
"id": "2",
"name": "併發編程",
"parentid": "1",
"url": "http://www.aliouchen.com",
"order": 1,
"children": []
},
{
"id": "3",
"name": "多線程",
"parentid": "1",
"url": "http://www.aliouchen.com",
"order": 2,
"children": [
"id": "4",
"name": "Thread",
"parentid": "3",
"url": "http://www.aliouchen.com",
"order": 1,
"children":[]
]
}
]
},
{
"id": "5",
"name": "Python",
"parentid": "0",
"url": "http://www.aliouchen.com",
"order": 2,
"children": []
}
]
}