From 5b6bd5990e4f4a47fe11f5e63ffc2292044ff82a Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Mon, 18 Dec 2023 11:46:51 +0800
Subject: [PATCH] 处理当前部门用户

---
 src/main/java/org/springblade/modules/evaluate/controller/EvaluateTaskController.java |   38 ++++++++++++++++++++++++++------------
 1 files changed, 26 insertions(+), 12 deletions(-)

diff --git a/src/main/java/org/springblade/modules/evaluate/controller/EvaluateTaskController.java b/src/main/java/org/springblade/modules/evaluate/controller/EvaluateTaskController.java
index daca43a..1b4eb08 100644
--- a/src/main/java/org/springblade/modules/evaluate/controller/EvaluateTaskController.java
+++ b/src/main/java/org/springblade/modules/evaluate/controller/EvaluateTaskController.java
@@ -49,6 +49,8 @@
 import springfox.documentation.annotations.ApiIgnore;
 import java.util.Map;
 import java.util.List;
+import java.util.Set;
+import java.util.stream.Collectors;
 import javax.servlet.http.HttpServletResponse;
 
 /**
@@ -93,20 +95,32 @@
 		wrapper.or().like("candidate_num",deptId).or().like("candidate_num",userId);
 		IPage<EvaluateTaskEntity> pages = evaluateTaskService.page(Condition.getPage(query), wrapper);
 
-		// 处理获取当前部门的用户
+		// 处理获取当前部门的用户 type为0时
 		List<EvaluateTaskVO> records = EvaluateTaskWrapper.build().listVO(pages.getRecords());
 		for (EvaluateTaskVO vo: records) {
-			List<CandidateJsonObj> candidateJsonObjs = JSONArray.parseArray(JSON.toJSONString(vo.getCandidateNum()),CandidateJsonObj.class);
-			for (CandidateJsonObj obj:candidateJsonObjs) {
-				if (deptId.equals(obj.getDeptId()) && obj.getUsers().size() > 0) {
-					vo.setSelfCandidate(obj.getUsers());
-					break;
-				} else if(deptId.equals(obj.getDeptId()) && obj.getUsers().size() == 0) {
-					QueryWrapper<User> userQueryWrapper = new QueryWrapper<>();
-					userQueryWrapper.like("dept_id", deptId);
-					List<User> users = userService.list(userQueryWrapper);
-					vo.setSelfCandidate(users);
-					break;
+			if (vo.getType() == 0) {
+				List<CandidateJsonObj> candidateJsonObjs = JSONArray.parseArray(JSON.toJSONString(vo.getCandidateNum()),CandidateJsonObj.class);
+				QueryWrapper<User> userQueryWrapper = new QueryWrapper<>();
+				userQueryWrapper.like("dept_id", deptId);
+				List<User> users = userService.list(userQueryWrapper);
+
+				for (CandidateJsonObj obj:candidateJsonObjs) {
+					if (deptId.equals(obj.getDeptId()) && obj.getUsers().size() > 0) {
+
+						//处理自己部门用户
+						List<Object> commonItems = obj.getUsers().stream()
+							.filter(item1 -> users.stream()
+								.anyMatch(item2 -> item1.getId().equals(item2.getId())))
+							.collect(Collectors.toList());
+
+						vo.setSelfCandidate(commonItems);
+
+						break;
+					} else if(deptId.equals(obj.getDeptId()) && obj.getUsers().size() == 0) {
+
+						vo.setSelfCandidate(users);
+						break;
+					}
 				}
 			}
 

--
Gitblit v1.9.3