Просмотр исходного кода

删除项目一并删除项目团队

lyhzzz 2 лет назад
Родитель
Сommit
2885cfd601

+ 4 - 0
src/main/java/com/fdkankan/site/controller/ProjectController.java

@@ -15,6 +15,7 @@ import com.fdkankan.site.request.ProjectDelSceneParam;
 import com.fdkankan.site.request.ProjectListParam;
 import com.fdkankan.site.service.IProjectNumService;
 import com.fdkankan.site.service.IProjectService;
+import com.fdkankan.site.service.IProjectTeamService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
@@ -34,6 +35,8 @@ public class ProjectController extends BaseController {
     IProjectService projectService;
     @Autowired
     IProjectNumService projectNumService;
+    @Autowired
+    IProjectTeamService projectTeamService;
 
     @GetMapping("/userData")
     public ResultData userData(){
@@ -59,6 +62,7 @@ public class ProjectController extends BaseController {
         VUtils.isTure(param.getProjectId() == null).throwMessage(ResultCode.PARAM_MISS);
         projectService.removeById(param.getProjectId());
         projectNumService.delByProjectId(param.getProjectId());
+        projectTeamService.delByProjectId(param.getProjectId());
         return ResultData.ok();
     }
 

+ 2 - 0
src/main/java/com/fdkankan/site/service/IProjectTeamService.java

@@ -41,4 +41,6 @@ public interface IProjectTeamService extends IService<ProjectTeam> {
     void updateRemark(ProjectTeam param);
 
     Set<Integer> getProjectIdByCreater(String projectCreater);
+
+    void delByProjectId(Integer projectId);
 }

+ 1 - 1
src/main/java/com/fdkankan/site/service/impl/ProjectServiceImpl.java

@@ -71,7 +71,7 @@ public class ProjectServiceImpl extends ServiceImpl<IProjectMapper, Project> imp
             LambdaQueryWrapper<Project> wrapper = new LambdaQueryWrapper<>();
             wrapper.in(Project::getProjectId,projectIds);
             wrapper.eq(Project::getProjectStatus,1);
-             count = this.count(wrapper);
+            count = this.count(wrapper);
         }
 
         userData.setProjectCount((long) projectIds.size());

+ 7 - 0
src/main/java/com/fdkankan/site/service/impl/ProjectTeamServiceImpl.java

@@ -148,4 +148,11 @@ public class ProjectTeamServiceImpl extends ServiceImpl<IProjectTeamMapper, Proj
         wrapper.eq(ProjectTeam::getIsCreater,1);
         return this.list(wrapper).stream().map(ProjectTeam::getProjectId).collect(Collectors.toSet());
     }
+
+    @Override
+    public void delByProjectId(Integer projectId) {
+        LambdaQueryWrapper<ProjectTeam> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(ProjectTeam::getProjectId,projectId);
+        this.remove(wrapper);
+    }
 }