From a955f188c0380df9705c84f3a0eacccd2fc2375a Mon Sep 17 00:00:00 2001 From: dengzedong <dengzedong@email> Date: 星期二, 07 一月 2025 16:02:25 +0800 Subject: [PATCH] modelResult 迁移到influxdb --- iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/schedule/impl/ScheduleModelHandlerImpl.java | 18 ++++++++++++++---- 1 files changed, 14 insertions(+), 4 deletions(-) diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/schedule/impl/ScheduleModelHandlerImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/schedule/impl/ScheduleModelHandlerImpl.java index bf9930d..aa14af6 100644 --- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/schedule/impl/ScheduleModelHandlerImpl.java +++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/schedule/impl/ScheduleModelHandlerImpl.java @@ -4,7 +4,7 @@ import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.iail.model.IAILModel; -import com.iailab.module.model.common.enums.CommonConstant; +import com.iailab.module.model.enums.CommonConstant; import com.iailab.module.model.mcs.sche.entity.StScheduleModelEntity; import com.iailab.module.model.mcs.sche.entity.StScheduleModelSettingEntity; import com.iailab.module.model.mcs.sche.entity.StScheduleSchemeEntity; @@ -52,7 +52,8 @@ private SampleConstructor sampleConstructor; @Override - public ScheduleResultVO doSchedule(String schemeCode, Date scheduleTime, Map<Integer, Integer> dynamicDataLength) throws ModelInvokeException { + public ScheduleResultVO doSchedule(String schemeCode, Date scheduleTime, Map<Integer, Integer> dynamicDataLength, + Map<String, String> dynamicSettings) throws ModelInvokeException { ScheduleResultVO scheduleResult = new ScheduleResultVO(); StScheduleSchemeEntity scheduleScheme = stScheduleSchemeService.getByCode(schemeCode); StScheduleModelEntity scheduleModel = stScheduleModelService.get(scheduleScheme.getModelId()); @@ -73,7 +74,7 @@ } IAILModel newModelBean = composeNewModelBean(scheduleModel); - HashMap<String, Object> settings = getScheduleSettingsByModelId(modelId); + HashMap<String, Object> settings = getScheduleSettingsByModelId(modelId, dynamicSettings); if (settings == null) { log.error("模型setting不存在,modelId=" + modelId); return null; @@ -127,13 +128,22 @@ * 根据模型id获取参数map * * @param modelId + * @param dynamicSettings * @return */ - private HashMap<String, Object> getScheduleSettingsByModelId(String modelId) { + private HashMap<String, Object> getScheduleSettingsByModelId(String modelId, Map<String, String> dynamicSettings) { List<StScheduleModelSettingEntity> list = stScheduleModelSettingService.getByModelId(modelId); if (CollectionUtils.isEmpty(list)) { return null; } + //如果输入参数中包含setting,则覆盖默认的setting + if (!CollectionUtils.isEmpty(dynamicSettings)){ + list.forEach(setting -> { + if (dynamicSettings.containsKey(setting.getKey())) { + setting.setValue(dynamicSettings.get(setting.getKey())); + } + }); + } HashMap<String, Object> result = new HashMap<>(); for (StScheduleModelSettingEntity entry : list) { String valueType = entry.getValuetype().trim(); //去除两端空格 -- Gitblit v1.9.3