package com.lxzn.ucenter.service.support;
|
|
import com.lxzn.framework.domain.ucenter.Menu;
|
import com.lxzn.framework.domain.ucenter.MenuButton;
|
import com.lxzn.framework.domain.ucenter.MenuButtonPermission;
|
import com.lxzn.framework.model.response.CommonJsonTree;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
public class MenuButtonTreeWrapper {
|
public static List<CommonJsonTree> loadTree(List<Menu> menuList, List<MenuButton> menuButtonList, Map<String, MenuButtonPermission> map) {
|
List<CommonJsonTree> tree = new ArrayList<CommonJsonTree>();// TreeNode集合，存放所有树对象。
|
List<CommonJsonTree> result = new ArrayList<CommonJsonTree>();// TreeNode集合，存放所有树对象。
|
Map<String, CommonJsonTree> menuTreeMap = new HashMap<>();
|
menuList.forEach(item -> {
|
CommonJsonTree node = new CommonJsonTree();
|
node.setId(item.getMenuId());
|
node.setLabel(item.getMenuName());
|
node.setDisabled(false);
|
menuTreeMap.put(item.getMenuId(), node);
|
tree.add(node);
|
});
|
menuButtonList.forEach(item -> {
|
CommonJsonTree node = new CommonJsonTree();
|
node.setId(item.getMenuButtonId());
|
node.setLabel(item.getButtonAlias());
|
node.setRField(item.getButtonId());
|
node.setParentId(item.getMenuId());
|
if(map != null && !map.isEmpty() &&map.containsKey(item.getMenuId() + "_" + item.getButtonId())) {
|
node.setChecked(true);
|
}
|
if(menuTreeMap.containsKey(item.getMenuId())) {
|
CommonJsonTree parent = menuTreeMap.get(item.getMenuId());
|
parent.addChildren(node);
|
}
|
});
|
for(CommonJsonTree commonJsonTree : tree) {
|
if(commonJsonTree.hasChildren()) {
|
result.add(commonJsonTree);
|
}
|
}
|
return result;
|
}
|
}
|
