diff --git a/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java b/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java index f524451c..820f4334 100644 --- a/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java +++ b/src/main/java/cn/lihongjie/coal/dataCollector/controller/DataCollectorController.java @@ -31,6 +31,9 @@ public class DataCollectorController { return this.service.create(request); } + + + @PostMapping("/update") public DataCollectorDto update(@RequestBody UpdateDataCollectorDto request) { return this.service.update(request); @@ -42,6 +45,12 @@ public class DataCollectorController { return true; } + @PostMapping("/createQueue") + public Object createQueue(@RequestBody IdRequest request) { + this.service.createQueue(request); + return true; + } + @PostMapping("/getById") public DataCollectorDto getById(@RequestBody IdRequest request) { return this.service.getById(request.getId()); diff --git a/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java b/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java index a5248fa4..4b30f585 100644 --- a/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java +++ b/src/main/java/cn/lihongjie/coal/dataCollector/service/DataCollectorService.java @@ -83,7 +83,7 @@ public class DataCollectorService this.repository.deleteAllById(request.getIds()); } public void createQueue(IdRequest request) { - this.repository.deleteAllById(request.getIds()); + this.repository.findAllById(request.getIds()).forEach(this::createQueue); } public DataCollectorDto getById(String id) {