简单的部门树接口

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 childrenData
* @param isLine
@@ -174,7 +175,7 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
}
//变电站
targetData = targetData.stream().peek(terminalTree -> {
if(isLine){
if (isLine) {
//变电站集合
List<TerminalTree> terminalTrees = groupLine.get(terminalTree.getId());
@@ -184,10 +185,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
return pid[LineBaseEnum.DEVICE_LEVEL.getCode()];
}).distinct().collect(Collectors.toList()).size();
terminalTree.setName(terminalTree.getName()+""+size+"台装置)");
terminalTree.setName(terminalTree.getName() + "" + size + "台装置)");
terminalTree.setChildren(groupLine.get(terminalTree.getId()));
}else{
} else {
terminalTree.setChildren(groupLine.get(terminalTree.getId()));
}
}).collect(Collectors.toList());
@@ -234,31 +235,31 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
deviceInfoParam.setServerName(ServerEnum.HARMONIC.getName());
List<GeneralDeviceDTO> generalDeviceDTOList = generalDeviceService.getDeviceInfo(deviceInfoParam, Stream.of(0).collect(Collectors.toList()), Stream.of(1).collect(Collectors.toList()));
for(GeneralDeviceDTO generalDeviceDTO : generalDeviceDTOList){
if(CollectionUtil.isEmpty(generalDeviceDTO.getLineIndexes())){
for (GeneralDeviceDTO generalDeviceDTO : generalDeviceDTOList) {
if (CollectionUtil.isEmpty(generalDeviceDTO.getLineIndexes())) {
continue;
}
// 通过供电公司索引查询省会
List<TerminalSelectTreeVO> children = new ArrayList<>();
if(level == 4){
List<Line> devList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId,generalDeviceDTO.getDeviceIndexes()));
for(Line line: devList){
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);
}
} else {
List<Line> devList = lineMapper.selectList(new LambdaQueryWrapper<Line>().in(Line::getId, generalDeviceDTO.getDeviceIndexes()));
for (Line line : devList) {
TerminalSelectTreeVO terminalSelectTreeVO = new TerminalSelectTreeVO();
terminalSelectTreeVO.setValue(line.getId());
terminalSelectTreeVO.setLabel(line.getName());
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.setValue(generalDeviceDTO.getIndex());
@@ -298,10 +299,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService {
if (CollectionUtil.isNotEmpty(allList)) {
List<TerminalTree> projectList;
List<TerminalTree> provinceList;
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());
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());
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());
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)) {
List<TerminalTree> projectList;
List<TerminalTree> provinceList;
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());
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());
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());
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)) {
List<TerminalTree> projectList;
List<TerminalTree> provinceList;
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());
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());
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());
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)) {
List<TerminalTree> projectList;
List<TerminalTree> provinceList;
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());
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());
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());
provinceList = allList.stream().filter(item -> item.getLevel().equals(LineBaseEnum.PROVINCE_LEVEL.getCode())).collect(Collectors.toList());
}