Parcourir la source

修改添加成员

yumeng il y a 5 ans
Parent
commit
8381a07896

+ 2 - 0
jeecg-boot-module-system/src/main/java/org/jeecg/modules/ctop/controller/ProjectController.java

@@ -253,11 +253,13 @@ public class ProjectController {
         SysUser sysUser = sysUserService.getById(project.getResponsibleId());
         if (!Check.isNull(sysUser)) {
             project.setResponsibleName(sysUser.getRealname());
+            projectMemberService.addMember(sysUser, project);
         }
 
         SysUser designSysUser = sysUserService.getById(project.getDesignResponsibleId());
         if (!Check.isNull(designSysUser)) {
             project.setDesignResponsibleName(designSysUser.getRealname());
+            projectMemberService.addMember(designSysUser, project);
         }
         Project projectEntity = projectService.getById(project.getId());
         if (projectEntity == null) {

+ 3 - 0
jeecg-boot-module-system/src/main/java/org/jeecg/modules/ctop/service/IProjectMemberService.java

@@ -1,7 +1,9 @@
 package org.jeecg.modules.ctop.service;
 
 import com.baomidou.mybatisplus.extension.service.IService;
+import org.jeecg.modules.ctop.entity.Project;
 import org.jeecg.modules.ctop.entity.ProjectMember;
+import org.jeecg.modules.system.entity.SysUser;
 
 /**
  * 项目成员
@@ -11,5 +13,6 @@ import org.jeecg.modules.ctop.entity.ProjectMember;
  * @date 2019-11-19
  */
 public interface IProjectMemberService extends IService<ProjectMember> {
+    void addMember(SysUser sysUser, Project project);
 
 }

+ 28 - 0
jeecg-boot-module-system/src/main/java/org/jeecg/modules/ctop/service/impl/ProjectMemberServiceImpl.java

@@ -1,11 +1,19 @@
 package org.jeecg.modules.ctop.service.impl;
 
+import cn.com.ctop.bytedance.mapper.MaterialReportMapper;
+import cn.com.ctop.common.module.utils.Check;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.jeecg.modules.ctop.entity.Project;
 import org.jeecg.modules.ctop.entity.ProjectMember;
 import org.jeecg.modules.ctop.mapper.ProjectMemberMapper;
 import org.jeecg.modules.ctop.service.IProjectMemberService;
+import org.jeecg.modules.system.entity.SysUser;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.HashMap;
+import java.util.Map;
+
 /**
  * 项目成员
  *
@@ -16,4 +24,24 @@ import org.springframework.stereotype.Service;
 @Service
 public class ProjectMemberServiceImpl extends ServiceImpl<ProjectMemberMapper, ProjectMember> implements IProjectMemberService {
 
+    @Autowired
+    private MaterialReportMapper materialReportMapper;
+
+    @Override
+    public void addMember(SysUser sysUser, Project project) {
+        Map<String, Object> map = new HashMap<>();
+        map.put("project_id", project.getId());
+        map.put("user_id", sysUser.getId());
+        this.removeByMap(map);
+        ProjectMember member = new ProjectMember();
+        member.setProjectId(project.getId());
+        member.setProjectName(project.getProjectName());
+        member.setUserId(sysUser.getId());
+        member.setUserName(sysUser.getRealname());
+        String roleCode = materialReportMapper.getRoleCodeByUserId(sysUser.getId());
+        if (!Check.isNull(roleCode)) {
+            member.setRoleCode(roleCode);
+        }
+        this.save(member);
+    }
 }