From 8774f95f86420215f8cc9e357598f250710158ed Mon Sep 17 00:00:00 2001
From: letsgocoding <sdxt_0802@163.com>
Date: 星期二, 11 二月 2025 17:14:54 +0800
Subject: [PATCH] !61 能流图调试 Merge pull request !61 from letsgocoding/wt_from_develop1.0

---
 zhitan-admin/src/main/java/com/zhitan/web/controller/comprehensivestatistics/YearComprehensiveController.java |    4 ++++
 zhitan-system/src/main/java/com/zhitan/statisticalAnalysis/domain/dto/FlowChartsDTO.java                      |    5 ++++-
 zhitan-system/src/main/java/com/zhitan/dataitem/service/impl/DataItemServiceImpl.java                         |   13 ++++++++++++-
 zhitan-admin/src/main/java/com/zhitan/web/controller/keyequipment/YearKeyEquipmentController.java             |    9 ++++++++-
 4 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/zhitan-admin/src/main/java/com/zhitan/web/controller/comprehensivestatistics/YearComprehensiveController.java b/zhitan-admin/src/main/java/com/zhitan/web/controller/comprehensivestatistics/YearComprehensiveController.java
index ead5b3f..fc5a46c 100644
--- a/zhitan-admin/src/main/java/com/zhitan/web/controller/comprehensivestatistics/YearComprehensiveController.java
+++ b/zhitan-admin/src/main/java/com/zhitan/web/controller/comprehensivestatistics/YearComprehensiveController.java
@@ -88,6 +88,10 @@
     @GetMapping("/listChart")
     @ApiOperation(value = "鑾峰彇缁煎悎鎸囨爣鍒嗘瀽鍥捐〃锛堝勾锛夋暟鎹�")
     public AjaxResult listChart(DataItem dataItem){
+
+        dataItem.setBeginTime(DateUtil.beginOfYear(dataItem.getDataTime()));
+        dataItem.setEndTime(DateUtil.endOfYear(dataItem.getDataTime()));
+
         List<YearComperhensive> list = yearComprehensive.getListChart(dataItem.getIndexId(),dataItem.getBeginTime(),dataItem.getEndTime(), dataItem.getTimeType(),dataItem.getIndexStorageId());
         return AjaxResult.success(list);
     }
diff --git a/zhitan-admin/src/main/java/com/zhitan/web/controller/keyequipment/YearKeyEquipmentController.java b/zhitan-admin/src/main/java/com/zhitan/web/controller/keyequipment/YearKeyEquipmentController.java
index 00afae6..8f0c4c5 100644
--- a/zhitan-admin/src/main/java/com/zhitan/web/controller/keyequipment/YearKeyEquipmentController.java
+++ b/zhitan-admin/src/main/java/com/zhitan/web/controller/keyequipment/YearKeyEquipmentController.java
@@ -1,8 +1,10 @@
 package com.zhitan.web.controller.keyequipment;
 
+import cn.hutool.core.date.DateUtil;
 import com.zhitan.common.core.controller.BaseController;
 import com.zhitan.common.core.domain.AjaxResult;
 import com.zhitan.common.core.page.TableDataInfo;
+import com.zhitan.comprehensivestatistics.domain.YearComperhensive;
 import com.zhitan.keyequipment.domain.YearKeyEquipment;
 import com.zhitan.keyequipment.service.IYearKeyEquipmentService;
 import com.zhitan.model.domain.EnergyIndex;
@@ -53,9 +55,13 @@
         }
         List<String> indexIds = energyList.stream().map(EnergyIndex::getIndexId).collect(Collectors.toList());
         List<YearKeyEquipment> dataList=new ArrayList<>();
+
+        dataItem.setBeginTime(DateUtil.beginOfYear(dataItem.getDataTime()));
+        dataItem.setEndTime(DateUtil.endOfYear(dataItem.getDataTime()));
+
         DateFormat df = new SimpleDateFormat("yyyy");
         SimpleDateFormat sf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-        String aa= df.format(dataItem.getBeginTime());
+        String aa= df.format(dataItem.getDataTime());
         String bb="";
         int i = 1;
         while (i <= 12) {
@@ -78,6 +84,7 @@
     @GetMapping("/listChart")
     @ApiOperation(value = "閲嶇偣璁惧鑳借�楃粺璁★紙骞达級鍥捐〃")
     public AjaxResult listChart(DataItem dataItem){
+
         List<YearKeyEquipment> list = yearKeyEquipmentService.getListChart(dataItem.getIndexId(),dataItem.getBeginTime(),dataItem.getEndTime(), dataItem.getTimeType(),dataItem.getIndexStorageId());
         return AjaxResult.success(list);
     }
diff --git a/zhitan-system/src/main/java/com/zhitan/dataitem/service/impl/DataItemServiceImpl.java b/zhitan-system/src/main/java/com/zhitan/dataitem/service/impl/DataItemServiceImpl.java
index 2f95c17..27fb12a 100644
--- a/zhitan-system/src/main/java/com/zhitan/dataitem/service/impl/DataItemServiceImpl.java
+++ b/zhitan-system/src/main/java/com/zhitan/dataitem/service/impl/DataItemServiceImpl.java
@@ -1,5 +1,7 @@
 package com.zhitan.dataitem.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.zhitan.common.enums.TimeType;
 import com.zhitan.dataitem.domain.StagseDataEntry;
 import com.zhitan.dataitem.domain.vo.NodeIndexValueVO;
@@ -149,7 +151,16 @@
     public FlowChartsVO getFlowCharts(FlowChartsDTO dto) {
         FlowChartsVO flowChartsVO = new FlowChartsVO();
         // 鐖惰妭鐐筰d
-        String nodeId = dto.getNodeId();
+        LambdaQueryWrapper<ModelNode> wrapper = Wrappers.<ModelNode>lambdaQuery()
+                .eq(ModelNode::getModelCode, dto.getModelCode())
+                .isNull(ModelNode::getParentId);
+        List<ModelNode> modelNodes = modelNodeMapper.selectList(wrapper);
+        if(ObjectUtils.isEmpty(modelNodes)){
+            throw new RuntimeException("鏈煡璇㈠埌妯″瀷淇℃伅");
+        }
+        Optional<ModelNode> modelNodeInfo = modelNodes.stream().findFirst();
+        String nodeId = modelNodeInfo.map(ModelNode::getNodeId).toString();
+        dto.setNodeId(nodeId);
         String energyType = dto.getEnergyType();
         LocalDate queryTime = dto.getQueryTime();
         TimeType timeType = dto.getTimeType();
diff --git a/zhitan-system/src/main/java/com/zhitan/statisticalAnalysis/domain/dto/FlowChartsDTO.java b/zhitan-system/src/main/java/com/zhitan/statisticalAnalysis/domain/dto/FlowChartsDTO.java
index b9dc2f9..968c4d7 100644
--- a/zhitan-system/src/main/java/com/zhitan/statisticalAnalysis/domain/dto/FlowChartsDTO.java
+++ b/zhitan-system/src/main/java/com/zhitan/statisticalAnalysis/domain/dto/FlowChartsDTO.java
@@ -44,10 +44,13 @@
     /**
      * 鑺傜偣id
      */
-    @NotBlank(message = "鑺傜偣id涓嶈兘涓虹┖")
     @ApiModelProperty(value = "鑺傜偣id")
     private String nodeId;
 
+    @NotBlank(message = "妯″瀷缂栫爜涓嶈兘涓虹┖")
+    @ApiModelProperty(value = "妯″瀷缂栫爜")
+    private String modelCode;
+
     public TimeType getTimeType() {
         if (ObjectUtils.isEmpty(timeType)) {
             return TimeType.DAY;

--
Gitblit v1.9.3