package com.lxzn.ucenter.service.support;
|
|
import com.lxzn.framework.domain.ucenter.ObjectBase;
|
import com.lxzn.framework.domain.ucenter.ObjectButton;
|
import com.lxzn.framework.domain.ucenter.ObjectButtonPermission;
|
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 ObjectButtonTreeWrapper {
|
|
public static List<CommonJsonTree> loadTree(List<ObjectBase> objectBaseList, List<ObjectButton> objectButtonList, Map<String, ObjectButtonPermission> map) {
|
List<CommonJsonTree> tree = new ArrayList<CommonJsonTree>();// TreeNode集合，存放所有树对象。
|
List<CommonJsonTree> result = new ArrayList<CommonJsonTree>();// TreeNode集合，存放所有树对象。
|
Map<String, CommonJsonTree> objectTreeMap = new HashMap<>();
|
objectBaseList.forEach(item -> {
|
CommonJsonTree node = new CommonJsonTree();
|
node.setId(item.getObjectId());
|
node.setLabel(item.getObjectName());
|
node.setDisabled(false);
|
objectTreeMap.put(item.getObjectId(), node);
|
tree.add(node);
|
});
|
objectButtonList.forEach(item -> {
|
CommonJsonTree node = new CommonJsonTree();
|
node.setId(item.getObjectButtonId());
|
node.setLabel(item.getButtonAlias());
|
node.setRField(item.getButtonId());
|
node.setParentId(item.getObjectId());
|
if(map != null && !map.isEmpty() && map.containsKey(item.getObjectId() + "_" + item.getButtonId())) {
|
node.setChecked(true);
|
}
|
if(objectTreeMap.containsKey(item.getObjectId())) {
|
CommonJsonTree parent = objectTreeMap.get(item.getObjectId());
|
parent.addChildren(node);
|
}
|
});
|
for(CommonJsonTree commonJsonTree : tree) {
|
if(commonJsonTree.hasChildren()) {
|
result.add(commonJsonTree);
|
}
|
}
|
return result;
|
}
|
}
|
