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

@@ -16,21 +16,27 @@
 */
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 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;
@@ -61,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;
      }
@@ -74,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) {
@@ -90,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)));
      }
      // 顶部菜单请求返回对应角色权限菜单
@@ -133,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);
   }
@@ -145,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()));
   }

@@ -204,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());
@@ -214,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
@@ -267,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);
@@ -295,9 +324,208 @@
      return saveOrUpdate(menu);
   }

   /**
    * 获取菜单下面的按钮，别乱调用，这个方法是针对主数据管理按钮查询的
    * @param classifyId
    * @param btmType 业务类型
    * @param authType
    * @return
    */
    @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 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("获取分类信息失败！");
      }
      // 返回的分类oid是当前节点为第一个，后面依次是他的上层节点
      List<String> classifyOidList = listR.getData();
      final List<String> roleIds = Func.toStrList(",",AuthUtil.getUser().getRoleId());
      // 先查询按钮id列表
      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){
         // 校验是否存在错误数据，同一个角色和同一个分类id存在多条授权记录
         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对象的buttonIds
            .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排序与classify的btmtypeid对应*/
      // 超管不用根据角色来查询
      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);
   }

   /**
    * 根据父级菜单的code查询按钮信息
    * @param code
    * @return
    */
   @Override
   public List<Menu> getButtonByParentCode(String code) {
      return menuMapper.getButtonsByRoleIdAndCode(null,code);
   }

   /**
    * 对KeepAlive值转换成布尔类型进行封装
    *
    * @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);
   }

}