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

@@ -20,10 +20,15 @@
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import com.vci.ubcs.starter.enumpack.CodeTableNameEnum;
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.*;
import com.vci.ubcs.system.user.entity.User;
import com.vci.ubcs.system.user.feign.IUserClient;
import com.vci.ubcs.system.vo.MenuVO;
import com.vci.ubcs.system.mapper.MenuMapper;
import com.vci.ubcs.system.service.IMenuService;
@@ -47,6 +52,7 @@
import org.springframework.cache.annotation.Cacheable;
import org.springframework.stereotype.Service;

import javax.management.relation.Role;
import java.util.*;
import java.util.stream.Collectors;

@@ -314,16 +320,38 @@
   /**
    * 根据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排序与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);
   }