package org.jeecg.modules.eam.wrapper;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
import org.jeecg.common.api.vo.CommonGenericTree;
|
import org.jeecg.common.util.StrUtils;
|
import org.jeecg.modules.eam.entity.Area;
|
import org.jeecg.modules.eam.entity.ProductionLine;
|
import org.jeecg.modules.eam.entity.Site;
|
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
|
@SuppressWarnings("rawtypes")
|
public class SiteAreaLineTreeWrapper {
|
|
/**
|
* 工厂/车间树
|
*/
|
public static List<CommonGenericTree> loadTree(List<Site> siteList, List<Area> areaList) {
|
//return loadTree(siteList, areaList, Collections.emptyList());
|
return null;
|
}
|
|
/**
|
* 工厂/车间/产线树
|
*/
|
@SuppressWarnings("unchecked")
|
public static List<CommonGenericTree> loadTree(String title,List<Site> siteList, List<Area> areaList,
|
List<ProductionLine> productionLineList) {
|
List<CommonGenericTree> list = new ArrayList<>();
|
Map<String, CommonGenericTree> siteMap = new HashMap<>();
|
Map<String, CommonGenericTree> areaMap = new HashMap<>();
|
CommonGenericTree<Site> root = new CommonGenericTree<>();
|
root.setKey("-1");
|
root.setTitle("中国航发");
|
root.setType(0);
|
list.add(root);
|
if (CollectionUtils.isNotEmpty(siteList)) {
|
CommonGenericTree<Site> siteNode;
|
CommonGenericTree<Area> areaNode;
|
CommonGenericTree<ProductionLine> lineNode;
|
for (Site site : siteList) {
|
siteNode = new CommonGenericTree<>();
|
siteNode.setKey(site.getId());
|
siteNode.setTitle(site.getName());
|
siteNode.setParentId(root.getKey());
|
siteNode.setIcon("");
|
siteNode.setType(1);
|
siteNode.setEntity(site);
|
root.addChildren(siteNode);
|
siteMap.put(site.getId(), siteNode);
|
}
|
for (Area area : areaList) {
|
if (siteMap.containsKey(area.getSiteId())) {
|
siteNode = siteMap.get(area.getSiteId());
|
areaNode = new CommonGenericTree<>();
|
areaNode.setKey(area.getId());
|
areaNode.setTitle(area.getName());
|
areaNode.setParentId(area.getSiteId());
|
areaNode.setIcon("");
|
areaNode.setType(2);
|
areaNode.setEntity(area);
|
siteNode.addChildren(areaNode);
|
areaMap.put(area.getId(), areaNode);
|
}
|
}
|
for (ProductionLine line : productionLineList) {
|
if (areaMap.containsKey(line.getAreaId())) {
|
areaNode = areaMap.get(line.getAreaId());
|
lineNode = new CommonGenericTree<>();
|
lineNode.setKey(line.getId());
|
lineNode.setTitle(line.getName());
|
lineNode.setParentId(line.getAreaId());
|
lineNode.setIcon("");
|
lineNode.setType(3);
|
lineNode.setEntity(line);
|
lineNode.setDisabled(true);
|
areaNode.addChildren(lineNode);
|
}
|
}
|
}
|
return list;
|
}
|
|
}
|
