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

@@ -1,32 +1,34 @@
package com.vci.ubcs.system.service.impl;

import cn.hutool.db.ds.pooled.ConnectionWraper;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
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.web.util.VciBaseUtil;
import com.vci.ubcs.system.entity.ClassifyAuth;
import com.vci.ubcs.system.entity.Menu;
import com.vci.ubcs.system.mapper.ClassifyAuthMapper;
import com.vci.ubcs.system.service.IClassifyAuthService;
import com.vci.ubcs.system.service.IMenuService;
import com.vci.ubcs.system.vo.ClassifyAuthVO;
import com.vci.ubcs.system.vo.MenuVO;
import com.vci.ubcs.system.wrapper.ClassifyAuthWrapper;
import lombok.AllArgsConstructor;
import lombok.RequiredArgsConstructor;
import org.redisson.api.condition.Conditions;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.Func;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 分类授权
 * 分类授权,数据授权
 * @author ludc
 * @date 2023/12/25 15:35
 */
@@ -35,6 +37,10 @@
public class ClassifyAuthServiceImpl extends ServiceImpl<ClassifyAuthMapper,ClassifyAuth> implements IClassifyAuthService {

   private final ClassifyAuthMapper classifyAuthMapper;

   private final IMenuService menuService;

   private final ICodeClassifyClient codeClassifyClient;

   /**
    * 分类授权保存接口
@@ -47,6 +53,17 @@
      if(classifyAuthList.isEmpty()){
         R.fail("授权列表不能为空！");
      }
      // 判重，查看是否存在同一个classid下配置了相同的角色
      Map<String, Long> roleidCounts = classifyAuthList.stream()
         .collect(Collectors.groupingBy(ClassifyAuth::getRoleId, Collectors.counting()));

      // 检查是否有roleid出现次数大于1的情况
      boolean hasDuplicateRoleid = roleidCounts.values().stream()
         .anyMatch(count -> count > 1);
      if(hasDuplicateRoleid){
         R.fail("角色和分类已经存在，请重新配置！");
      }

      // 如果传过来的集合中该分类id下删除了部分角色的授权，就需要将该库中classifyId下不存在的数据删掉
      List<String> roleIds = classifyAuthList.stream().map(ClassifyAuth::getRoleId).collect(Collectors.toList());
      // 删除
@@ -72,12 +89,105 @@
         throw new ServiceException("缺少必传参数分类id");
      }
      LambdaQueryWrapper<ClassifyAuth> wrapper = Wrappers.<ClassifyAuth>query()
         .lambda().eq(ClassifyAuth::getClassifyId,classifyAuthVO.getClassifyId());
         .lambda().eq(ClassifyAuth::getClassifyId,classifyAuthVO.getClassifyId())
         .eq(ClassifyAuth::getAuthType,classifyAuthVO.getAuthType());
      List<ClassifyAuth> classifyAuths = this.classifyAuthMapper.selectList(wrapper);
      if(!classifyAuths.isEmpty()){
         return ClassifyAuthWrapper.build().listVO(classifyAuths);
      if(classifyAuths.isEmpty()){
         return new ArrayList<ClassifyAuthVO>();
      }
      return new ArrayList<ClassifyAuthVO>();
      return ClassifyAuthWrapper.build().listVO(classifyAuths);
   }

   /**
    * 查询该分类下，当前登录的角色有哪些按钮权限
    * @param classifyId
    * @param menuCode
    * @param authType
    * @return
    */
   public Map<String,Boolean> getAuthButtonList(String classifyId,String menuCode,String authType){
      List<Menu> menuList = this.getButtonList(classifyId, menuCode, authType);
      if(menuList.isEmpty()){
         return new HashMap<>();
      }
      Map<String, Boolean> buttonMaps = menuList.stream()
         .collect(Collectors.toMap(Menu::getCode, menu -> true));
      return buttonMaps;
   }

   /**
    * 查询该主数据下，当前登录的角色有哪些按钮菜单权限
    * @param classifyId
    * @param menuCode
    * @param authType
    * @return
    */
   public List<Menu> getAuthMenuButtonList(String classifyId,String menuCode,String authType){
      List<Menu> buttonList = getButtonList(classifyId, menuCode, authType);
      return buttonList;
   }

   private List<Menu> getButtonList(String classifyId,String menuCode,String authType){
      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 String roleIds = 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 = this.classifyAuthMapper.selectList(wrapper);
      //如果当前分类没有找到授权配置，就依次从当前节点往上层节点找授权配置，找到了就停止，没找到就一直找到最后
      if(classifyAuths.isEmpty()){
         // 下标从1开始因为当前节点0已经查询过
         for (int i = 1; i < classifyOidList.size(); i++) {
            classifyAuths = this.classifyAuthMapper.selectList(
               Wrappers.<ClassifyAuth>query()
                  .lambda().eq(ClassifyAuth::getClassifyId, classifyOidList.get(i))
                  .in(ClassifyAuth::getRoleId, roleIds)
            );
            if(!classifyAuths.isEmpty()){
               break;
            }
         }
      }
      //出现了错误数据，同一个角色和同一个分类id存在多条授权记录
      if(classifyAuths.size()>1){
         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){
         ids.addAll(Arrays.asList(classifyAuths.get(0).getButtonIds().split(",")));
      }
      return menuService.getMenuListByCode(ids,menuCode,roleIds);
   }

   /**
    * 根据角色id查看有哪些分类具备查看权限
    * @param roleIds
    * @return
    */
   @Override
   public List<String> getViewClassByRoleIds(List<String> roleIds,String authType,String buttonCode,String menuCode) {
      if(roleIds.isEmpty()){
         return new ArrayList<>();
      }
      return this.classifyAuthMapper.getViewClassByRoleIds(roleIds, authType,buttonCode,menuCode);
   }

}