From 83d95b122720adf9ef365bb0f88036f96196c249 Mon Sep 17 00:00:00 2001
From: ArgonarioD <arnordorian@sina.com>
Date: Tue, 5 Jul 2022 20:47:17 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9=E4=BA=86ProjectGroup?=
 =?UTF-8?q?=E7=9B=B8=E5=85=B3=E7=9A=84=E4=B8=9C=E8=A5=BF=EF=BC=8C=E9=80=82?=
 =?UTF-8?q?=E9=85=8D=E4=BA=86=E6=96=B0=E7=9A=84=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E7=BB=93=E6=9E=84=E5=8F=98=E5=8A=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../entity/ProjectGroup.java                  |  1 +
 .../service/IProjectGroupService.java         |  6 +++-
 .../service/impl/ProjectGroupServiceImpl.java | 29 ++++++++-----------
 3 files changed, 18 insertions(+), 18 deletions(-)

diff --git a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/entity/ProjectGroup.java b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/entity/ProjectGroup.java
index 73abb2d..e897d83 100644
--- a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/entity/ProjectGroup.java
+++ b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/entity/ProjectGroup.java
@@ -14,4 +14,5 @@ public class ProjectGroup {
     Integer staffId;
     Integer projectId;
     String projectStaffPosition;
+    Integer projectAccessLevel;
 }
diff --git a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/IProjectGroupService.java b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/IProjectGroupService.java
index d6fc875..225671d 100644
--- a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/IProjectGroupService.java
+++ b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/IProjectGroupService.java
@@ -9,5 +9,9 @@ import com.baomidou.mybatisplus.extension.service.IService;
  */
 public interface IProjectGroupService extends IService<ProjectGroup> {
     Boolean addCreator(Integer projectId, Integer staffId);
-    Integer getUserLevelInGroup(String token,Integer projectId);
+
+    /**
+     * @return 如果不存在就返回0，否则返回AccessLevel
+     */
+    Integer getUserLevelInGroup(String token, Integer projectId);
 }
\ No newline at end of file
diff --git a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/impl/ProjectGroupServiceImpl.java b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/impl/ProjectGroupServiceImpl.java
index 1687e5c..3fd794b 100644
--- a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/impl/ProjectGroupServiceImpl.java
+++ b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/service/impl/ProjectGroupServiceImpl.java
@@ -1,7 +1,6 @@
 package cn.edu.hfut.rmdjzz.projectmanagement.service.impl;
 
 import cn.edu.hfut.rmdjzz.projectmanagement.entity.ProjectGroup;
-import cn.edu.hfut.rmdjzz.projectmanagement.entity.Staff;
 import cn.edu.hfut.rmdjzz.projectmanagement.mapper.ProjectGroupMapper;
 import cn.edu.hfut.rmdjzz.projectmanagement.service.IProjectGroupService;
 import cn.edu.hfut.rmdjzz.projectmanagement.utils.TokenUtils;
@@ -9,8 +8,6 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springframework.stereotype.Service;
 
-import java.io.Serializable;
-
 /**
  * @author 张韬
  * created at 2022/6/30 21:38
@@ -24,25 +21,23 @@ public class ProjectGroupServiceImpl extends ServiceImpl<ProjectGroupMapper, Pro
         ProjectGroup projectGroup = new ProjectGroup();
         projectGroup.setProjectId(projectId);
         projectGroup.setStaffId(staffId);
-        projectGroup.setProjectStaffPosition("负责人");
+        projectGroup.setProjectStaffPosition("项目经理");
+        projectGroup.setProjectAccessLevel(1);
         return baseMapper.insert(projectGroup) == 1;
     }
 
-    //TODO:
     @Override
-    public Integer getUserLevelInGroup(String token,Integer projectId){
-        try{
-            Integer staffId= TokenUtils.getStaffId(token);
-            ProjectGroup projectGroup=baseMapper.selectOne(Wrappers.<ProjectGroup>lambdaQuery().eq(ProjectGroup::getStaffId, staffId).eq(ProjectGroup::getProjectId, projectId));
-            if(projectGroup == null)
+    public Integer getUserLevelInGroup(String token, Integer projectId) {
+        try {
+            Integer staffId = TokenUtils.getStaffId(token);
+            ProjectGroup projectGroup = baseMapper.selectOne(Wrappers.<ProjectGroup>lambdaQuery()
+                    .eq(ProjectGroup::getStaffId, staffId)
+                    .eq(ProjectGroup::getProjectId, projectId)
+            );
+            if (projectGroup == null)
                 return 0;
-            else if(projectGroup.getProjectStaffPosition().equals("负责人"))
-                return 3;
-            else if(projectGroup.getProjectStaffPosition().length()>0)
-                return 1;
-            else
-                return 0;
-        }catch(Exception e){//需要调整
+            return projectGroup.getProjectAccessLevel();
+        } catch (Exception e) {
             return 0;
         }
     }