diff --git a/src/main/java/cn/palmte/work/service/ProjectService.java b/src/main/java/cn/palmte/work/service/ProjectService.java index a2a4386..5c6aabe 100644 --- a/src/main/java/cn/palmte/work/service/ProjectService.java +++ b/src/main/java/cn/palmte/work/service/ProjectService.java @@ -966,4 +966,12 @@ public class ProjectService { projectRepository.saveAndFlush(project); return ResponseMsg.buildSuccessMsg("保存成功"); } + + public List visibleProjects(List projectList, Admin admin) { + if (admin.getRoleType() == SysRole.ROLE_TYPE_SYSTEM) { + return projectList; + } + List projectIds = projectVisibleRepository.findProjectIdByTypeAndTid(ProjectVisible.TYPE_USER, admin.getId()); + return projectList.stream().filter(i -> i.getCreatorId() == admin.getId() || projectIds.contains(i.getId())).collect(Collectors.toList()); + } }