diff --git a/cloud-task-server/src/main/java/com/muyu/controller/TaskJoinController.java b/cloud-task-server/src/main/java/com/muyu/controller/TaskJoinController.java index e84a329..ba836af 100644 --- a/cloud-task-server/src/main/java/com/muyu/controller/TaskJoinController.java +++ b/cloud-task-server/src/main/java/com/muyu/controller/TaskJoinController.java @@ -1,8 +1,10 @@ package com.muyu.controller; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.muyu.common.core.domain.Result; import com.muyu.domain.TaskJoinInfo; import com.muyu.mapper.TaskJoinMapper; +import com.muyu.service.TaskJoinService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; @@ -18,15 +20,12 @@ import org.springframework.web.bind.annotation.*; public class TaskJoinController { @Autowired - private TaskJoinMapper taskJoinMapper; + private TaskJoinService taskJoinService; @PostMapping(value = "addTaskJoin") public Result addTaskJoin(@RequestBody TaskJoinInfo taskJoinInfo){ - int insert = taskJoinMapper.insert(taskJoinInfo); - if(insert<=0){ - throw new RuntimeException("失败"); - } - return Result.success(null,"添加成功"); + + return Result.success(null,taskJoinService.addTaskJoin(taskJoinInfo)); }; } diff --git a/cloud-task-server/src/main/java/com/muyu/service/TaskJoinService.java b/cloud-task-server/src/main/java/com/muyu/service/TaskJoinService.java new file mode 100644 index 0000000..f4c9511 --- /dev/null +++ b/cloud-task-server/src/main/java/com/muyu/service/TaskJoinService.java @@ -0,0 +1,9 @@ +package com.muyu.service; + +import com.baomidou.mybatisplus.extension.service.IService; +import com.muyu.domain.TaskJoinInfo; + +public interface TaskJoinService extends IService { + + String addTaskJoin(TaskJoinInfo taskJoinInfo); +} diff --git a/cloud-task-server/src/main/java/com/muyu/service/impl/TaskJoinServiceImpl.java b/cloud-task-server/src/main/java/com/muyu/service/impl/TaskJoinServiceImpl.java new file mode 100644 index 0000000..996d026 --- /dev/null +++ b/cloud-task-server/src/main/java/com/muyu/service/impl/TaskJoinServiceImpl.java @@ -0,0 +1,39 @@ +package com.muyu.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.muyu.domain.TaskJoinInfo; +import com.muyu.mapper.TaskJoinMapper; +import com.muyu.service.TaskJoinService; +import org.apache.ibatis.annotations.Mapper; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; + +/** + * @PackageName:com.muyu.service.impl + * @ClassName:TaskJoinServiceImpl + * @Description: + * @author: ¥陈思豪¥ + * @date: 2024/9/2 20:14 + */ +@Service +public class TaskJoinServiceImpl extends ServiceImpl implements TaskJoinService{ + + @Autowired + private TaskJoinMapper taskJoinMapper; + + @Override + public String addTaskJoin(TaskJoinInfo taskJoinInfo) { + LambdaQueryWrapper taskJoinInfoLambdaQueryWrapper = new LambdaQueryWrapper<>(); + taskJoinInfoLambdaQueryWrapper.eq(TaskJoinInfo::getNodeId,taskJoinInfo.getNodeId()); + TaskJoinInfo taskJoinInfo1 = taskJoinMapper.selectOne(taskJoinInfoLambdaQueryWrapper); + if(taskJoinInfo1!=null){ + taskJoinMapper.deleteById(taskJoinInfo1.getId()); + } + int insert = taskJoinMapper.insert(taskJoinInfo); + if(insert<=0){ + throw new RuntimeException("失败"); + } + return "添加成功"; + } +}