package org.jeecg.modules.dnc.service.support;
|
|
import org.jeecg.modules.dnc.dto.DeviceGroupExt;
|
import org.jeecg.modules.dnc.entity.DeviceGroup;
|
import org.jeecg.modules.dnc.response.CommonGenericTree;
|
import org.springframework.beans.BeanUtils;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
public class DeviceGroupTreeWrapper {
|
public static List<CommonGenericTree> loadTree(List<DeviceGroupExt> deviceGroupExtList) {
|
Map<String, CommonGenericTree> treeMap = new HashMap<>();
|
for(DeviceGroupExt ext : deviceGroupExtList) {
|
treeMap = getBaseNodeMap(ext, treeMap);
|
}
|
return CommonGenericTree.convertMapToList(treeMap);
|
}
|
|
public static List<CommonGenericTree> loadDepartTree(List<DeviceGroupExt> deviceGroupExtList) {
|
List<CommonGenericTree> tree = new ArrayList<CommonGenericTree>();// TreeNode集合，存放所有树对象。
|
CommonGenericTree<DeviceGroup> node;
|
DeviceGroup deviceGroup;
|
for(DeviceGroupExt ext : deviceGroupExtList) {
|
node = new CommonGenericTree<DeviceGroup>();
|
node.setId(ext.getGroupId());
|
node.setLabel(ext.getGroupName());
|
node.setParentId(null);
|
//设置菜单节点图标 Start
|
String iconStr = "";
|
/*判断是否设置了菜单导航图标样式*/
|
if (!ext.hasChild()) {//导航页面的菜单节点
|
iconStr = "";//默认图标
|
}
|
node.setIconClass(iconStr);
|
node.setType(1);
|
deviceGroup = new DeviceGroup();
|
BeanUtils.copyProperties(ext, deviceGroup);
|
node.setEntity(deviceGroup);
|
node = loadChildrenNodes(ext, node);
|
tree.add(node);
|
}
|
return tree;
|
}
|
|
private static CommonGenericTree loadChildrenNodes(DeviceGroupExt info, CommonGenericTree node) {
|
if(info.hasChild()) {
|
List<DeviceGroupExt> children = info.getChildren();
|
CommonGenericTree<DeviceGroup> childNode;
|
DeviceGroup deviceGroup;
|
for (DeviceGroupExt ext : children) {
|
childNode = new CommonGenericTree();
|
childNode.setId(ext.getGroupId());
|
childNode.setLabel(ext.getGroupName());
|
childNode.setParentId(ext.getParentId());
|
//设置菜单节点图标 Start
|
String iconStr = "";
|
/*判断是否设置了菜单导航图标样式*/
|
if (!ext.hasChild()) {//导航页面的菜单节点
|
iconStr = "";//默认图标
|
}
|
childNode.setIconClass(iconStr);
|
childNode.setType(1);
|
deviceGroup = new DeviceGroup();
|
BeanUtils.copyProperties(ext, deviceGroup);
|
childNode.setEntity(deviceGroup);
|
childNode = loadChildrenNodes(ext, childNode);
|
node.addChildren(childNode);
|
}
|
}
|
return node;
|
}
|
|
private static Map<String,CommonGenericTree> getBaseNodeMap(DeviceGroupExt ext, Map<String,CommonGenericTree> map) {
|
CommonGenericTree<DeviceGroup> node;
|
DeviceGroup deviceGroup;
|
if(ext.getRankLevel() == 1) {
|
node = new CommonGenericTree<DeviceGroup>();
|
node.setId(ext.getGroupId());
|
node.setLabel(ext.getGroupName());
|
node.setParentId(null);
|
//设置菜单节点图标 Start
|
String iconStr = "";
|
/*判断是否设置了菜单导航图标样式*/
|
if (!ext.hasChild()) {//导航页面的菜单节点
|
iconStr = "";//默认图标
|
}
|
node.setIconClass(iconStr);
|
node.setType(1);
|
deviceGroup = new DeviceGroup();
|
BeanUtils.copyProperties(ext, deviceGroup);
|
node.setEntity(deviceGroup);
|
map.put(node.getId(), node);
|
return map;
|
}else {
|
node = new CommonGenericTree<DeviceGroup>();
|
node.setId(ext.getGroupId());
|
node.setLabel(ext.getGroupName());
|
node.setParentId(ext.getParentId());
|
//设置菜单节点图标 Start
|
String iconStr = "";
|
/*判断是否设置了菜单导航图标样式*/
|
if (!ext.hasChild()) {//导航页面的菜单节点
|
iconStr = "";//默认图标
|
}
|
node.setIconClass(iconStr);
|
node.setType(1);
|
deviceGroup = new DeviceGroup();
|
BeanUtils.copyProperties(ext, deviceGroup);
|
node.setEntity(deviceGroup);
|
map.put(node.getId(), node);
|
return getBaseNodeMap(ext.getParent(), map);
|
}
|
}
|
}
|
