From 02e7ca80d849ad8a99974bd08f9e68a78f36d601 Mon Sep 17 00:00:00 2001
From: ludc
Date: 星期二, 16 一月 2024 21:51:45 +0800
Subject: [PATCH] 分类授权数据授权界面编写与接口修改

---
 Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java |   98 ++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 77 insertions(+), 21 deletions(-)

diff --git a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java b/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java
index 50fdfa6..9139a1f 100644
--- a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java
+++ b/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java
@@ -21,18 +21,17 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.vci.ubcs.code.feign.ICodeClassifyClient;
 import com.vci.ubcs.starter.util.MybatisParameterUtil;
 import com.vci.ubcs.starter.web.util.VciBaseUtil;
 import com.vci.ubcs.system.cache.NacosConfigCache;
 import com.vci.ubcs.system.cache.SysCache;
 import com.vci.ubcs.system.dto.MenuDTO;
 import com.vci.ubcs.system.entity.*;
+import com.vci.ubcs.system.mapper.ClassifyAuthMapper;
+import com.vci.ubcs.system.service.*;
 import com.vci.ubcs.system.vo.MenuVO;
 import com.vci.ubcs.system.mapper.MenuMapper;
-import com.vci.ubcs.system.service.IMenuService;
-import com.vci.ubcs.system.service.IRoleMenuService;
-import com.vci.ubcs.system.service.IRoleScopeService;
-import com.vci.ubcs.system.service.ITopMenuSettingService;
 import com.vci.ubcs.system.wrapper.MenuWrapper;
 import lombok.AllArgsConstructor;
 import org.springblade.core.log.exception.ServiceException;
@@ -70,6 +69,8 @@
 	private final IRoleScopeService roleScopeService;
 	private final MenuMapper menuMapper;
 	private final ITopMenuSettingService topMenuSettingService;
+	private final ICodeClassifyClient codeClassifyClient;
+	private final ClassifyAuthMapper classifyAuthMapper;
 	private final static String PARENT_ID = "parentId";
 	private final static Integer MENU_CATEGORY = 1;
 
@@ -324,18 +325,63 @@
 	}
 
 	/**
-	 * 鑾峰彇鑿滃崟涓嬮潰鐨勬寜閽�
+	 * 鑾峰彇鑿滃崟涓嬮潰鐨勬寜閽紝鍒贡璋冪敤锛岃繖涓柟娉曟槸閽堝涓绘暟鎹鐞嗘寜閽煡璇㈢殑
+	 * @param classifyId
 	 * @param btmType 涓氬姟绫诲瀷
-	 * @return List<Menu>
+	 * @param authType
+	 * @return
 	 */
     @Override
 	//@Cacheable(cacheNames = MENU_CACHE, key = "'auth:menuButton:'+ #btmType +':'+ #userId ")
-    public List<Menu> getMenuButtonByType(String btmType,Long userId) {
-		List<String> roleIds = null;
-		if(!VciBaseUtil.checkAdminTenant()){
-			roleIds = Arrays.asList(AuthUtil.getUser().getRoleId().split(","));
+    public List<Menu> getMenuButtonByType(String classifyId,String btmType,String authType) {
+		// baseMapper.selectMenuChildByBtnType(btmType,roleIds);
+		if(Func.isBlank(classifyId)){
+			throw new ServiceException("蹇呬紶鍙傛暟鍒嗙被oid涓嶈兘涓虹┖锛�");
 		}
-		return baseMapper.selectMenuChildByBtnType(btmType,roleIds);
+		//鏌ヨ鍒嗙被鑺傜偣鐨勬墍鏈夌埗绾ц妭鐐�
+		R<List<String>> listR = codeClassifyClient.selectAllParentOid(classifyId);
+		if (!listR.isSuccess() && !listR.getData().isEmpty()) {
+			throw new ServiceException("鑾峰彇鍒嗙被淇℃伅澶辫触锛�");
+		}
+		// 杩斿洖鐨勫垎绫籵id鏄綋鍓嶈妭鐐逛负绗竴涓紝鍚庨潰渚濇鏄粬鐨勪笂灞傝妭鐐�
+		List<String> classifyOidList = listR.getData();
+		final String roleIds = AuthUtil.getUser().getRoleId();
+		// 鍏堟煡璇㈡寜閽甶d鍒楄〃
+		LambdaQueryWrapper<ClassifyAuth> wrapper = Wrappers.<ClassifyAuth>query()
+			.lambda().eq(ClassifyAuth::getClassifyId, classifyId)
+			.eq(ClassifyAuth::getAuthType,authType)
+			.in(ClassifyAuth::getRoleId, roleIds);
+		List<ClassifyAuth> classifyAuths = classifyAuthMapper.selectList(wrapper);
+		//濡傛灉褰撳墠鍒嗙被娌℃湁鎵惧埌鎺堟潈閰嶇疆锛屽氨渚濇浠庡綋鍓嶈妭鐐瑰線涓婂眰鑺傜偣鎵炬巿鏉冮厤缃紝鎵惧埌浜嗗氨鍋滄锛屾病鎵惧埌灏变竴鐩存壘鍒版渶鍚�
+		if(classifyAuths.isEmpty()){
+			// 涓嬫爣浠�1寮�濮嬪洜涓哄綋鍓嶈妭鐐�0宸茬粡鏌ヨ杩�
+			for (int i = 1; i < classifyOidList.size(); i++) {
+				classifyAuths = classifyAuthMapper.selectList(
+					Wrappers.<ClassifyAuth>query()
+						.lambda().eq(ClassifyAuth::getClassifyId, classifyOidList.get(i))
+						.in(ClassifyAuth::getRoleId, roleIds)
+				);
+				if(!classifyAuths.isEmpty()){
+					break;
+				}
+			}
+		}
+		//鍑虹幇浜嗛敊璇暟鎹紝鍚屼竴涓鑹插拰鍚屼竴涓垎绫籭d瀛樺湪澶氭潯鎺堟潈璁板綍
+		if(classifyAuths.size()>1){
+			throw new ServiceException("瑙掕壊鍜屽垎绫婚厤缃瓨鍦ㄥ鏉¤褰曪紝璇疯仈绯荤鐞嗕汉鍛樻竻鐞嗛敊璇厤缃紒");
+		}
+		// 鏄惁涓鸿秴绠�
+		Boolean isAdmin = VciBaseUtil.checkAdminTenant();
+		// 鏈厤缃寜閽潈闄�
+		if(!isAdmin && (classifyAuths.isEmpty() || Func.isBlank(classifyAuths.get(0).getButtonIds()))){
+			return new ArrayList<>();
+		}
+		List<String> ids = new ArrayList<>();
+		// 濡傛灉涓嶆槸瓒呯鐢ㄦ埛
+		if(!isAdmin){
+			ids.addAll(Arrays.asList(classifyAuths.get(0).getButtonIds().split(",")));
+		}
+		return this.getMenuListByCode(ids,btmType,roleIds);
     }
 
 	/**
@@ -381,8 +427,7 @@
 	 */
 	@Override
 	public List<Menu> getButtonByParentCode(String code) {
-		List<Menu> buttonList = menuMapper.getButtonByParentCode(code);
-		return buttonList;
+		return menuMapper.getButtonsByRoleIdAndCode(null,code);
 	}
 
 	/**
@@ -444,20 +489,31 @@
 	/**
 	 * 鏍规嵁涓婚敭鑾峰彇鑿滃崟淇℃伅
 	 * @param ids
+	 * @param menuCode
+	 * @param roleIds
 	 * @return
 	 */
 	@Override
-	public List<Menu> getMenuListById(List<String> ids,String parentId) {
-		LambdaQueryWrapper<Menu> wrapper = Wrappers.<Menu>query()
-			.lambda().eq(Menu::getCategory, "2")
-			.eq(Menu::getIsDeleted,BladeConstant.DB_NOT_DELETED);
+	public List<Menu> getMenuListByCode(List<String> ids,String menuCode,String roleIds){
+		List<Menu> menuButtonList = null;
 		if(VciBaseUtil.checkAdminTenant()){
-			wrapper.eq(Menu::getParentId,parentId);
+			// 姝e父鎯呭喌涓嬫潵璇磋繖涓笉瀛樺湪涓虹┖鐨勬儏鍐�
+			// 鏌ヨ璇ヨ彍鍗曚笅鐨勬墍鏈夋寜閽�
+			menuButtonList = menuMapper.getButtonByIdsOrByParentCode(null, menuCode, null);
 		}else {
-			wrapper.in(Menu::getId, ids);
+			menuButtonList = menuMapper.getButtonByIdsOrByParentCode(roleIds,null,ids);
 		}
-		List<Menu> menuList = this.list(wrapper);
-		return menuList;
+		return menuButtonList;
+	}
+
+	/**
+	 * 鏍规嵁瑙掕壊id鑾峰彇宸叉巿鏉冪殑鎸夐挳淇℃伅
+	 * @param roleId
+	 * @return
+	 */
+	@Override
+	public List<Menu> getButtonsByRoleId(String roleId, String menuCode) {
+		return menuMapper.getButtonsByRoleIdAndCode(roleId,menuCode);
 	}
 
 }

--
Gitblit v1.9.3