From 9be058b9e922f28bfbef8ca80c9f37eaa1d5490d Mon Sep 17 00:00:00 2001 From: hongawen <83944980@qq.com> Date: Tue, 3 Dec 2024 10:24:08 +0800 Subject: [PATCH 1/4] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E8=B0=83=E6=95=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../advance/enums/AdvanceResponseEnum.java | 2 ++ .../govern/voltage/SgEventController.java | 7 +++++++ .../bpm/controller/BpmTaskController.java | 19 +++++++++++++++++++ .../service/task/impl/BpmTaskServiceImpl.java | 2 +- 4 files changed, 29 insertions(+), 1 deletion(-) diff --git a/pqs-advance/advance-api/src/main/java/com/njcn/advance/enums/AdvanceResponseEnum.java b/pqs-advance/advance-api/src/main/java/com/njcn/advance/enums/AdvanceResponseEnum.java index 6e34eed28..1e415cdee 100644 --- a/pqs-advance/advance-api/src/main/java/com/njcn/advance/enums/AdvanceResponseEnum.java +++ b/pqs-advance/advance-api/src/main/java/com/njcn/advance/enums/AdvanceResponseEnum.java @@ -78,6 +78,8 @@ public enum AdvanceResponseEnum { MACHINE_HAS_UNIT("A0102","当前设备存在元器件"), EVENT_TIME_ERROR("A0102","暂降事件时间格式有误,请检查"), + + INVALID_FILE_TYPE("A0102","请选择CSV文件"), ; private final String code; diff --git a/pqs-advance/advance-boot/src/main/java/com/njcn/advance/controller/govern/voltage/SgEventController.java b/pqs-advance/advance-boot/src/main/java/com/njcn/advance/controller/govern/voltage/SgEventController.java index c5a3335a7..f5288f2f9 100644 --- a/pqs-advance/advance-boot/src/main/java/com/njcn/advance/controller/govern/voltage/SgEventController.java +++ b/pqs-advance/advance-boot/src/main/java/com/njcn/advance/controller/govern/voltage/SgEventController.java @@ -106,6 +106,13 @@ public class SgEventController extends BaseController { @OperateInfo(info = LogEnum.BUSINESS_COMMON, operateType = OperateType.ADD) public HttpResult importEventData(@RequestParam("file") MultipartFile file, @RequestParam("incomingLineId") String incomingLineId, @RequestParam("productLineId") String productLineId, HttpServletResponse response) { String methodDescribe = getMethodDescribe("importEventData"); + + // 检查文件MIME类型是否为csv + String contentType = file.getContentType(); + if (!"text/csv".equals(contentType)) { + throw new BusinessException(AdvanceResponseEnum.INVALID_FILE_TYPE); + } + ImportParams params = new ImportParams(); params.setHeadRows(1); params.setNeedVerify(true); diff --git a/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/controller/BpmTaskController.java b/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/controller/BpmTaskController.java index 2cb0f48d5..2b52e4a32 100644 --- a/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/controller/BpmTaskController.java +++ b/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/controller/BpmTaskController.java @@ -3,6 +3,7 @@ package com.njcn.bpm.controller; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollectionUtil; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.njcn.bpm.enums.BpmTaskStatusEnum; import com.njcn.bpm.pojo.dto.PageResult; import com.njcn.bpm.pojo.param.task.*; import com.njcn.bpm.pojo.po.BpmCategory; @@ -217,6 +218,24 @@ public class BpmTaskController extends BaseController { page.setTotal(bpmTaskRespVOPageResult.getTotal()); page.setSize(PageFactory.getPageSize(bpmTaskQueryParam)); page.setCurrent(PageFactory.getPageNum(bpmTaskQueryParam)); + + List records = page.getRecords(); + for (BpmTaskVO record : records) { + if(record.getStatus().equals(BpmTaskStatusEnum.CANCEL.getStatus()) && record.getReason().equals("系统自动取消")){ + record.setStatus(BpmTaskStatusEnum.APPROVE.getStatus()); + record.setReason(BpmTaskStatusEnum.APPROVE.getName()); + // 找到由谁完成的 + HttpResult> taskListByProcessInstanceId = this.getTaskListByProcessInstanceId(record.getProcessInstanceId()); + if(CollectionUtil.isNotEmpty(taskListByProcessInstanceId.getData())){ + for (BpmTaskVO datum : taskListByProcessInstanceId.getData()) { + if(datum.getStatus().equals(BpmTaskStatusEnum.APPROVE.getStatus()) && datum.getTaskDefinitionKey().equals(record.getTaskDefinitionKey())){ + record.setReason("由".concat(datum.getAssigneeUser().getName()).concat("完成审批")); + } + } + } + } + } + return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, page, methodDescribe); } diff --git a/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/service/task/impl/BpmTaskServiceImpl.java b/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/service/task/impl/BpmTaskServiceImpl.java index 9f65991ed..6eda4c886 100644 --- a/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/service/task/impl/BpmTaskServiceImpl.java +++ b/pqs-bpm/bpm-boot/src/main/java/com/njcn/bpm/service/task/impl/BpmTaskServiceImpl.java @@ -140,7 +140,7 @@ public class BpmTaskServiceImpl implements IBpmTaskService { // 审批时间倒序 .orderByHistoricTaskInstanceEndTime().desc(); if (StrUtil.isNotBlank(bpmTaskQueryParam.getSearchValue())) { - taskQuery.taskNameLike("%" + bpmTaskQueryParam.getSearchValue() + "%"); + taskQuery.processDefinitionNameLike("%" + bpmTaskQueryParam.getSearchValue() + "%"); } String beginTimeStr = bpmTaskQueryParam.getSearchBeginTime(); String endTimeStr = bpmTaskQueryParam.getSearchEndTime(); From dce85a5f4c8e09fab3413132c485ad98f8e2e5e7 Mon Sep 17 00:00:00 2001 From: hzj <826100833@qq.com> Date: Tue, 3 Dec 2024 10:53:46 +0800 Subject: [PATCH 2/4] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=8A=A5=E8=A1=A8?= =?UTF-8?q?=E6=97=B6=E9=97=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../harmonic/controller/ExportModelController.java | 10 +++++++++- .../harmonic/service/impl/CustomReportServiceImpl.java | 2 +- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/controller/ExportModelController.java b/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/controller/ExportModelController.java index 8fba0440a..5a03ffc7c 100644 --- a/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/controller/ExportModelController.java +++ b/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/controller/ExportModelController.java @@ -1,8 +1,10 @@ package com.njcn.harmonic.controller; +import cn.hutool.core.date.DatePattern; import cn.hutool.core.date.DateTime; import cn.hutool.core.date.DateUtil; +import cn.hutool.core.date.LocalDateTimeUtil; import cn.hutool.core.util.ObjectUtil; import com.njcn.common.pojo.enums.response.CommonResponseEnum; import com.njcn.common.pojo.exception.BusinessException; @@ -44,6 +46,7 @@ import sun.misc.BASE64Encoder; import javax.servlet.http.HttpServletResponse; import java.io.*; import java.text.SimpleDateFormat; +import java.time.LocalDateTime; import java.util.*; import java.util.stream.Collectors; @@ -177,7 +180,12 @@ public class ExportModelController extends BaseController { ReportQueryParam param = new ReportQueryParam(); param.setLineId(lineIndex); param.setStartTime(DateUtil.format(startDate, "yyyy-MM-dd HH:mm:ss")); - param.setEndTime(DateUtil.format(endDate, "yyyy-MM-dd HH:mm:ss")); + LocalDateTime parse = LocalDateTimeUtil.parse(DateUtil.format(endDate, "yyyy-MM-dd HH:mm:ss"), DatePattern.NORM_DATETIME_PATTERN); + if(LocalDateTime.now().isBefore(parse)){ + param.setEndTime(LocalDateTimeUtil.format(LocalDateTime.now(),DatePattern.NORM_DATETIME_PATTERN)); + }else { + param.setEndTime(DateUtil.format(endDate, "yyyy-MM-dd HH:mm:ss")); + } param.setB(day == 0); //限值 diff --git a/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/service/impl/CustomReportServiceImpl.java b/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/service/impl/CustomReportServiceImpl.java index c177514bb..e519053e3 100644 --- a/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/service/impl/CustomReportServiceImpl.java +++ b/pqs-harmonic/harmonic-boot/src/main/java/com/njcn/harmonic/service/impl/CustomReportServiceImpl.java @@ -969,7 +969,7 @@ public class CustomReportServiceImpl implements CustomReportService { String localTime = InfluxDbSqlConstant.END_TIME; LocalDate localDate = LocalDateTimeUtil.parseDate(reportSearchParam.getEndTime(), DatePattern.NORM_DATE_PATTERN); if(LocalDate.now().isEqual(localDate)){ - localTime = LocalTime.now().format(DatePattern.NORM_TIME_FORMATTER); + localTime =" "+ LocalTime.now().format(DatePattern.NORM_TIME_FORMATTER); } son.set("v", reportSearchParam.getStartTime() + InfluxDbSqlConstant.START_TIME + "_" + reportSearchParam.getEndTime() + localTime); } else { From 7cfbb8fb5338cc4eeb0d1f01cb31107953ec2139 Mon Sep 17 00:00:00 2001 From: hongawen <83944980@qq.com> Date: Tue, 3 Dec 2024 11:30:43 +0800 Subject: [PATCH 3/4] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E8=B0=83=E6=95=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../advance/service/impl/EventRelevantAnalysisServiceImpl.java | 1 - 1 file changed, 1 deletion(-) diff --git a/pqs-advance/advance-boot/src/main/java/com/njcn/advance/service/impl/EventRelevantAnalysisServiceImpl.java b/pqs-advance/advance-boot/src/main/java/com/njcn/advance/service/impl/EventRelevantAnalysisServiceImpl.java index 4530d4d5f..54cc0a62a 100644 --- a/pqs-advance/advance-boot/src/main/java/com/njcn/advance/service/impl/EventRelevantAnalysisServiceImpl.java +++ b/pqs-advance/advance-boot/src/main/java/com/njcn/advance/service/impl/EventRelevantAnalysisServiceImpl.java @@ -277,7 +277,6 @@ public class EventRelevantAnalysisServiceImpl extends ServiceImpl lambdaQueryWrapper = new LambdaQueryWrapper<>(); if (StrUtil.isNotBlank(baseParam.getSearchValue())) { lambdaQueryWrapper.like(RmpEventDetailAssPO::getContentDes, baseParam.getSearchValue()); - } lambdaQueryWrapper.between(RmpEventDetailAssPO::getTimeId, timeV.get(0), timeV.get(1)) .orderByAsc(RmpEventDetailAssPO::getTimeId); From c6475a862cebe02ae0be8b031a72f551c3c2dbca Mon Sep 17 00:00:00 2001 From: wr <1754607820@qq.com> Date: Tue, 3 Dec 2024 11:55:15 +0800 Subject: [PATCH 4/4] =?UTF-8?q?1.=E8=B0=83=E6=95=B4=E5=86=80=E5=8C=97?= =?UTF-8?q?=E7=BB=88=E7=AB=AF=E6=A0=91=E5=B1=95=E7=A4=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../njcn/device/pq/pojo/vo/TerminalTree.java | 5 +++ .../controller/YwZtSubstationController.java | 11 ++++++- .../pq/mapper/YwZtSubstationMapper.java | 8 ++++- .../mapper/mapping/YwZtSubstationMapper.xml | 28 +++++++++++++++++ .../pq/service/IYwZtSubstationService.java | 9 ++++++ .../service/impl/TerminalTreeServiceImpl.java | 31 +++++++++++++------ .../impl/YwZtSubstationServiceImpl.java | 26 ++++++++++++++++ 7 files changed, 106 insertions(+), 12 deletions(-) diff --git a/pqs-device/pq-device/pq-device-api/src/main/java/com/njcn/device/pq/pojo/vo/TerminalTree.java b/pqs-device/pq-device/pq-device-api/src/main/java/com/njcn/device/pq/pojo/vo/TerminalTree.java index bf089ce01..e6e018b95 100644 --- a/pqs-device/pq-device/pq-device-api/src/main/java/com/njcn/device/pq/pojo/vo/TerminalTree.java +++ b/pqs-device/pq-device/pq-device-api/src/main/java/com/njcn/device/pq/pojo/vo/TerminalTree.java @@ -70,4 +70,9 @@ public class TerminalTree implements Serializable { * 电网侧变电站 */ private String powerSubstationName; + + /** + * 电网侧变电站 + */ + private String objName; } diff --git a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/controller/YwZtSubstationController.java b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/controller/YwZtSubstationController.java index 72cc37533..cfa05d12f 100644 --- a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/controller/YwZtSubstationController.java +++ b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/controller/YwZtSubstationController.java @@ -5,6 +5,8 @@ import com.njcn.common.pojo.enums.common.LogEnum; import com.njcn.common.pojo.enums.response.CommonResponseEnum; import com.njcn.common.pojo.response.HttpResult; import com.njcn.common.utils.HttpResultUtil; +import com.njcn.device.biz.pojo.dto.SubGetBase; +import com.njcn.device.biz.pojo.param.SubstationParam; import com.njcn.device.pq.pojo.po.YwZtSubstation; import com.njcn.device.pq.service.IYwZtSubstationService; import com.njcn.web.controller.BaseController; @@ -51,5 +53,12 @@ public class YwZtSubstationController extends BaseController { return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, ywZtSubstation, methodDescribe); } - + @OperateInfo(info = LogEnum.BUSINESS_COMMON) + @PostMapping("/tagOrIdGetYWZtSub") + @ApiOperation("获取变电站信息") + public HttpResult> tagOrIdGetSub(@RequestBody SubstationParam substationParam) { + String methodDescribe = getMethodDescribe("tagOrIdGetSub"); + List subGetBases = ywZtSubstationService.getSubstation(substationParam); + return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, subGetBases, methodDescribe); + } } diff --git a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/YwZtSubstationMapper.java b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/YwZtSubstationMapper.java index 540e53617..56ab1a0cd 100644 --- a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/YwZtSubstationMapper.java +++ b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/YwZtSubstationMapper.java @@ -1,11 +1,16 @@ package com.njcn.device.pq.mapper; import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.njcn.device.biz.pojo.dto.SubGetBase; +import com.njcn.device.biz.pojo.param.SubstationParam; import com.njcn.device.pq.pojo.po.YwZtSubstation; +import org.apache.ibatis.annotations.Param; + +import java.util.List; /** *

- * Mapper 接口 + * Mapper 接口 *

* * @author wr @@ -13,4 +18,5 @@ import com.njcn.device.pq.pojo.po.YwZtSubstation; */ public interface YwZtSubstationMapper extends BaseMapper { + List selectSubstationList(@Param("param") SubstationParam substationParam); } diff --git a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/mapping/YwZtSubstationMapper.xml b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/mapping/YwZtSubstationMapper.xml index a21457d22..11adbd2fe 100644 --- a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/mapping/YwZtSubstationMapper.xml +++ b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/mapper/mapping/YwZtSubstationMapper.xml @@ -2,4 +2,32 @@ + diff --git a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/IYwZtSubstationService.java b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/IYwZtSubstationService.java index 9e1a4938b..4f620f934 100644 --- a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/IYwZtSubstationService.java +++ b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/IYwZtSubstationService.java @@ -1,6 +1,8 @@ package com.njcn.device.pq.service; import com.baomidou.mybatisplus.extension.service.IService; +import com.njcn.device.biz.pojo.dto.SubGetBase; +import com.njcn.device.biz.pojo.param.SubstationParam; import com.njcn.device.pq.pojo.po.YwZtSubstation; import java.util.List; @@ -19,4 +21,11 @@ public interface IYwZtSubstationService extends IService { Boolean syncYwZtSubstation(); List getYwZtSubstation(String orgId); + + /** + * 线上管控获取所有中台变电站信息 + * @param substationParam + * @return + */ + List getSubstation(SubstationParam substationParam); } diff --git a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/TerminalTreeServiceImpl.java b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/TerminalTreeServiceImpl.java index 55b1ef0ac..c519241bb 100644 --- a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/TerminalTreeServiceImpl.java +++ b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/TerminalTreeServiceImpl.java @@ -3,6 +3,7 @@ package com.njcn.device.pq.service.impl; import cn.hutool.core.bean.BeanUtil; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollectionUtil; +import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.njcn.common.pojo.enums.common.ServerEnum; @@ -319,6 +320,7 @@ public class TerminalTreeServiceImpl implements TerminalTreeService { if(powerFlagMap.containsKey(x.getId())){ x.setPowerFlag(powerFlagMap.get(x.getId()).getPowerFlag()); x.setPowerSubstationName(powerFlagMap.get(x.getId()).getPowerSubstationName()); + x.setObjName(powerFlagMap.get(x.getId()).getObjName()); }else{ x.setPowerFlag(LineBaseEnum.POWER_FLAG.getCode()); } @@ -333,8 +335,7 @@ public class TerminalTreeServiceImpl implements TerminalTreeService { lineObjNameMap.forEach((key, value)->{ List terminalTrees = Collections.unmodifiableList(collect1); //获取变电信息 - List collect = subList.stream().filter(x -> x.getName().equals(key)).collect(Collectors.toList()); - + List collect = subList.stream().filter(x -> x.getName().equals(key.replace("±",""))).collect(Collectors.toList()); //电网侧监测点 List powerFlag = value.stream().filter(x -> x.getPowerFlag().equals(LineBaseEnum.POWER_FLAG.getCode())).collect(Collectors.toList()); if(CollUtil.isEmpty(powerFlag)){ @@ -342,13 +343,24 @@ public class TerminalTreeServiceImpl implements TerminalTreeService { powerFlag.addAll(collect); } List subChildren = new ArrayList<>(); - getRecursionChildren(powerFlag,subChildren, terminalTrees); - + TerminalTree recursion = getRecursionChildren(powerFlag, terminalTrees); + if(ObjectUtil.isNotNull(recursion)){ + subChildren.add(recursion); + } //非电网侧监测点 List powerFlagNot = value.stream().filter(x -> x.getPowerFlag().equals(LineBaseEnum.POWER_FLAG_NOT.getCode())).collect(Collectors.toList()); List notSubChildren = new ArrayList<>(); - getRecursionChildren(powerFlagNot,notSubChildren,terminalTrees); + if(CollUtil.isNotEmpty(powerFlagNot)){ + Map> objNameMap = powerFlagNot.stream().collect(Collectors.groupingBy(TerminalTree::getObjName)); + objNameMap.forEach((objNameKey,objNameValue)->{ + TerminalTree recursionChildren = getRecursionChildren(objNameValue, terminalTrees); + if(ObjectUtil.isNotNull(recursionChildren)){ + recursionChildren.setName(objNameKey); + notSubChildren.add(recursionChildren); + } + }); + } if(CollUtil.isNotEmpty(collect)){ List valueList = new ArrayList<>(); TerminalTree sub=new TerminalTree(); @@ -506,7 +518,7 @@ public class TerminalTreeServiceImpl implements TerminalTreeService { * @param all 检测对象 * @return */ - public void getRecursionChildren(List line, List info,List all) { + public TerminalTree getRecursionChildren(List line, List all) { List aa=new ArrayList<>(); for (TerminalTree lineTree : line) { findParentsRecursive(lineTree, all,aa); @@ -520,11 +532,10 @@ public class TerminalTreeServiceImpl implements TerminalTreeService { List lines = aa.stream().distinct().sorted(Comparator.comparing(TerminalTree::getSort)).collect(Collectors.toList()); List collect = distinctList.stream().filter(x -> x.getLevel() == LineBaseEnum.SUB_LEVEL.getCode()).collect(Collectors.toList()); if(CollUtil.isNotEmpty(collect)){ - for (TerminalTree tree : collect) { - TerminalTree terminalTree = BeanUtil.copyProperties(tree, TerminalTree.class); - info.add(findParents(lines,terminalTree)); - } + TerminalTree terminalTree = BeanUtil.copyProperties(collect.get(0), TerminalTree.class); + return findParents(lines,terminalTree); } + return null; } private static void findParentsRecursive(TerminalTree tree, List all,List newParents) { diff --git a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/YwZtSubstationServiceImpl.java b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/YwZtSubstationServiceImpl.java index 186b0c5da..1d069a5ca 100644 --- a/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/YwZtSubstationServiceImpl.java +++ b/pqs-device/pq-device/pq-device-boot/src/main/java/com/njcn/device/pq/service/impl/YwZtSubstationServiceImpl.java @@ -3,6 +3,8 @@ package com.njcn.device.pq.service.impl; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.njcn.device.biz.pojo.dto.SubGetBase; +import com.njcn.device.biz.pojo.param.SubstationParam; import com.njcn.device.pq.mapper.SubstationMapper; import com.njcn.device.pq.mapper.YwZtSubstationMapper; import com.njcn.device.pq.pojo.po.Substation; @@ -12,11 +14,13 @@ import com.njcn.device.pq.service.IYwZtSubstationService; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.njcn.user.api.DeptFeignClient; import com.njcn.user.pojo.dto.DeptDTO; +import com.njcn.user.pojo.po.Dept; import com.njcn.web.utils.WebUtil; import lombok.RequiredArgsConstructor; import org.springframework.stereotype.Service; import java.util.ArrayList; +import java.util.Collections; import java.util.List; import java.util.Map; import java.util.function.Function; @@ -98,6 +102,28 @@ public class YwZtSubstationServiceImpl extends ServiceImpl().in(CollUtil.isNotEmpty(orgIds),YwZtSubstation::getOrgId,orgIds)); } + @Override + public List getSubstation(SubstationParam substationParam) { + List orgIds = substationParam.getOrgIds(); + List deptList = deptFeignClient.getDeptInfoListByIds(orgIds).getData(); + List deptCode = deptList.stream().map(Dept::getCode).collect(Collectors.toList()); + substationParam.setOrgIds(deptCode); + List subGetBases = this.baseMapper.selectSubstationList(substationParam); + if(CollUtil.isNotEmpty(subGetBases)){ + Map deptMap = deptList.stream().collect(Collectors.toMap(Dept::getCode, Function.identity())); + subGetBases.stream().forEach(x->{ + if(deptMap.containsKey(x.getOrgId())){ + Dept deptDTO = deptMap.get(x.getOrgId()); + x.setOrgName(deptDTO.getName()); + x.setOrgId(deptDTO.getId()); + } + }); + return subGetBases; + } + return Collections.emptyList(); + } + + public String getDzmc(String dzmc) { if(StrUtil.isNotBlank(dzmc)) { dzmc = dzmc.replace("变电站", "")