From 07794736f879adb9cf02a69e5c6ada6a94b7bbde Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Mon, 06 Nov 2023 14:10:58 +0800
Subject: [PATCH] 考试报名查询修改,分新报名和考试未通过

---
 src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java |   43 +++++++++++++++++++++++++++++++++----------
 1 files changed, 33 insertions(+), 10 deletions(-)

diff --git a/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java b/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java
index 0853aba..b54ef0a 100644
--- a/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java
+++ b/src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java
@@ -752,18 +752,41 @@
 	 */
 	@Override
 	public List<TreeNode> getSecurityApplyTree(UserVO user) {
-		Map<Long, TreeNode> map = baseMapper.getSecurityApplyTree(user);
-		List<TreeNode> tree = new ArrayList<>();
-		if (map.size() > 1) {
-			map.forEach((id, treeNode) -> {
-				if (map.containsKey(treeNode.getParentId())) {
-					map.get(treeNode.getParentId()).getChildren().add(treeNode);
-				} else {
-					tree.add(treeNode);
+		// 总的
+		List<TreeNode> allList = new ArrayList<>();
+		// 考试未通过
+		List<TreeNode> notList = new ArrayList<>();
+		// 新申报报名的
+		List<TreeNode> newList = new ArrayList<>();
+		// 查询
+		List<TreeNode> treeNodeList = baseMapper.getSecurityApplyTree(user);
+		// 遍历
+		if (treeNodeList.size()>0){
+			for (TreeNode treeNode : treeNodeList) {
+				if (treeNode.getType()==1){
+					notList.add(treeNode);
 				}
-			});
+				if (treeNode.getType()==2){
+					newList.add(treeNode);
+				}
+			}
 		}
-		return tree;
+		if (newList.size()>0){
+			// 新申报报名的
+			TreeNode treeNode = new TreeNode();
+			treeNode.setName("新申报名单");
+			treeNode.setChildren(newList);
+			allList.add(treeNode);
+		}
+		if (notList.size()>0){
+			// 考试未通过
+			TreeNode treeNode = new TreeNode();
+			treeNode.setName("未通过考试名单");
+			treeNode.setChildren(notList);
+			allList.add(treeNode);
+		}
+		// 返回
+		return allList;
 	}
 
 	/**

--
Gitblit v1.9.3