From 483af85d1bf87622467173818a4fb0f2383b6f3a Mon Sep 17 00:00:00 2001 From: ludc Date: 星期五, 20 十月 2023 09:52:24 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java | 51 +++++++++++++++++++++++++++++++++++++++++++++++---- 1 files changed, 47 insertions(+), 4 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 aa566ac..06b47de 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 @@ -20,6 +20,8 @@ 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.system.cache.NacosConfigCache; import com.vci.ubcs.system.cache.SysCache; import com.vci.ubcs.system.dto.MenuDTO; import com.vci.ubcs.system.entity.*; @@ -67,6 +69,7 @@ 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) { @@ -148,13 +151,15 @@ @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())); return ForestNodeMerger.merge(tenantPackageTree(menuTree, user.getTenantId())); } @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())); } @@ -217,12 +222,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 @@ -308,6 +313,44 @@ } /** + * 鏍规嵁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,0) /*鏈鍒犻櫎*/ + .eq(Menu::getCategory,1) /*鑿滃崟绫诲瀷涓嶈兘涓烘寜閽�*/ + .orderByAsc(Menu::getCode); /*鏍规嵁code鎺掑簭涓巆lassify鐨刡tmtypeid瀵瑰簲*/ + // 瓒呯涓嶇敤鏍规嵁瑙掕壊鏉ユ煡璇� + if(!AuthUtil.getTenantId().equals(NacosConfigCache.getAdminUserInfo().getTenantId()) + && !AuthUtil.getUserId().equals(NacosConfigCache.getAdminUserInfo().getUserId()) + ){ + 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); + } + + /** * 瀵筀eepAlive鍊艰浆鎹㈡垚甯冨皵绫诲瀷杩涜灏佽 * * @param childMenu -- Gitblit v1.9.3