|
@@ -36,12 +36,22 @@ public class ProjectUserService extends ServiceImpl<ProjectUserMapper, ProjectUs
|
|
|
* @description 查询项目组成员
|
|
|
* @date 2019/12/13 2:40 PM
|
|
|
*/
|
|
|
- public List<ProjectUserResponse> userList(Long projectId, Long mtCompanyId) {
|
|
|
+ public List<ProjectUserResponse> userList(Long projectId, Long mtCompanyId, Long userId, String currentRoleCode) {
|
|
|
List<ProjectUserResponse> userList = getUserList(projectId, mtCompanyId);
|
|
|
- return userList.stream().filter(user -> (
|
|
|
- CommonEnum.DefaultRole.MTWORK.getCode().equals(user.getUserRole())
|
|
|
- || CommonEnum.DefaultRole.REGION_DIRECTOR.getCode().equals(user.getUserRole())
|
|
|
- )).sorted(Comparator.comparing(ProjectUserResponse::getUserRole)).collect(Collectors.toList());
|
|
|
+ List<ProjectUserResponse> workerList = new ArrayList<>();
|
|
|
+ if (CommonEnum.DefaultRole.MTWORK.getCode().equals(currentRoleCode)) {
|
|
|
+ userList.forEach(entry -> {
|
|
|
+ if (Objects.equals(userId, entry.getUserId())) {
|
|
|
+ workerList.add(entry);
|
|
|
+ }
|
|
|
+ });
|
|
|
+ return workerList;
|
|
|
+ } else {
|
|
|
+ return userList.stream().filter(
|
|
|
+ entry -> (CommonEnum.DefaultRole.MTWORK.getCode().equals(entry.getUserRole())
|
|
|
+ || CommonEnum.DefaultRole.REGION_DIRECTOR.getCode().equals(entry.getUserRole())))
|
|
|
+ .collect(Collectors.toList());
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
/**
|