From 63d93f64e7643624a4d078c0ac3ec3d465897b11 Mon Sep 17 00:00:00 2001
From: ludc
Date: 星期五, 29 十二月 2023 13:05:23 +0800
Subject: [PATCH] 分类授权:主题库定义按钮授权相关接口提交

---
 Source/UBCS/ubcs-service/ubcs-code/src/main/java/com/vci/ubcs/code/service/impl/CodeClassifyServiceImpl.java |   83 ++++++++++++++++++++++++++++++++++++-----
 1 files changed, 73 insertions(+), 10 deletions(-)

diff --git a/Source/UBCS/ubcs-service/ubcs-code/src/main/java/com/vci/ubcs/code/service/impl/CodeClassifyServiceImpl.java b/Source/UBCS/ubcs-service/ubcs-code/src/main/java/com/vci/ubcs/code/service/impl/CodeClassifyServiceImpl.java
index e021231..f2d1d0b 100644
--- a/Source/UBCS/ubcs-service/ubcs-code/src/main/java/com/vci/ubcs/code/service/impl/CodeClassifyServiceImpl.java
+++ b/Source/UBCS/ubcs-service/ubcs-code/src/main/java/com/vci/ubcs/code/service/impl/CodeClassifyServiceImpl.java
@@ -49,6 +49,7 @@
 import com.vci.ubcs.starter.web.util.LangBaseUtil;
 import com.vci.ubcs.starter.web.util.VciBaseUtil;
 import com.vci.ubcs.system.cache.NacosConfigCache;
+import com.vci.ubcs.system.feign.ISysClient;
 import org.apache.poi.hssf.util.HSSFColor;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -92,7 +93,7 @@
 	private IBtmTypeClient btmTypeClient;
 
 	@Resource
-	private CommonsMapper commonsMapper;
+	private ISysClient sysClient;
 
 	/**
 	 * 鏃ュ織
@@ -582,18 +583,67 @@
 			return s.getId() + " " + s.getName() + (FrameworkDataLCStatus.DISABLED.getValue().equalsIgnoreCase(s
 				.getLcStatus()) ? (" 銆愬仠鐢ㄣ�� ") : "");
 		});
-
-		Iterator var6 =  tree.listIterator();
-		while(var6.hasNext()){
-			Tree trees = (Tree) var6.next();
-			boolean checkHasChild=codeClassifyMapper.checkHasChild(trees.getOid());
-			if(checkHasChild){
-				trees.setLeaf(false);
-			}else{
-				trees.setLeaf(true);
+		//瓒呯鏄剧ず鎵�鏈夊垎绫�
+		if(VciBaseUtil.checkAdminTenant()){
+			Iterator var6 =  tree.listIterator();
+			while(var6.hasNext()){
+				Tree trees = (Tree) var6.next();
+				boolean checkHasChild=codeClassifyMapper.checkHasChild(trees.getOid());
+				if(checkHasChild){
+					trees.setLeaf(false);
+				}else{
+					trees.setLeaf(true);
+				}
 			}
+		}else {
+			// 閭d簺鍒嗙被鍏峰鏌ョ湅鏉冮檺
+			R<List<String>> viewClassByRoleIds = sysClient.getViewClassByRoleIds(Arrays.asList(AuthUtil.getUser().getRoleId().split(",")));
+			// 璇锋眰澶辫触鎴栬�呰姹傚緱鍒扮殑鍏峰鏌ョ湅鏉冮檺鐨勫垎绫籭d闆嗗悎涓虹┖
+			if(!viewClassByRoleIds.isSuccess() && !viewClassByRoleIds.getData().isEmpty()){
+				return new ArrayList<>();
+			}
+			// 杩囨护
+			filterTreeNodes(tree,viewClassByRoleIds.getData());
 		}
 		return tree;
+	}
+
+	/**
+	 * 鍒嗙被鎺堟潈杩囨护鎺夋病鏈夋潈闄愮殑鍒嗙被
+	 * @param trees
+	 * @param classifyIds
+	 */
+	private void filterTreeNodes(List<Tree> trees, List<String> classifyIds) {
+		Iterator<Tree> iterator = trees.iterator();
+		while (iterator.hasNext()) {
+			Tree tree = iterator.next();
+			Boolean checkHasChild = codeClassifyMapper.checkHasChild(tree.getOid());
+			tree.setLeaf(!checkHasChild);
+			if (classifyIds.contains(tree.getOid())) {
+				// 濡傛灉椤跺眰鑺傜偣瀛樺湪浜� classifyIds 涓紝鐩存帴淇濈暀鍏跺瓙鑺傜偣闆嗗悎
+				continue;
+			}
+			if (tree.getChildren() != null && !tree.getChildren().isEmpty()) {
+				filterTreeNodes(tree.getChildren(), classifyIds);
+			}
+			if (!hasMatchingChild(tree, classifyIds)) {
+				iterator.remove();
+			}
+		}
+	}
+
+	private boolean hasMatchingChild(Tree tree, List<String> classifyIds) {
+		if (classifyIds.contains(tree.getOid())) {
+			return true;
+		}
+		if (tree.getChildren() != null) {
+			for (Tree child : tree.getChildren()) {
+				if (hasMatchingChild(child, classifyIds)) {
+					return true;
+				}
+			}
+		}
+		return false;
 	}
 
 	/**
@@ -1408,6 +1458,19 @@
 	}
 
 	/**
+	 * 鏌ヨ鎵�鏈変笂灞傜埗鑺傜偣鐨刼id
+	 * @param oid
+	 * @return
+	 */
+	@Override
+	public List<String> selectAllParentOid(String oid){
+		if(Func.isBlank(oid)){
+			return new ArrayList<>();
+		}
+		return this.codeClassifyMapper.selectAllParentOid(oid);
+	}
+
+	/**
 	 * 浣跨敤鍒嗙被涓婚敭鑾峰彇鍒嗙被鐩稿叧鐨勬墍鏈変俊鎭�
 	 *
 	 * @param codeClassifyOid 鍒嗙被鐨勪富閿�

--
Gitblit v1.9.3