linwei
2024-01-31 b3b566ebdfed4005aaa513da3d5d2fd3924903cc
src/main/java/org/springblade/modules/system/service/impl/RoleServiceImpl.java
@@ -26,6 +26,7 @@
import org.springblade.core.tool.constant.BladeConstant;
import org.springblade.core.tool.constant.RoleConstant;
import org.springblade.core.tool.node.ForestNodeMerger;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.CollectionUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.system.entity.Role;
@@ -72,13 +73,21 @@
   @Override
   public List<RoleVO> tree(String tenantId) {
      String userRole = AuthUtil.getUserRole();
      String roleId = AuthUtil.getUser().getRoleId();
      String excludeRole = null;
      if (!CollectionUtil.contains(Func.toStrArray(userRole), RoleConstant.ADMIN) && !CollectionUtil.contains(Func.toStrArray(userRole), RoleConstant.ADMINISTRATOR)) {
         excludeRole = RoleConstant.ADMIN;
      }
      return ForestNodeMerger.merge(baseMapper.tree(tenantId, excludeRole));
      List<RoleVO> merge = ForestNodeMerger.merge(baseMapper.tree(tenantId, excludeRole));
      List<RoleVO> copy = BeanUtil.copy(merge, RoleVO.class);
      if (!AuthUtil.isAdministrator()) {
         List<RoleVO> collect = copy.stream().filter(item -> roleId.contains(item.getId().toString())).collect(Collectors.toList());
         return collect;
      }
      return merge;
   }
   @Override
   @Transactional(rollbackFor = Exception.class)
   public boolean grant(@NotEmpty List<Long> roleIds, List<Long> menuIds, List<Long> dataScopeIds, List<Long> apiScopeIds) {