From aae9bc1ebc7f65cebaec65ac7128f79ca6c827d1 Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Wed, 07 Feb 2024 10:24:39 +0800
Subject: [PATCH] 代码优化

---
 src/main/java/org/springblade/common/cache/SysCache.java |   34 +++++++++++++++++++++-------------
 1 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/src/main/java/org/springblade/common/cache/SysCache.java b/src/main/java/org/springblade/common/cache/SysCache.java
index d816792..451b0ad 100644
--- a/src/main/java/org/springblade/common/cache/SysCache.java
+++ b/src/main/java/org/springblade/common/cache/SysCache.java
@@ -156,25 +156,33 @@
 	/**
 	 * 获取子部门ID集合
 	 *
-	 * @param deptId 主键
+	 * @param deptIds 主键
 	 * @return 子部门ID
 	 */
-	public static List<Long> getDeptChildIds(Long deptId) {
-		if (deptId == null) {
+	public static List<Long> getDeptChildIds(String deptIds) {
+		if (Strings.isBlank(deptIds)) {
 			return null;
 		}
-		List<Long> deptIdList = CacheUtil.get(SYS_CACHE, DEPT_CHILDIDS_ID, deptId, List.class);
-		if (deptIdList == null) {
-			deptIdList = new ArrayList<>();
-			List<Dept> deptChild = getDeptChild(deptId);
-			if (deptChild != null) {
-				List<Long> collect = deptChild.stream().map(Dept::getId).collect(Collectors.toList());
-				deptIdList.addAll(collect);
+		List<Long> list = new ArrayList<>();
+		List<String> deptIdList = new ArrayList<>(Arrays.asList(deptIds.split(",")));
+		for (String deptId : deptIdList) {
+			List<Long> deptIdLists = CacheUtil.get(SYS_CACHE, DEPT_CHILDIDS_ID, deptId, List.class);
+			if (deptIdLists == null) {
+				deptIdLists = new ArrayList<>();
+				List<Dept> deptChild = getDeptChild(Long.parseLong(deptId));
+				if (deptChild != null) {
+					List<Long> collect = deptChild.stream().map(Dept::getId).collect(Collectors.toList());
+					deptIdLists.addAll(collect);
+				}
+				deptIdLists.add(Long.parseLong(deptId));
+				CacheUtil.put(SYS_CACHE, DEPT_CHILDIDS_ID, deptId, deptIdLists);
 			}
-			deptIdList.add(deptId);
-			CacheUtil.put(SYS_CACHE, DEPT_CHILDIDS_ID, deptId, deptIdList);
+			list.addAll(deptIdLists);
 		}
-		return deptIdList;
+		// 去重
+		List<Long> collect = list.stream().distinct().collect(Collectors.toList());
+		// 返回
+		return collect;
 	}
 
 	/**

--
Gitblit v1.9.3