From 52fd1e55b71fa38ffe89db738acc4e6c7e4da0df Mon Sep 17 00:00:00 2001
From: ludc
Date: 星期五, 02 二月 2024 16:47:35 +0800
Subject: [PATCH] 分类授权，数据授权界面请求参数修改

---
 Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java |  276 +++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 255 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 73a2601..764c61d 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
@@ -16,23 +16,30 @@
  */
 package com.vci.ubcs.system.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.Wrapper;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+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 com.vci.ubcs.core.log.exception.ServiceException;
+import org.springblade.core.log.exception.ServiceException;
+import org.springblade.core.mp.support.Condition;
+import org.springblade.core.mp.support.Query;
 import org.springblade.core.secure.BladeUser;
 import org.springblade.core.secure.utils.AuthUtil;
+import org.springblade.core.tool.api.R;
 import org.springblade.core.tool.constant.BladeConstant;
 import org.springblade.core.tool.node.ForestNodeMerger;
 import org.springblade.core.tool.node.TreeNode;
@@ -60,12 +67,16 @@
 
 	private final IRoleMenuService roleMenuService;
 	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;
 
 	@Override
 	public List<MenuVO> lazyList(Long parentId, Map<String, Object> param) {
+		//int i = 1 / 0;
 		if (Func.isEmpty(Func.toStr(param.get(PARENT_ID)))) {
 			parentId = null;
 		}
@@ -73,13 +84,12 @@
 	}
 
 	@Override
-	public List<MenuVO> lazyMenuList(Long parentId, Map<String, Object> param) {
+	public IPage<MenuVO> lazyMenuPage(Long parentId, Map<String, Object> param, Query query) {
 		if (Func.isEmpty(Func.toStr(param.get(PARENT_ID)))) {
 			parentId = null;
 		}
-		return baseMapper.lazyMenuList(parentId, param);
+		return baseMapper.lazyMenuPage(parentId, param, Condition.getPage(query));
 	}
-
 
 	@Override
 	public List<MenuVO> routes(String roleId, Long topMenuId) {
@@ -89,11 +99,11 @@
 		List<Menu> allMenus = baseMapper.allMenu();
 		List<Menu> roleMenus;
 		// 瓒呯骇绠＄悊鍛樺苟涓斾笉鏄《閮ㄨ彍鍗曡姹傚垯杩斿洖鍏ㄩ儴鑿滃崟
-		if (AuthUtil.isAdministrator() && Func.isEmpty(topMenuId)) {
+		if (VciBaseUtil.checkAdminTenant() && Func.isEmpty(topMenuId)) {
 			roleMenus = allMenus;
 		}
 		// 闈炶秴绾х鐞嗗憳骞朵笖涓嶆槸椤堕儴鑿滃崟璇锋眰鍒欒繑鍥炲搴旇鑹叉潈闄愯彍鍗�
-		else if (!AuthUtil.isAdministrator() && Func.isEmpty(topMenuId)) {
+		else if (!VciBaseUtil.checkAdminTenant() && Func.isEmpty(topMenuId)) {
 			roleMenus = tenantPackageMenu(baseMapper.roleMenuByRoleId(Func.toLongList(roleId)));
 		}
 		// 椤堕儴鑿滃崟璇锋眰杩斿洖瀵瑰簲瑙掕壊鏉冮檺鑿滃崟
@@ -132,7 +142,7 @@
 
 	@Override
 	public List<MenuVO> buttons(String roleId) {
-		List<Menu> buttons = (AuthUtil.isAdministrator()) ? baseMapper.allButtons() : baseMapper.buttons(Func.toLongList(roleId));
+		List<Menu> buttons = (VciBaseUtil.checkAdminTenant()) ? baseMapper.allButtons() : baseMapper.buttons(Func.toLongList(roleId));
 		MenuWrapper menuWrapper = new MenuWrapper();
 		return menuWrapper.listNodeVO(buttons);
 	}
@@ -144,13 +154,28 @@
 
 	@Override
 	public List<TreeNode> grantTree(BladeUser user) {
-		List<TreeNode> menuTree = user.getTenantId().equals(BladeConstant.ADMIN_TENANT_ID) ? baseMapper.grantTree() : baseMapper.grantTreeByRole(Func.toLongList(user.getRoleId()));
+		/*List<TreeNode> menuTree = user.getTenantId().equals(nacosConfigCache.getAdminUserInfo().getTenantId())
+			&& user.getUserId().equals(nacosConfigCache.getAdminUserInfo().getUserId())
+			? baseMapper.grantTree() : baseMapper.grantTreeByRole(Func.toLongList(user.getRoleId()));*/
+		// 淇敼涓烘墍鏈夌敤鎴烽兘鑳芥煡璇㈡墍鏈夛紝鍥犱负娌℃潈闄愮殑鐢ㄦ埛涓嶅叿澶囪繖涓彍鍗曢〉闈㈢殑鏉冮檺锛屼篃灏辨病蹇呰闄愬埗鐢ㄦ埛鏌ヨ嚜宸辩殑
+		List<TreeNode> menuTree = baseMapper.grantTree();
 		return ForestNodeMerger.merge(tenantPackageTree(menuTree, user.getTenantId()));
+	}
+
+	/**
+	 * 鏍规嵁瑙掕壊id鑾峰彇鑿滃崟鏍戝舰缁撴瀯
+	 * @param roleId
+	 * @return
+	 */
+	@Override
+	public List<TreeNode> grantTreeByRoleIds(List<Long> roleId) {
+		List<TreeNode> menuTree = baseMapper.grantTreeByRole(roleId);
+		return ForestNodeMerger.merge(menuTree);
 	}
 
 	@Override
 	public List<TreeNode> grantTopTree(BladeUser user) {
-		List<TreeNode> menuTree = user.getTenantId().equals(BladeConstant.ADMIN_TENANT_ID) ? baseMapper.grantTopTree() : baseMapper.grantTopTreeByRole(Func.toLongList(user.getRoleId()));
+		List<TreeNode> menuTree = user.getTenantId().equals(NacosConfigCache.getAdminUserInfo().getTenantId()) ? baseMapper.grantTopTree() : baseMapper.grantTopTreeByRole(Func.toLongList(user.getRoleId()));
 		return ForestNodeMerger.merge(tenantPackageTree(menuTree, user.getTenantId()));
 	}
 
@@ -203,6 +228,7 @@
 	 * 绉熸埛鑿滃崟鏉冮檺鑷畾涔夌瓫閫�
 	 */
 	private List<Menu> tenantPackageMenu(List<Menu> menu) {
+		// 绉熸埛鍖呴厤缃煡璇�
 		TenantPackage tenantPackage = SysCache.getTenantPackage(AuthUtil.getTenantId());
 		if (Func.isNotEmpty(tenantPackage) && tenantPackage.getId() > 0L) {
 			List<Long> menuIds = Func.toLongList(tenantPackage.getMenuId());
@@ -213,12 +239,12 @@
 
 	@Override
 	public List<TreeNode> grantDataScopeTree(BladeUser user) {
-		return ForestNodeMerger.merge(user.getTenantId().equals(BladeConstant.ADMIN_TENANT_ID) ? baseMapper.grantDataScopeTree() : baseMapper.grantDataScopeTreeByRole(Func.toLongList(user.getRoleId())));
+		return ForestNodeMerger.merge(user.getTenantId().equals(NacosConfigCache.getAdminUserInfo().getTenantId()) ? baseMapper.grantDataScopeTree() : baseMapper.grantDataScopeTreeByRole(Func.toLongList(user.getRoleId())));
 	}
 
 	@Override
 	public List<TreeNode> grantApiScopeTree(BladeUser user) {
-		return ForestNodeMerger.merge(user.getTenantId().equals(BladeConstant.ADMIN_TENANT_ID) ? baseMapper.grantApiScopeTree() : baseMapper.grantApiScopeTreeByRole(Func.toLongList(user.getRoleId())));
+		return ForestNodeMerger.merge(user.getTenantId().equals(NacosConfigCache.getAdminUserInfo().getTenantId()) ? baseMapper.grantApiScopeTree() : baseMapper.grantApiScopeTreeByRole(Func.toLongList(user.getRoleId())));
 	}
 
 	@Override
@@ -266,15 +292,19 @@
 	@Override
 	public boolean submit(Menu menu) {
 		LambdaQueryWrapper<Menu> menuQueryWrapper = Wrappers.lambdaQuery();
+		// 鏂板
 		if (menu.getId() == null) {
-			menuQueryWrapper.eq(Menu::getCode, menu.getCode()).or(
-				wrapper -> wrapper.eq(Menu::getName, menu.getName()).eq(Menu::getCategory, MENU_CATEGORY)
-			);
+			//妫�楠屾槸鍚﹂噸澶嶈彍鍗曞埆鍚嶏紝鍙牎楠屽悓涓�鐖跺垎绫昏彍鍗曚笅鐨勭紪鍙锋槸鍚﹂噸澶�
+			menuQueryWrapper.eq(Menu::getParentId,menu.getParentId()!=null ? menu.getParentId():BladeConstant.TOP_PARENT_ID)
+				.eq(Menu::getCode, menu.getCode())
+				.or(wrapper -> wrapper.eq(Menu::getName, menu.getName()).eq(Menu::getCategory, MENU_CATEGORY))
+			;
 		} else {
+			// 淇敼
 			menuQueryWrapper.ne(Menu::getId, menu.getId()).and(
-				wrapper -> wrapper.eq(Menu::getCode, menu.getCode()).or(
-					o -> o.eq(Menu::getName, menu.getName()).eq(Menu::getCategory, MENU_CATEGORY)
-				)
+				wrapper -> wrapper.eq(Menu::getCode, menu.getCode())
+					.eq(Menu::getParentId,menu.getParentId())
+					.or(o -> o.eq(Menu::getName, menu.getName()).eq(Menu::getCategory, MENU_CATEGORY))
 			);
 		}
 		Long cnt = baseMapper.selectCount(menuQueryWrapper);
@@ -294,4 +324,208 @@
 		return saveOrUpdate(menu);
 	}
 
+	/**
+	 * 鑾峰彇鑿滃崟涓嬮潰鐨勬寜閽紝鍒贡璋冪敤锛岃繖涓柟娉曟槸閽堝涓绘暟鎹鐞嗘寜閽煡璇㈢殑
+	 * @param classifyId
+	 * @param btmType 涓氬姟绫诲瀷
+	 * @param authType
+	 * @return
+	 */
+    @Override
+	//@Cacheable(cacheNames = MENU_CACHE, key = "'auth:menuButton:'+ #btmType +':'+ #userId ")
+    public List<Menu> getMenuButtonByType(String classifyId,String btmType,String authType) {
+		// baseMapper.selectMenuChildByBtnType(btmType,roleIds);
+		if(Func.isBlank(classifyId)){
+			throw new ServiceException("蹇呬紶鍙傛暟鍒嗙被oid涓嶈兘涓虹┖锛�");
+		}
+		//鏌ヨ鍒嗙被鑺傜偣鐨勬墍鏈夌埗绾ц妭鐐�
+		R<List<String>> listR = codeClassifyClient.selectAllParentOid(classifyId);
+		if (!listR.isSuccess() && !listR.getData().isEmpty()) {
+			throw new ServiceException("鑾峰彇鍒嗙被淇℃伅澶辫触锛�");
+		}
+		// 杩斿洖鐨勫垎绫籵id鏄綋鍓嶈妭鐐逛负绗竴涓紝鍚庨潰渚濇鏄粬鐨勪笂灞傝妭鐐�
+		List<String> classifyOidList = listR.getData();
+		final List<String> roleIds = Func.toStrList(",",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))
+						.eq(ClassifyAuth::getAuthType,authType)
+						.in(ClassifyAuth::getRoleId, roleIds)
+				);
+				if(!classifyAuths.isEmpty()){
+					break;
+				}
+			}
+		}
+		//鍑虹幇浜嗗鏉℃暟鎹�
+		if(classifyAuths.size()>1){
+			// 鏍￠獙鏄惁瀛樺湪閿欒鏁版嵁锛屽悓涓�涓鑹插拰鍚屼竴涓垎绫籭d瀛樺湪澶氭潯鎺堟潈璁板綍
+			List<ClassifyAuth> finalClassifyAuths = classifyAuths;
+			boolean hasDuplicate = classifyAuths.stream()
+				.anyMatch(auth1 -> finalClassifyAuths.stream()
+					.filter(auth2 -> auth1 != auth2)
+					.anyMatch(auth2 -> auth1.getRoleId().equals(auth2.getRoleId()) && auth1.getClassifyId().equals(auth2.getClassifyId())));
+			if (hasDuplicate) {
+				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){
+			String concatenatedButtonIds = classifyAuths.stream()
+				.map(ClassifyAuth::getButtonIds) // 鑾峰彇姣忎釜classifyAuths瀵硅薄鐨刡uttonIds
+				.collect(Collectors.joining(",")); // 鐢ㄩ�楀彿鍒嗛殧鎷兼帴鎴愪竴涓瓧绗︿覆
+			ids.addAll(Arrays.asList(concatenatedButtonIds.split(",")));
+		}
+		return this.getMenuListByCode(ids,btmType,roleIds);
+    }
+
+	/**
+	 * 鏍规嵁code鏌ヨ鑿滃崟淇℃伅
+	 * @param codes
+	 * @param userId
+	 * @return
+	 */
+	@Override
+	public List<Menu> getMenuByCodes(List<String> codes,Long userId) {
+		if(codes.isEmpty()){
+			return new ArrayList<>();
+		}
+		// 鏌ヨ鑿滃崟淇℃伅
+		LambdaQueryWrapper<Menu> wrapper = Wrappers.<Menu>query()
+			.lambda()
+			.in(Menu::getCode, codes)
+			.eq(Menu::getIsDeleted,BladeConstant.DB_NOT_DELETED) /*鏈鍒犻櫎*/
+			.eq(Menu::getCategory,1) /*鑿滃崟绫诲瀷涓嶈兘涓烘寜閽�*/
+			.orderByAsc(Menu::getCode);  /*鏍规嵁code鎺掑簭涓巆lassify鐨刡tmtypeid瀵瑰簲*/
+		// 瓒呯涓嶇敤鏍规嵁瑙掕壊鏉ユ煡璇�
+		if(!VciBaseUtil.checkAdminTenant()){
+			if(Func.isBlank(userId.toString()) && Func.isBlank(AuthUtil.getUserId().toString())){
+				throw new ServiceException("鑾峰彇鐢ㄦ埛id澶辫触");
+			}
+			List<Long> menuIds = roleMenuService.getMenuIdByUserId(userId);
+			if(menuIds.isEmpty()){
+				return new ArrayList<>();
+			}
+			if(menuIds.size()>=1000){
+				MybatisParameterUtil.cutInParameter(wrapper,Menu::getId,menuIds);
+			}else {
+				wrapper.in(Menu::getId,menuIds);
+			}
+		}
+		return this.list(wrapper);
+	}
+
+	/**
+	 * 鏍规嵁鐖剁骇鑿滃崟鐨刢ode鏌ヨ鎸夐挳淇℃伅
+	 * @param code
+	 * @return
+	 */
+	@Override
+	public List<Menu> getButtonByParentCode(String code) {
+		return menuMapper.getButtonsByRoleIdAndCode(null,code);
+	}
+
+	/**
+	 * 瀵筀eepAlive鍊艰浆鎹㈡垚甯冨皵绫诲瀷杩涜灏佽
+	 *
+	 * @param childMenu
+	 * @return
+	 */
+	@Override
+	public void handleKeepAlive(List<MenuVO> childMenu) {
+		childMenu.forEach(list->{
+			list.getMeta().put("keepAlive","true".equals(list.getKeepAlive()));
+			if(list.getChildren().size()>0){
+				handleKeepAlive(list.getChildren());
+			}
+		});
+	}
+
+	/**
+	 * 鍏嬮殕鍏朵粬鑿滃崟涓嬫寜閽�
+	 * @param menuId 瑕佸厠闅嗙殑鑿滃崟鎸夐挳涓婚敭
+	 * @param buttonIds 琚厠闅嗙殑鎸夐挳涓婚敭
+	 * @return
+	 */
+	@Override
+	public R cloneMenuButton(Long menuId, List<String> buttonIds) {
+		if(Func.isEmpty(menuId)){
+			return R.fail("瑕佸厠闅嗙殑鑿滃崟涓婚敭涓嶈兘涓虹┖锛�");
+		}
+		if(buttonIds.isEmpty() || buttonIds.size() <= 0){
+			return R.fail("琚厠闅嗙殑鎸夐挳涓婚敭涓嶈兘涓虹┖锛�");
+		}
+		// 鍏堟牴鎹富閿煡璇㈠嚭鎵�鏈夋寜閽殑淇℃伅
+		List<Menu> buttons = this.listByIds(buttonIds);
+		List<Menu> newButtons = new ArrayList<>();
+		List<String> addButtonCodes = new ArrayList();
+		buttons.parallelStream().forEach(item->{
+			// 鍒ゆ柇鏄惁涓烘寜閽紝闈炴寜閽笉澶勭悊
+			if(item.getCategory().equals(2)){
+				// 鏀瑰彉鐖惰妭鐐逛俊鎭�
+				item.setParentId(menuId);
+				// 灏嗕富閿祴绌�
+				item.setId(null);
+				addButtonCodes.add(item.getCode());
+				newButtons.add(item);
+			}
+		});
+		//妫�楠屾槸鍚﹂噸澶嶈彍鍗曞埆鍚嶏紝鍙牎楠屽悓涓�鐖跺垎绫昏彍鍗曚笅鐨勭紪鍙锋槸鍚﹂噸澶�
+		LambdaQueryWrapper<Menu> menuQueryWrapper = Wrappers.<Menu>lambdaQuery()
+			.eq(Menu::getParentId,menuId)
+			.and(a -> a.in( Menu::getCode, addButtonCodes));
+		Long cnt = baseMapper.selectCount(menuQueryWrapper);
+		if (cnt > 0L) {
+			return R.fail("璇ヨ彍鍗曚笅宸插瓨鍦ㄧ殑缂栧彿涓庤鍏嬮殕鐨勬寜閽紪鍙峰瓨鍦ㄩ噸澶�!");
+		}
+		return this.saveBatch(newButtons) ? R.success("鎸夐挳鍏嬮殕鎴愬姛锛�"):R.fail("鎸夐挳鍏嬮殕澶辫触锛�");
+	}
+
+	/**
+	 * 鏍规嵁涓婚敭鑾峰彇鑿滃崟淇℃伅
+	 * @param ids
+	 * @param menuCode
+	 * @param roleIds
+	 * @return
+	 */
+	@Override
+	public List<Menu> getMenuListByCode(List<String> ids,String menuCode,List<String> roleIds){
+		List<Menu> menuButtonList = null;
+		if(VciBaseUtil.checkAdminTenant()){
+			// 姝ｅ父鎯呭喌涓嬫潵璇磋繖涓笉瀛樺湪涓虹┖鐨勬儏鍐�
+			// 鏌ヨ璇ヨ彍鍗曚笅鐨勬墍鏈夋寜閽�
+			menuButtonList = menuMapper.getButtonByIdsOrByParentCode(null, menuCode, null);
+		}else {
+			menuButtonList = menuMapper.getButtonByIdsOrByParentCode(roleIds,null,ids);
+		}
+		return menuButtonList;
+	}
+
+	/**
+	 * 鏍规嵁瑙掕壊id鑾峰彇宸叉巿鏉冪殑鎸夐挳淇℃伅
+	 * @param roleId
+	 * @return
+	 */
+	@Override
+	public List<Menu> getButtonsByRoleId(String roleId, String menuCode) {
+		return menuMapper.getButtonsByRoleIdAndCode(roleId,menuCode);
+	}
+
 }

--
Gitblit v1.9.3