From e61a898fe563214cb13e97c08804534a61aad46b Mon Sep 17 00:00:00 2001 From: ludc Date: 星期二, 07 十一月 2023 17:32:40 +0800 Subject: [PATCH] 主数据管理按钮返回缓存问题修改 --- Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java | 57 ++++++++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 46 insertions(+), 11 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 cc258f7..035269d 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,10 +16,13 @@ */ 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.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; @@ -68,7 +71,6 @@ private final ITopMenuSettingService topMenuSettingService; private final static String PARENT_ID = "parentId"; private final static Integer MENU_CATEGORY = 1; - private final NacosConfigCache nacosConfigCache; @Override public List<MenuVO> lazyList(Long parentId, Map<String, Object> param) { @@ -95,7 +97,8 @@ List<Menu> allMenus = baseMapper.allMenu(); List<Menu> roleMenus; // 瓒呯骇绠$悊鍛樺苟涓斾笉鏄《閮ㄨ彍鍗曡姹傚垯杩斿洖鍏ㄩ儴鑿滃崟 - if (AuthUtil.isAdministrator() && Func.isEmpty(topMenuId)) { + // if (AuthUtil.isAdministrator() && Func.isEmpty(topMenuId)) { + if (VciBaseUtil.checkAdminTenant() && Func.isEmpty(topMenuId)) { roleMenus = allMenus; } // 闈炶秴绾х鐞嗗憳骞朵笖涓嶆槸椤堕儴鑿滃崟璇锋眰鍒欒繑鍥炲搴旇鑹叉潈闄愯彍鍗� @@ -138,7 +141,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); } @@ -150,15 +153,17 @@ @Override public List<TreeNode> grantTree(BladeUser user) { - List<TreeNode> menuTree = user.getTenantId().equals(nacosConfigCache.getAdminUserInfo().getTenantId()) + /*List<TreeNode> menuTree = user.getTenantId().equals(nacosConfigCache.getAdminUserInfo().getTenantId()) && user.getUserId().equals(nacosConfigCache.getAdminUserInfo().getUserId()) - ? baseMapper.grantTree() : baseMapper.grantTreeByRole(Func.toLongList(user.getRoleId())); + ? baseMapper.grantTree() : baseMapper.grantTreeByRole(Func.toLongList(user.getRoleId()));*/ + // 淇敼涓烘墍鏈夌敤鎴烽兘鑳芥煡璇㈡墍鏈夛紝鍥犱负娌℃潈闄愮殑鐢ㄦ埛涓嶅叿澶囪繖涓彍鍗曢〉闈㈢殑鏉冮檺锛屼篃灏辨病蹇呰闄愬埗鐢ㄦ埛鏌ヨ嚜宸辩殑 + List<TreeNode> menuTree = baseMapper.grantTree(); return ForestNodeMerger.merge(tenantPackageTree(menuTree, user.getTenantId())); } @Override public List<TreeNode> grantTopTree(BladeUser user) { - List<TreeNode> menuTree = user.getTenantId().equals(nacosConfigCache.getAdminUserInfo().getTenantId()) ? 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())); } @@ -221,12 +226,12 @@ @Override public List<TreeNode> grantDataScopeTree(BladeUser user) { - return ForestNodeMerger.merge(user.getTenantId().equals(nacosConfigCache.getAdminUserInfo().getTenantId()) ? 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(nacosConfigCache.getAdminUserInfo().getTenantId()) ? 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 @@ -306,24 +311,54 @@ return saveOrUpdate(menu); } + /** + * 鑾峰彇鑿滃崟涓嬮潰鐨勬寜閽� + * @param btmType 涓氬姟绫诲瀷 + * @return List<Menu> + */ @Override - public List<Menu> getMenuButtonByType(String btmType) { - return baseMapper.selectMenuChildByBtnType(btmType); + @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(",")); + } + return baseMapper.selectMenuChildByBtnType(btmType,roleIds); } /** * 鏍规嵁code鏌ヨ鑿滃崟淇℃伅 * @param codes + * @param userId * @return */ @Override - public List<Menu> getMenuByCodes(List<String> codes) { + 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,0) /*鏈鍒犻櫎*/ .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); } -- Gitblit v1.9.3