From f09e05514d9a9e2623cfa73c4de1ffa98bb30bf8 Mon Sep 17 00:00:00 2001
From: xiejun <xiejun@vci-tech.com>
Date: 星期二, 23 一月 2024 10:09:07 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 Source/UBCS/ubcs-service/ubcs-code/src/main/java/com/vci/ubcs/code/service/impl/CodeClassifyServiceImpl.java |   48 ++++++++++++++++++++++++++----------------------
 1 files changed, 26 insertions(+), 22 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 ad4b63b..aced92e 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
@@ -71,6 +71,7 @@
 import java.awt.event.ItemEvent;
 import java.io.File;
 import java.util.*;
+import java.util.concurrent.ForkJoinPool;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -116,6 +117,7 @@
 
 	@Resource
 	private CodeBasicSecMapper codeBasicSecMapper;
+
 	/**
 	 * 涓婄骇鑺傜偣鐨勫睘鎬у悕绉�
 	 */
@@ -510,7 +512,7 @@
 	 * @return 涓婚搴撳垎绫绘樉绀烘爲
 	 */
 	@Override
-	public List<Tree> referTree(TreeQueryObject treeQueryObject) {
+	public List<Tree> referTree(TreeQueryObject treeQueryObject) throws ServiceException{
 		if(treeQueryObject == null){
 			treeQueryObject = new TreeQueryObject();
 		}
@@ -520,7 +522,7 @@
 		if(treeQueryObject.getConditionMap().containsKey(LC_STATUS)) {
 			treeQueryObject.getConditionMap().remove(LC_STATUS);
 		}
-		if(treeQueryObject.getExtandParamsMap() ==null || !treeQueryObject.getExtandParamsMap().containsKey(REFER_SHOW_DISABLED_QUERY_KEY)) {
+		if(treeQueryObject.getExtandParamsMap() == null || !treeQueryObject.getExtandParamsMap().containsKey(REFER_SHOW_DISABLED_QUERY_KEY)) {
 		}
 		treeQueryObject.getConditionMap().put(LC_STATUS, FRAMEWORK_DATA_ENABLED);
 		return treeCodeClassify(treeQueryObject);
@@ -550,7 +552,7 @@
 	 * @return 涓婚搴撳垎绫� 鏄剧ず鏍�
 	 */
 	@Override
-	public List<Tree> treeCodeClassify(TreeQueryObject treeQueryObject) {
+	public List<Tree> treeCodeClassify(TreeQueryObject treeQueryObject) throws ServiceException{
 		List<CodeClassify> doList = null;
 		String id = null;
 		String lcStatus = null;
@@ -578,33 +580,35 @@
 		List<CodeClassifyVO> voList = codeClassifyDO2VOs(doList);
 		TreeWrapperOptions treeWrapperOptions = new TreeWrapperOptions(PARENT_FIELD_NAME);
 		treeWrapperOptions.copyFromTreeQuery(treeQueryObject);
-		List<Tree> tree= revisionModelUtil.doList2Trees(voList,treeWrapperOptions,(CodeClassifyVO s) ->{
+		List<Tree> tree = revisionModelUtil.doList2Trees(voList,treeWrapperOptions,(CodeClassifyVO s) ->{
 			//鍙互鍦ㄨ繖閲屽鐞嗘爲鑺傜偣鐨勬樉绀�
 			return s.getId() + " " + s.getName() + (FrameworkDataLCStatus.DISABLED.getValue().equalsIgnoreCase(s
 				.getLcStatus()) ? (" 銆愬仠鐢ㄣ�� ") : "");
 		});
-		//瓒呯鏄剧ず鎵�鏈夊垎绫�
-		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 {
+		// 闈炶秴绠¤繃婊ゆ湭鎺堟潈鐨勫垎绫�
+		if(!VciBaseUtil.checkAdminTenant()){
 			// 閭d簺鍒嗙被鍏峰鏌ョ湅鏉冮檺
-			R<List<String>> viewClassByRoleIds = sysClient.getViewClassByRoleIds(Arrays.asList(AuthUtil.getUser().getRoleId().split(",")));
+			R<List<String>> viewClassByRoleIds = sysClient.getViewClassByRoleIds(
+				Arrays.asList(AuthUtil.getUser().getRoleId().split(",")),
+				treeQueryObject.getConditionMap().getOrDefault("authType","classify_auth"),
+				treeQueryObject.getConditionMap().getOrDefault("buttonCode","classify_view"),
+				treeQueryObject.getConditionMap().getOrDefault("menuCode","classifyTree")
+			);
 			// 璇锋眰澶辫触鎴栬�呰姹傚緱鍒扮殑鍏峰鏌ョ湅鏉冮檺鐨勫垎绫籭d闆嗗悎涓虹┖
-			if(!viewClassByRoleIds.isSuccess() && !viewClassByRoleIds.getData().isEmpty()){
-				return new ArrayList<>();
+			if(!viewClassByRoleIds.isSuccess() && viewClassByRoleIds.getData().isEmpty()){
+				throw new ServiceException("涓绘暟鎹煡鐪嬫潈闄愭湭閰嶇疆锛屾垨閰嶇疆鏈夎锛�");
 			}
+			long startTime = System.currentTimeMillis();
 			// 杩囨护
 			filterTreeNodes(tree,viewClassByRoleIds.getData());
+			long endTime = System.currentTimeMillis();
+			System.out.println(("鎵ц鏃堕棿锛�"+(endTime-startTime)/1000)+"s");
 		}
+		// 鍔犺浇鍒嗙被鏄惁鍏锋湁瀛愬垎绫�
+		tree.parallelStream().forEach(item -> {
+			boolean checkHasChild = checkHasChild(item.getOid());
+			item.setLeaf(!checkHasChild);
+		});
 		return tree;
 	}
 
@@ -617,8 +621,8 @@
 		Iterator<Tree> iterator = trees.iterator();
 		while (iterator.hasNext()) {
 			Tree tree = iterator.next();
-			Boolean checkHasChild = codeClassifyMapper.checkHasChild(tree.getOid());
-			tree.setLeaf(!checkHasChild);
+			/*Boolean checkHasChild = checkHasChild(tree.getOid());
+			tree.setLeaf(!checkHasChild);*/
 			if (classifyIds.contains(tree.getOid())) {
 				// 濡傛灉椤跺眰鑺傜偣瀛樺湪浜� classifyIds 涓紝鐩存帴淇濈暀鍏跺瓙鑺傜偣闆嗗悎
 				continue;

--
Gitblit v1.9.3