zh

src/main/java/com/zh/project0512/controller/TaskController.java

@@ -168,12 +168,52 @@ public class TaskController {
         return MybatisPlusUtil.sqlResult(taskBrandService.remove(Wrappers.<TaskBrand>query().lambda().eq(TaskBrand::getTaskId, param.getId()).eq(TaskBrand::getBrandId, param.getBrandId())), "删除");
     }
 
+    @Data
+    static class AddTaskReferenceParam {
+        @NotNull(message = "id不能为空")
+        @Min(value = 1, message = "id最小值为1")
+        @Schema(title = "任务id")
+        private Integer id;
+        @NotNull(message = "referenceId不能为空")
+        @Min(value = 1, message = "referenceId最小值为1")
+        @Schema(title = "素材Id")
+        private Integer referenceId;
+    }
+
+    @Operation(summary = "新增任务素材")
+    @PostMapping("/addReference")
+    public Result addBrand(@Validated @RequestBody AddTaskReferenceParam param) {
+        Integer taskId = param.getId();
+        Integer referenceId = param.getReferenceId();
+        if (null != taskReferenceService.getMap(Wrappers.<TaskReference>query().lambda().eq(TaskReference::getTaskId, taskId).eq(TaskReference::getReferenceId, referenceId))) {
+            return Result.success("数据已存在");
+        }
+        taskReferenceService.saveOrUpdate(new TaskReference().setTaskId(taskId).setReferenceId(referenceId).setCreatAt(LocalDateTime.now()));
+        return Result.success("添加完成");
+    }
+
+    @Data
+    static class DelTaskReferenceParam {
+        @NotNull(message = "id不能为空")
+        @Min(value = 1, message = "id最小值为1")
+        @Schema(title = "任务id")
+        private Integer id;
+        @NotNull(message = "referenceId不能为空")
+        @Min(value = 1, message = "referenceId最小值为1")
+        @Schema(title = "素材id")
+        private Integer referenceId;
+    }
+
+    @Operation(summary = "删除任务素材")
+    @PostMapping("/delReference")
+    public Result delTag(@Validated @RequestBody DelTaskReferenceParam param) {
+        return MybatisPlusUtil.sqlResult(taskReferenceService.remove(Wrappers.<TaskReference>query().lambda().eq(TaskReference::getTaskId, param.getId()).eq(TaskReference::getReferenceId, param.getReferenceId())), "删除");
+    }
+
     @Operation(summary = "更新")
     @PostMapping("/upd")
     public Result upd(@Validated(TaskValidGroup2.class) @RequestBody Task task) {
        task.setUpdateAt(LocalDateTime.now());
-//        UpdateWrapper<Task> updateWrapper = new UpdateWrapper<>();
-//        updateWrapper.eq("id", id);
         return MybatisPlusUtil.sqlResult(taskService.updateById(task), "修改");
     }