Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/MenuServiceImpl.java

@@ -16,11 +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;
@@ -69,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) {
@@ -96,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;
      }
      // 非超级管理员并且不是顶部菜单请求则返回对应角色权限菜单
@@ -139,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);
   }
@@ -161,7 +163,7 @@

   @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()));
   }

@@ -224,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
@@ -309,9 +311,19 @@
      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:' + #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);
    }

   /**
@@ -333,9 +345,7 @@
         .eq(Menu::getCategory,1) /*菜单类型不能为按钮*/
         .orderByAsc(Menu::getCode);  /*根据code排序与classify的btmtypeid对应*/
      // 超管不用根据角色来查询
      if(!AuthUtil.getTenantId().equals(NacosConfigCache.getAdminUserInfo().getTenantId())
         && !AuthUtil.getUserId().equals(NacosConfigCache.getAdminUserInfo().getUserId())
      ){
      if(!VciBaseUtil.checkAdminTenant()){
         if(Func.isBlank(userId.toString()) && Func.isBlank(AuthUtil.getUserId().toString())){
            throw new ServiceException("获取用户id失败");
         }