소스 검색

普通管理员,部门只能看到自己

lyhzzz 2 년 전
부모
커밋
1c003198c7
1개의 변경된 파일16개의 추가작업 그리고 0개의 파일을 삭제
  1. 16 0
      src/main/java/com/fdkankan/fusion/service/impl/TmDepartmentServiceImpl.java

+ 16 - 0
src/main/java/com/fdkankan/fusion/service/impl/TmDepartmentServiceImpl.java

@@ -9,11 +9,15 @@ import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.fdkankan.fusion.common.ResultCode;
 import com.fdkankan.fusion.common.enums.IdPreEnum;
 import com.fdkankan.fusion.entity.TmDepartment;
+import com.fdkankan.fusion.entity.TmRole;
 import com.fdkankan.fusion.entity.TmUser;
+import com.fdkankan.fusion.entity.TmUserRole;
 import com.fdkankan.fusion.exception.BusinessException;
 import com.fdkankan.fusion.mapper.ITmDepartmentMapper;
 import com.fdkankan.fusion.service.ITmDepartmentService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.fdkankan.fusion.service.ITmRoleService;
+import com.fdkankan.fusion.service.ITmUserRoleService;
 import com.fdkankan.fusion.service.ITmUserService;
 import org.apache.catalina.User;
 import org.apache.commons.lang3.StringUtils;
@@ -39,6 +43,10 @@ public class TmDepartmentServiceImpl extends ServiceImpl<ITmDepartmentMapper, Tm
 
     @Autowired
     ITmUserService tmUserService;
+    @Autowired
+    ITmUserRoleService tmUserRoleService;
+    @Autowired
+    ITmRoleService tmRoleService;
 
     @Override
     public void insertDept(TmDepartment dept) {
@@ -98,6 +106,14 @@ public class TmDepartmentServiceImpl extends ServiceImpl<ITmDepartmentMapper, Tm
         if(tmUser == null){
             return new ArrayList<>();
         }
+        List<TmUserRole> tmUserRoles = tmUserRoleService.getByUserId(tmUser.getId());
+        if(tmUserRoles.size() >0){
+            TmUserRole tmUserRole = tmUserRoles.get(0);
+            TmRole tmRole = tmRoleService.getById(tmUserRole.getRoleId());
+            if(tmRole != null && tmRole.getRoleKey().equals("admin-ordinary")){
+                return Arrays.asList(this.getById(tmUser.getDeptId()));
+            }
+        }
         return getDeptList(tmUser.getDeptId());
     }