diff --git a/business-css/src/main/java/com/yfd/business/css/controller/ModelTrainController.java b/business-css/src/main/java/com/yfd/business/css/controller/ModelTrainController.java index 86ecc5b..f56f297 100644 --- a/business-css/src/main/java/com/yfd/business/css/controller/ModelTrainController.java +++ b/business-css/src/main/java/com/yfd/business/css/controller/ModelTrainController.java @@ -101,7 +101,8 @@ public class ModelTrainController { @GetMapping("/list") public ResponseResult list(@RequestParam(defaultValue = "1") Integer current, @RequestParam(defaultValue = "10") Integer size, - @RequestParam(required = false) String algoType, + @RequestParam(value = "algorithmType", required = false) String algorithmType, + @RequestParam(value = "algoType", required = false) String algoType, @RequestParam(required = false) String deviceType, @RequestParam(required = false) String status, @RequestParam(required = false) String name) { @@ -109,8 +110,9 @@ public class ModelTrainController { Page page = new Page<>(current, size); QueryWrapper query = new QueryWrapper<>(); - if (algoType != null && !algoType.isBlank()) { - query.eq("algorithm_type", algoType); + String finalAlgorithmType = (algorithmType != null && !algorithmType.isBlank()) ? algorithmType : algoType; + if (finalAlgorithmType != null && !finalAlgorithmType.isBlank()) { + query.eq("algorithm_type", finalAlgorithmType); } if (deviceType != null && !deviceType.isBlank()) { query.eq("device_type", deviceType);