From 2c90fcd0530de4fdd7efdb32b0a514d70e8061e0 Mon Sep 17 00:00:00 2001
From: 176yunxuan <362612387@qq.com~>
Date: Sat, 16 Mar 2024 16:42:34 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BF=94=E5=9B=9E=E4=BF=A1=E6=81=AF=E8=B0=83?=
 =?UTF-8?q?=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/jsl/oa/mapper/ProjectMapper.java |  2 ++
 .../oa/services/impl/ProjectServiceImpl.java  | 20 ++++++++++++++++++-
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/jsl/oa/mapper/ProjectMapper.java b/src/main/java/com/jsl/oa/mapper/ProjectMapper.java
index c9d84bc..ff59dad 100644
--- a/src/main/java/com/jsl/oa/mapper/ProjectMapper.java
+++ b/src/main/java/com/jsl/oa/mapper/ProjectMapper.java
@@ -110,4 +110,6 @@ public interface ProjectMapper {
 
    @Select("select * from organize_oa.oa_project_work where id=#{id}")
    ProjectWorkDO getWorkById(Integer id);
+@Select("select principal_id from organize_oa.oa_project_work where id=#{pid}")
+    Long getPirIdbyWorkid(Long pid);
 }
diff --git a/src/main/java/com/jsl/oa/services/impl/ProjectServiceImpl.java b/src/main/java/com/jsl/oa/services/impl/ProjectServiceImpl.java
index b3a0718..222b449 100644
--- a/src/main/java/com/jsl/oa/services/impl/ProjectServiceImpl.java
+++ b/src/main/java/com/jsl/oa/services/impl/ProjectServiceImpl.java
@@ -29,6 +29,7 @@ import javax.servlet.http.HttpServletRequest;
 import java.sql.Timestamp;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
 
 /**
  * <h1>项目服务层实现类</h1>
@@ -64,7 +65,24 @@ public class ProjectServiceImpl implements ProjectService {
     @Override
     public BaseResponse projecWorktAdd(HttpServletRequest request, ProjectWorkVO projectWorkVO) {
         log.info("\t> 执行 Service 层 ProjectService.projectWorkAdd 方法");
-        projectDAO.projectWorkAdd(projectWorkVO);
+        //获取用户id
+        Long userId = Processing.getAuthHeaderToUserId(request);
+        //是否是增加子系统
+        if(projectWorkVO.getType() == 0 ){
+            //是否是老师
+            if (Processing.checkUserIsTeacher(request, roleMapper)){
+                projectDAO.projectWorkAdd(projectWorkVO);
+            }else {
+                return ResultUtil.error(ErrorCode.NOT_PERMISSION);
+            }
+        }//增加子模块
+        else {
+            //是否是子系统的负责人
+            if(Objects.equals(userId, projectMapper.getPirIdbyWorkid(projectWorkVO.getPid()))){
+                projectDAO.projectWorkAdd(projectWorkVO);
+            }else return ResultUtil.error(ErrorCode.NOT_PERMISSION);
+        }
+
         return ResultUtil.success("添加成功");
     }