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排序与classify的btmtypeid对应*/
      // 超管不用根据角色来查询
      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);
   }

   /**
    * 对KeepAlive值转换成布尔类型进行封装
    *
    * @param childMenu
@@ -323,4 +366,44 @@
      });
   }

   /**
    * 克隆其他菜单下按钮
    * @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("按钮克隆失败！");
   }

}