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); } 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(); 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); } } 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); } 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); } } 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); }