diff --git a/src/main/java/com/genersoft/iot/vmp/aiot/controller/AiAlgorithmController.java b/src/main/java/com/genersoft/iot/vmp/aiot/controller/AiAlgorithmController.java index 794303ce1..e4b2f0eb4 100644 --- a/src/main/java/com/genersoft/iot/vmp/aiot/controller/AiAlgorithmController.java +++ b/src/main/java/com/genersoft/iot/vmp/aiot/controller/AiAlgorithmController.java @@ -1,5 +1,6 @@ package com.genersoft.iot.vmp.aiot.controller; +import com.alibaba.fastjson2.JSON; import com.genersoft.iot.vmp.aiot.bean.AiAlgorithm; import com.genersoft.iot.vmp.aiot.bean.AiRoiAlgoBind; import com.genersoft.iot.vmp.aiot.service.IAiAlgorithmService; @@ -65,6 +66,14 @@ public class AiAlgorithmController { @PostMapping("/global-params/{algoCode}") public void saveGlobalParams(@PathVariable String algoCode, @RequestBody Map body) { String globalParams = body.get("globalParams"); + if (globalParams == null || globalParams.isBlank()) { + throw new IllegalArgumentException("参数不能为空"); + } + try { + JSON.parseObject(globalParams); + } catch (Exception e) { + throw new IllegalArgumentException("参数不是合法的 JSON: " + e.getMessage()); + } algorithmService.saveGlobalParams(algoCode, globalParams); } @@ -80,6 +89,14 @@ public class AiAlgorithmController { @PathVariable String algoCode, @RequestBody Map body) { String params = body.get("params"); + if (params == null || params.isBlank()) { + throw new IllegalArgumentException("参数不能为空"); + } + try { + JSON.parseObject(params); + } catch (Exception e) { + throw new IllegalArgumentException("参数不是合法的 JSON: " + e.getMessage()); + } algorithmService.updateDeviceAlgoParams(deviceId, algoCode, params); } }