From 443339e98893164231eb8af1857f7b9e150002fa Mon Sep 17 00:00:00 2001
From: Jay <csj123456>
Date: 星期二, 29 四月 2025 16:42:18 +0800
Subject: [PATCH] 新增修改预测模型配置接口
---
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmModelArithSettingsServiceImpl.java | 5 +++++
iailab-module-model/iailab-module-model-api/src/main/java/com/iailab/module/model/api/mcs/McsApi.java | 5 ++++-
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/controller/admin/McsApiController.java | 7 +++++++
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/dao/MmModelArithSettingsDao.java | 2 ++
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/McsApiImpl.java | 15 +++++++++++++++
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java | 5 +++++
6 files changed, 38 insertions(+), 1 deletions(-)
diff --git a/iailab-module-model/iailab-module-model-api/src/main/java/com/iailab/module/model/api/mcs/McsApi.java b/iailab-module-model/iailab-module-model-api/src/main/java/com/iailab/module/model/api/mcs/McsApi.java
index 26839f3..23e129e 100644
--- a/iailab-module-model/iailab-module-model-api/src/main/java/com/iailab/module/model/api/mcs/McsApi.java
+++ b/iailab-module-model/iailab-module-model-api/src/main/java/com/iailab/module/model/api/mcs/McsApi.java
@@ -12,7 +12,6 @@
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestParam;
-import javax.validation.Valid;
import java.math.BigDecimal;
import java.util.List;
import java.util.Map;
@@ -169,4 +168,8 @@
@Operation(summary = "根据预测时间获取最后预测值")
Map<String, BigDecimal> getPredictValueByTime(@RequestBody PredictLastValueReqVO reqVO);
+ @GetMapping(PREFIX + "/predict/model/setting/update")
+ @Operation(summary = "修改预测模型配置")
+ Boolean updatePredictModelSetting(@RequestParam("modelCode") String modelCode, @RequestParam("key") String key, @RequestParam("value") String value);
+
}
\ No newline at end of file
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 19268d2..45a1ee3 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
@@ -23,6 +23,7 @@
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.entity.MmPredictModelEntity;
import com.iailab.module.model.mcs.pre.enums.PredGranularityEnum;
import com.iailab.module.model.mcs.pre.service.*;
import com.iailab.module.model.mcs.pre.vo.MmItemOutputRespVO;
@@ -117,6 +118,13 @@
@Autowired
private StScheduleModelSettingService stScheduleModelSettingService;
+
+ @Autowired
+ private MmPredictModelService mmPredictModelService;
+
+ @Autowired
+ private MmModelArithSettingsService mmModelArithSettingsService;
+
private int HOUR_MINS = 60;
@Override
@@ -989,6 +997,13 @@
return stScheduleSuggestService.getLastSuggest(params);
}
+ @Override
+ public Boolean updatePredictModelSetting(String modelCode, String key, String value) {
+ MmPredictModelEntity model = mmPredictModelService.getByModelCode(modelCode);
+ mmModelArithSettingsService.updateByModelIdAndKey(model.getId(), key, value);
+ return true;
+ }
+
private Date[] calResultTime(ItemVO predictItem, Date startTimeReq, Date endTimeReq, int lengthLeft, int lengthRight) {
Date[] result = new Date[3];
Date predictTime = predictItem.getLastTime();
diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/controller/admin/McsApiController.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/controller/admin/McsApiController.java
index 39fa9f8..bd37854 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/controller/admin/McsApiController.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/controller/admin/McsApiController.java
@@ -275,4 +275,11 @@
List<ScheduleSuggestRespDTO> data = mcsApi.getLastSuggest(params);
return CommonResult.success(data);
}
+
+ @GetMapping("/predict/model/setting/update")
+ @Operation(summary = "修改调度模型配置")
+ public CommonResult<Boolean> updatePredictModelSetting(@RequestParam("modelCode") String modelCode, @RequestParam("key") String key, @RequestParam("value") String value){
+ mcsApi.updatePredictModelSetting(modelCode, key, value);
+ return success( true);
+ }
}
\ No newline at end of file
diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/dao/MmModelArithSettingsDao.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/dao/MmModelArithSettingsDao.java
index 82b30a0..f2a40f2 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/dao/MmModelArithSettingsDao.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/dao/MmModelArithSettingsDao.java
@@ -33,4 +33,6 @@
void insertList(List<MmModelArithSettingsEntity> list);
void updatePyFile(@Param("likeValue") String likeValue,@Param("value") String value);
+
+ void updateByModelIdAndKey(@Param("modelId") String modelId,@Param("key") String key,@Param("value") String value);
}
diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmModelArithSettingsServiceImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmModelArithSettingsServiceImpl.java
index dc13a83..88fa486 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmModelArithSettingsServiceImpl.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmModelArithSettingsServiceImpl.java
@@ -60,4 +60,9 @@
// 清空缓存
modelIdMap.clear();
}
+
+ @Override
+ public void updateByModelIdAndKey(String modelId, String key, String value) {
+ baseMapper.updateByModelIdAndKey(modelId,key,value);
+ }
}
diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java
index 4738817..5aa03cf 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/MmPredictModelServiceImpl.java
@@ -61,6 +61,11 @@
}
@Override
+ public MmPredictModelEntity getByModelCode(String modelCode) {
+ return mmPredictModelDao.selectOne("modelno", modelCode);
+ }
+
+ @Override
public MmPredictModelEntity getInfo(String id) {
return mmPredictModelDao.selectById(id);
}
--
Gitblit v1.9.3