diff --git a/src/main/java/com/zh/project0512/controller/TaskController.java b/src/main/java/com/zh/project0512/controller/TaskController.java index a216cfa..9426d3c 100644 --- a/src/main/java/com/zh/project0512/controller/TaskController.java +++ b/src/main/java/com/zh/project0512/controller/TaskController.java @@ -168,12 +168,52 @@ public class TaskController { return MybatisPlusUtil.sqlResult(taskBrandService.remove(Wrappers.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.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.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 updateWrapper = new UpdateWrapper<>(); -// updateWrapper.eq("id", id); return MybatisPlusUtil.sqlResult(taskService.updateById(task), "修改"); }