From 8fd9f6efcb6575d68d39302efe046558d8b1534c Mon Sep 17 00:00:00 2001
From: 潘志宝 <979469083@qq.com>
Date: 星期二, 14 一月 2025 14:39:16 +0800
Subject: [PATCH] 预测值

---
 iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/McsApiImpl.java |  122 ++++++++++++++++++++++++++++++++++------
 1 files changed, 102 insertions(+), 20 deletions(-)

diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/McsApiImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/McsApiImpl.java
index 4d9fc79..ea66d6a 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/McsApiImpl.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/McsApiImpl.java
@@ -1,5 +1,6 @@
 package com.iailab.module.model.api;
 
+import com.iailab.framework.common.pojo.PageResult;
 import com.iailab.framework.common.util.date.DateUtils;
 import com.iailab.framework.common.util.object.ConvertUtils;
 import com.iailab.module.data.api.plan.PlanItemApi;
@@ -11,14 +12,19 @@
 import com.iailab.module.data.common.ApiDataQueryDTO;
 import com.iailab.module.model.api.mcs.McsApi;
 import com.iailab.module.model.api.mcs.dto.*;
-import com.iailab.module.model.common.enums.CommonConstant;
+import com.iailab.module.model.enums.CommonConstant;
 import com.iailab.module.model.common.enums.PreLineTypeEnum;
 import com.iailab.module.model.mcs.pre.entity.DmModuleEntity;
 import com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity;
 import com.iailab.module.model.mcs.pre.entity.MmPredictAlarmMessageEntity;
 import com.iailab.module.model.mcs.pre.service.*;
 import com.iailab.module.model.mcs.pre.vo.MmPredictAlarmMessageSaveReqVO;
+import com.iailab.module.model.mcs.sche.entity.StScheduleSuggestEntity;
+import com.iailab.module.model.mcs.sche.service.StScheduleSchemeService;
+import com.iailab.module.model.mcs.sche.service.StScheduleSuggestService;
+import com.iailab.module.model.mcs.sche.vo.StScheduleSuggestSaveReqVO;
 import com.iailab.module.model.mdk.vo.ItemVO;
+import com.iailab.module.model.mpk.service.ChartParamService;
 import com.iailab.module.model.mpk.service.ChartService;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
@@ -42,6 +48,9 @@
 public class McsApiImpl implements McsApi {
 
     @Autowired
+    private DataPointApi dataPointApi;
+
+    @Autowired
     private DmModuleService dmModuleService;
 
     @Autowired
@@ -57,9 +66,6 @@
     private MmItemResultLastPointService mmItemResultLastPointService;
 
     @Autowired
-    private DataPointApi dataPointApi;
-
-    @Autowired
     private MmItemResultJsonService mmItemResultJsonService;
 
     @Autowired
@@ -69,7 +75,19 @@
     private MmPredictAlarmMessageService mmPredictAlarmMessageService;
 
     @Autowired
+    private StScheduleSuggestService stScheduleSuggestService;
+
+    @Autowired
     private PlanItemApi planItemApi;
+
+    @Autowired
+    private MmPredictAlarmConfigService mmPredictAlarmConfigService;
+
+    @Autowired
+    private StScheduleSchemeService stScheduleSchemeService;
+
+    @Autowired
+    private ChartParamService chartParamService;
 
     private int HOUR_MINS = 60;
 
@@ -122,7 +140,7 @@
         Date predictTime = reqVO.getPredictTime();
         if (predictTime == null) {
             MmItemOutputEntity output = mmItemOutputService.getOutPutById(reqVO.getOutIds().get(0));
-            ItemVO predictItem = mmPredictItemService.getItemById(output.getItemid());
+            ItemVO predictItem = mmPredictItemService.getItemByIdFromCache(output.getItemid());
             if (predictItem.getLastTime() != null) {
                 predictTime = predictItem.getLastTime();
             } else {
@@ -235,20 +253,20 @@
         if (startTime == null) {
             Calendar calendar = Calendar.getInstance();
             calendar.setTime(predictItem.getLastTime());
-            calendar.add(Calendar.MINUTE, -1 * predictItem.getPredictLength());
+            calendar.add(Calendar.SECOND, -1 * predictItem.getPredictLength() * predictItem.getGranularity());
             startTime = calendar.getTime();
         }
         Date endTime = reqVO.getEndTime();
         if (endTime == null) {
             Calendar calendar = Calendar.getInstance();
             calendar.setTime(predictItem.getLastTime());
-            calendar.add(Calendar.MINUTE, predictItem.getPredictLength());
+            calendar.add(Calendar.SECOND, predictItem.getPredictLength() * predictItem.getGranularity());
             endTime = calendar.getTime();
         }
         if (endTime.getTime() <= startTime.getTime()) {
             Calendar calendar = Calendar.getInstance();
             calendar.setTime(startTime);
-            calendar.add(Calendar.MINUTE, predictItem.getPredictLength());
+            calendar.add(Calendar.SECOND, predictItem.getPredictLength() * predictItem.getGranularity());
             endTime = calendar.getTime();
         }
 
@@ -262,7 +280,9 @@
         for (MmItemOutputEntity out : outs) {
             legend.add(out.getResultName());
             PreDataSampleViewRespDTO viewDto = new PreDataSampleViewRespDTO();
-            viewDto.setRealData(getHisData(out.getPointid(), startTime, endTime));
+            if (StringUtils.isNotBlank(out.getPointid())) {
+                viewDto.setRealData(getHisData(out.getPointid(), startTime, endTime));
+            }
             viewDto.setPreDataN(mmItemResultService.getData(out.getId(), startTime, endTime, DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND));
             viewMap.put(out.getResultName(), viewDto);
         }
@@ -290,16 +310,18 @@
         if (resultStr == null) {
             return result;
         }
+        String resultIndex = chartParams.get(CommonConstant.RESULT_INDEX);
+
         ItemVO predictItem = mmPredictItemService.getItemByItemNo(itemCode);
         if (predictItem == null || predictItem.getLastTime() == null) {
             return result;
         }
         String timeFormat = StringUtils.isBlank(reqVO.getTimeFormat()) ? DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND : reqVO.getTimeFormat();
         PreLineTypeEnum lineType = chartParams.get(CommonConstant.LINE_TYPE) == null ? PreLineTypeEnum.TN : PreLineTypeEnum.getEumByCode(chartParams.get(CommonConstant.LINE_TYPE));
-        BigDecimal rangeH = chartParams.get(CommonConstant.RANGE_H) == null ? BigDecimal.ZERO : new BigDecimal(chartParams.get(CommonConstant.RANGE_H));
-        BigDecimal rangeL = chartParams.get(CommonConstant.RANGE_L) == null ? BigDecimal.ZERO : new BigDecimal(chartParams.get(CommonConstant.RANGE_L));
-        BigDecimal limitH = chartParams.get(CommonConstant.LIMIT_H) == null ? BigDecimal.ZERO : new BigDecimal(chartParams.get(CommonConstant.LIMIT_H));
-        BigDecimal limitL = chartParams.get(CommonConstant.LIMIT_L) == null ? BigDecimal.ZERO : new BigDecimal(chartParams.get(CommonConstant.LIMIT_L));
+        BigDecimal rangeH = chartParams.get(CommonConstant.RANGE_H) == null ? null : new BigDecimal(chartParams.get(CommonConstant.RANGE_H));
+        BigDecimal rangeL = chartParams.get(CommonConstant.RANGE_L) == null ? null : new BigDecimal(chartParams.get(CommonConstant.RANGE_L));
+        BigDecimal limitH = chartParams.get(CommonConstant.LIMIT_H) == null ? null : new BigDecimal(chartParams.get(CommonConstant.LIMIT_H));
+        BigDecimal limitL = chartParams.get(CommonConstant.LIMIT_L) == null ? null : new BigDecimal(chartParams.get(CommonConstant.LIMIT_L));
         int lengthLeft = chartParams.get(CommonConstant.LENGTH_LEFT) == null ? predictItem.getPredictLength() : new BigDecimal(chartParams.get(CommonConstant.LENGTH_LEFT)).intValue();
         int lengthRight = chartParams.get(CommonConstant.LENGTH_RIGHT) == null ? predictItem.getPredictLength() : new BigDecimal(chartParams.get(CommonConstant.LENGTH_RIGHT)).intValue();
 
@@ -310,7 +332,7 @@
 
         List<String> categories = DateUtils.getTimeScale(startTime, endTime, predictItem.getGranularity(), timeFormat);
         List<String> legend = new ArrayList<>();
-        MmItemOutputEntity outPut = mmItemOutputService.getByItemid(predictItem.getId(), resultStr);
+        MmItemOutputEntity outPut = mmItemOutputService.getByItemid(predictItem.getId(), resultStr, resultIndex);
         PreDataViewRespDTO dataView = new PreDataViewRespDTO();
         dataView.setItemId(predictItem.getId());
         dataView.setItemName(predictItem.getItemName());
@@ -319,8 +341,12 @@
         dataView.setRangeL(rangeL);
         dataView.setLimitH(limitH);
         dataView.setLimitL(limitL);
-        dataView.setRealData(getHisData(outPut.getPointid(), startTime, endTime, timeFormat));
-        dataView.setCurData(mmItemResultJsonService.getData(outPut.getId(), predictTime, timeFormat));
+        if (StringUtils.isNotBlank(outPut.getPointid())) {
+            dataView.setRealData(getHisData(outPut.getPointid(), startTime, endTime, timeFormat));
+        } else {
+            dataView.setRealData(new ArrayList<>());
+        }
+        // dataView.setCurData(mmItemResultJsonService.getData(outPut.getId(), predictTime, timeFormat));
         switch (lineType) {
             case TN:
                 dataView.setPreDataN(mmItemResultService.getData(outPut.getId(), startTime, endTime, timeFormat));
@@ -333,12 +359,24 @@
                 break;
         }
 
-        if (!CollectionUtils.isEmpty(dataView.getCurData())) {
-            List<Double> curList = dataView.getCurData().stream().map(t -> {
+        if (!CollectionUtils.isEmpty(dataView.getPreDataN())) {
+            List<Double> curList = dataView.getPreDataN().stream().map(t -> {
                 return new Double(t[1].toString());
             }).collect(Collectors.toList());
             dataView.setPreMax(new BigDecimal(curList.stream().mapToDouble(Double::doubleValue).max().getAsDouble()).setScale(2, BigDecimal.ROUND_HALF_UP));
             dataView.setPreMin(new BigDecimal(curList.stream().mapToDouble(Double::doubleValue).min().getAsDouble()).setScale(2, BigDecimal.ROUND_HALF_UP));
+            dataView.setPreLast(new BigDecimal(curList.get(curList.size() - 1)));
+        }
+
+        String alarmObj = chartParams.get(CommonConstant.ALARM_OBJ);
+        if (StringUtils.isNotBlank(alarmObj)) {
+            List<AlarmMessageRespDTO> alarmList = ConvertUtils.sourceToTarget(mmPredictAlarmMessageService.getList(alarmObj, predictTime), AlarmMessageRespDTO.class);
+            dataView.setAlarmList(alarmList);
+            if (CollectionUtils.isEmpty(alarmList)) {
+                dataView.setAlarmMessage("暂无预警信息");
+            } else {
+                dataView.setAlarmMessage(alarmList.get(0).getContent());
+            }
         }
 
         result.setPredictTime(predictTime);
@@ -347,6 +385,18 @@
         result.setCategories(categories);
         result.setLegend(legend);
         result.setDataView(dataView);
+        return result;
+    }
+
+    @Override
+    public Map<String, List<Object[]>> getPreDataCur(PreDataJsonReqVO reqVO) {
+        Map<String, List<Object[]>> result = new HashMap<>();
+        if (reqVO == null || reqVO.getPredictTime() == null || CollectionUtils.isEmpty(reqVO.getOutputIdList())) {
+            return result;
+        }
+        reqVO.getOutputIdList().forEach(outPutId -> {
+            result.put(outPutId, mmItemResultJsonService.getData(outPutId, reqVO.getPredictTime(), DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND));
+        });
         return result;
     }
 
@@ -479,6 +529,11 @@
     }
 
     @Override
+    public List<AlarmConfigRespDTO> listAlarmConfig(Map<String, Object> params) {
+        return mmPredictAlarmConfigService.list(params);
+    }
+
+    @Override
     public AlarmMessageRespDTO getLastAlarmMessage(String alarmObj) {
         MmPredictAlarmMessageEntity entity = mmPredictAlarmMessageService.getLast(alarmObj);
         return ConvertUtils.sourceToTarget(entity, AlarmMessageRespDTO.class);
@@ -486,12 +541,19 @@
 
     @Override
     public Boolean createScheduleSuggest(ScheduleSuggestRespDTO dto) {
+        stScheduleSuggestService.create(ConvertUtils.sourceToTarget(dto, StScheduleSuggestSaveReqVO.class));
         return true;
     }
 
     @Override
-    public List<ScheduleSuggestRespDTO> listScheduleSuggest(ScheduleSuggestReqDTO params) {
-        return null;
+    public List<ScheduleSuggestRespDTO> listScheduleSuggest(Map<String, Object> params) {
+        return Collections.emptyList();
+    }
+
+    @Override
+    public List<ScheduleSuggestRespDTO> getLastLimitScheduleSuggest(String scheduleObj, Integer limit) {
+        List<StScheduleSuggestEntity> list = stScheduleSuggestService.getList(scheduleObj, limit);
+        return ConvertUtils.sourceToTarget(list, ScheduleSuggestRespDTO.class);
     }
 
     @Override
@@ -504,6 +566,26 @@
         return true;
     }
 
+    @Override
+    public List<StScheduleSchemeDTO> listScheduleScheme(String triggerMethod, String triggerCondition) {
+        Map<String, Object> params = new HashMap<>();
+        params.put("triggerMethod", triggerMethod);
+        params.put("triggerCondition", triggerCondition);
+        return stScheduleSchemeService.list(params);
+    }
+
+    @Override
+    public List<ChartParamDTO> getChartParamList(String chartCode) {
+        if (StringUtils.isBlank(chartCode)) {
+            return null;
+        }
+        return chartParamService.list(chartCode);
+    }
+
+    @Override
+    public PageResult<StAlarmAndSuggestRespVO> getAlarmAndSuggestPage(StAlarmAndSuggestPageReqVO reqVO) {
+        return stScheduleSchemeService.getAlarmAndSuggestPage(reqVO);
+    }
 
     private Date[] calResultTime(ItemVO predictItem, Date startTimeReq, Date endTimeReq, int lengthLeft, int lengthRight) {
         Date[] result = new Date[3];

--
Gitblit v1.9.3