diff --git a/cloud-task-server/src/main/java/com/muyu/task/server/controller/NodeTableController.java b/cloud-task-server/src/main/java/com/muyu/task/server/controller/NodeTableController.java index ffe39e7..da594a2 100644 --- a/cloud-task-server/src/main/java/com/muyu/task/server/controller/NodeTableController.java +++ b/cloud-task-server/src/main/java/com/muyu/task/server/controller/NodeTableController.java @@ -1,5 +1,6 @@ package com.muyu.task.server.controller; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.muyu.common.core.domain.Result; import com.muyu.common.domian.NodeTable; import com.muyu.task.server.service.NodeTableService; @@ -13,7 +14,13 @@ public class NodeTableController { @Autowired private NodeTableService nodeTableService; - + @DeleteMapping + private Result delete(@PathVariable("id") Long id) { + QueryWrapper wrapper = new QueryWrapper<>(); + wrapper.eq("node_id", id); + boolean save = nodeTableService.remove(wrapper); + return Result.success(save); + } @PostMapping private Result addNodeTable(@RequestBody NodeTable nodeTable) { boolean save = nodeTableService.save(nodeTable); @@ -21,7 +28,9 @@ public class NodeTableController { } @GetMapping("/{id}") private Result getNodeTable(@PathVariable("id") Long id) { - NodeTable byId = nodeTableService.getById(id); + QueryWrapper wrapper = new QueryWrapper<>(); + wrapper.eq("node_id", id); + NodeTable byId = nodeTableService.getOne(wrapper); return Result.success(byId); } diff --git a/cloud-task-server/src/main/java/com/muyu/task/server/controller/TaskInputController.java b/cloud-task-server/src/main/java/com/muyu/task/server/controller/TaskInputController.java index 5ee672d..989c2ab 100644 --- a/cloud-task-server/src/main/java/com/muyu/task/server/controller/TaskInputController.java +++ b/cloud-task-server/src/main/java/com/muyu/task/server/controller/TaskInputController.java @@ -1,5 +1,6 @@ package com.muyu.task.server.controller; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.muyu.common.core.domain.Result; import com.muyu.common.domian.Structure; @@ -93,7 +94,9 @@ public class TaskInputController { @DeleteMapping("/{id}") @Operation(summary = "删除任务", description = "通过任务id删除任务信息") public Result delete(@PathVariable("id") Long id) { - taskInputService.removeById(id); + QueryWrapper taskInputQueryWrapper = new QueryWrapper<>(); + taskInputQueryWrapper.eq("node_id",id); + taskInputService.remove(taskInputQueryWrapper); return Result.success(null, "操作成功"); }