简单的部门树接口

This commit is contained in:
cdf
2024-05-13 20:34:47 +08:00
parent ecf53c9bbb
commit fb6ce3fc11

View File

@@ -155,6 +155,7 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
/** /**
* 处理变电站 * 处理变电站
*
* @param targetData * @param targetData
* @param childrenData * @param childrenData
* @param isLine * @param isLine
@@ -174,7 +175,7 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
} }
//变电站 //变电站
targetData = targetData.stream().peek(terminalTree -> { targetData = targetData.stream().peek(terminalTree -> {
if(isLine){ if (isLine) {
//变电站集合 //变电站集合
List<TerminalTree> terminalTrees = groupLine.get(terminalTree.getId()); List<TerminalTree> terminalTrees = groupLine.get(terminalTree.getId());
@@ -184,10 +185,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
return pid[LineBaseEnum.DEVICE_LEVEL.getCode()]; return pid[LineBaseEnum.DEVICE_LEVEL.getCode()];
}).distinct().collect(Collectors.toList()).size(); }).distinct().collect(Collectors.toList()).size();
terminalTree.setName(terminalTree.getName()+""+size+"台装置)"); terminalTree.setName(terminalTree.getName() + "" + size + "台装置)");
terminalTree.setChildren(groupLine.get(terminalTree.getId())); terminalTree.setChildren(groupLine.get(terminalTree.getId()));
}else{ } else {
terminalTree.setChildren(groupLine.get(terminalTree.getId())); terminalTree.setChildren(groupLine.get(terminalTree.getId()));
} }
}).collect(Collectors.toList()); }).collect(Collectors.toList());
@@ -234,31 +235,31 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
deviceInfoParam.setServerName(ServerEnum.HARMONIC.getName()); deviceInfoParam.setServerName(ServerEnum.HARMONIC.getName());
List<GeneralDeviceDTO> generalDeviceDTOList = generalDeviceService.getDeviceInfo(deviceInfoParam, Stream.of(0).collect(Collectors.toList()), Stream.of(1).collect(Collectors.toList())); List<GeneralDeviceDTO> generalDeviceDTOList = generalDeviceService.getDeviceInfo(deviceInfoParam, Stream.of(0).collect(Collectors.toList()), Stream.of(1).collect(Collectors.toList()));
for(GeneralDeviceDTO generalDeviceDTO : generalDeviceDTOList){ for (GeneralDeviceDTO generalDeviceDTO : generalDeviceDTOList) {
if(CollectionUtil.isEmpty(generalDeviceDTO.getLineIndexes())){ if (CollectionUtil.isEmpty(generalDeviceDTO.getLineIndexes())) {
continue; continue;
} }
// 通过供电公司索引查询省会 // 通过供电公司索引查询省会
List<TerminalSelectTreeVO> children = new ArrayList<>(); List<TerminalSelectTreeVO> children = new ArrayList<>();
if(level == 4){ if (level == 6) {
List<Line> devList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId,generalDeviceDTO.getDeviceIndexes())); List<Line> busList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId, generalDeviceDTO.getVoltageIndexes()));
for(Line line: devList){ Map<String, Line> map = busList.stream().collect(Collectors.toMap(Line::getId, Function.identity()));
List<Line> lineList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId, generalDeviceDTO.getLineIndexes()));
for (Line line : lineList) {
TerminalSelectTreeVO terminalSelectTreeVO = new TerminalSelectTreeVO();
terminalSelectTreeVO.setValue(line.getId());
terminalSelectTreeVO.setLabel(map.get(line.getPid()).getName() + "_" + line.getName());
children.add(terminalSelectTreeVO);
}
} else {
List<Line> devList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId, generalDeviceDTO.getDeviceIndexes()));
for (Line line : devList) {
TerminalSelectTreeVO terminalSelectTreeVO = new TerminalSelectTreeVO(); TerminalSelectTreeVO terminalSelectTreeVO = new TerminalSelectTreeVO();
terminalSelectTreeVO.setValue(line.getId()); terminalSelectTreeVO.setValue(line.getId());
terminalSelectTreeVO.setLabel(line.getName()); terminalSelectTreeVO.setLabel(line.getName());
children.add(terminalSelectTreeVO); children.add(terminalSelectTreeVO);
} }
}else if(level == 6){
List<Line> busList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId,generalDeviceDTO.getVoltageIndexes()));
Map<String,Line> map = busList.stream().collect(Collectors.toMap(Line::getId, Function.identity()));
List<Line> lineList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId,generalDeviceDTO.getLineIndexes()));
for(Line line: lineList){
TerminalSelectTreeVO terminalSelectTreeVO = new TerminalSelectTreeVO();
terminalSelectTreeVO.setValue(line.getId());
terminalSelectTreeVO.setLabel(map.get(line.getPid()).getName()+"_"+line.getName());
children.add(terminalSelectTreeVO);
}
} }
TerminalSelectTreeVO terminalSelectTreeVO = new TerminalSelectTreeVO(); TerminalSelectTreeVO terminalSelectTreeVO = new TerminalSelectTreeVO();
terminalSelectTreeVO.setValue(generalDeviceDTO.getIndex()); terminalSelectTreeVO.setValue(generalDeviceDTO.getIndex());
@@ -298,10 +299,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
if (CollectionUtil.isNotEmpty(allList)) { if (CollectionUtil.isNotEmpty(allList)) {
List<TerminalTree> projectList; List<TerminalTree> projectList;
List<TerminalTree> provinceList; List<TerminalTree> provinceList;
if(!"0".equals(areaId)){ if (!"0".equals(areaId)) {
projectList=allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList());
}else{ } else {
projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList());
} }
@@ -380,10 +381,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
if (CollectionUtil.isNotEmpty(allList)) { if (CollectionUtil.isNotEmpty(allList)) {
List<TerminalTree> projectList; List<TerminalTree> projectList;
List<TerminalTree> provinceList; List<TerminalTree> provinceList;
if(!"0".equals(areaId)){ if (!"0".equals(areaId)) {
projectList=allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList());
}else{ } else {
projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList());
} }
@@ -453,10 +454,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
if (CollectionUtil.isNotEmpty(allList)) { if (CollectionUtil.isNotEmpty(allList)) {
List<TerminalTree> projectList; List<TerminalTree> projectList;
List<TerminalTree> provinceList; List<TerminalTree> provinceList;
if(!"0".equals(areaId)){ if (!"0".equals(areaId)) {
projectList=allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList());
}else{ } else {
projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList());
} }
@@ -534,10 +535,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
if (CollectionUtil.isNotEmpty(allList)) { if (CollectionUtil.isNotEmpty(allList)) {
List<TerminalTree> projectList; List<TerminalTree> projectList;
List<TerminalTree> provinceList; List<TerminalTree> provinceList;
if(!"0".equals(areaId)){ if (!"0".equals(areaId)) {
projectList=allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode()) && lineMapper.selectProject(areaId).contains(item.getId())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode()) && item.getName().equals(areaId)).collect(Collectors.toList());
}else{ } else {
projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList()); projectList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROJECT_LEVEL.getCode())).collect(Collectors.toList());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList()); provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList());
} }