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

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,171 @@
package com.vci.ubcs.system.service.impl;

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 org.springblade.core.log.exception.ServiceException;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.Func;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

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

/**
 * åˆ†ç±»æŽˆæƒ
 * @author ludc
 * @date 2023/12/25 15:35
 */
@Service
@AllArgsConstructor
public class ClassifyAuthServiceImpl extends ServiceImpl<ClassifyAuthMapper,ClassifyAuth> implements IClassifyAuthService {

   private final ClassifyAuthMapper classifyAuthMapper;

   private final IMenuService menuService;

   private final ICodeClassifyClient codeClassifyClient;

   /**
    * åˆ†ç±»æŽˆæƒä¿å­˜æŽ¥å£
    * @param classifyAuthList
    * @return
    */
   @Override
   @Transactional(rollbackFor = Exception.class)
   public R submit(List<ClassifyAuth> classifyAuthList) {
      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());
      // åˆ é™¤
      LambdaUpdateWrapper<ClassifyAuth> updateWrapper = Wrappers.<ClassifyAuth>update()
         .lambda().eq(ClassifyAuth::getClassifyId, classifyAuthList.get(0).getClassifyId())
         .notIn(ClassifyAuth::getRoleId, roleIds);
      try {
         this.classifyAuthMapper.delete(updateWrapper);
      }catch (Exception e){
         throw new ServiceException("分类授权过程中出现错误，错误原因："+e.getMessage());
      }
      return R.status(saveOrUpdateBatch(classifyAuthList));
   }

   /**
    * èŽ·å–分类授权集合
    * @param classifyAuthVO
    * @return
    */
   @Override
   public List<ClassifyAuthVO> getClassifyAuthList(ClassifyAuthVO classifyAuthVO) {
      if(Func.isBlank(classifyAuthVO.getClassifyId())){
         throw new ServiceException("缺少必传参数分类id");
      }
      LambdaQueryWrapper<ClassifyAuth> wrapper = Wrappers.<ClassifyAuth>query()
         .lambda().eq(ClassifyAuth::getClassifyId,classifyAuthVO.getClassifyId());
      List<ClassifyAuth> classifyAuths = this.classifyAuthMapper.selectList(wrapper);
      if(classifyAuths.isEmpty()){
         return new ArrayList<ClassifyAuthVO>();
      }
      return ClassifyAuthWrapper.build().listVO(classifyAuths);
   }

   /**
    * æŸ¥è¯¢è¯¥åˆ†ç±»ä¸‹ï¼Œå½“前登录的角色有哪些按钮权限
    * @param classifyId
    * @return
    */
   public Map<String,Boolean> getAuthButtonList(String classifyId){
      if(Func.isBlank(classifyId)){
         return new HashMap<>();
      }
      //查询分类节点的所有父级节点
      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)
         .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 HashMap<>();
      }
      List<String> condition1 = new ArrayList<>();
      // å¦‚果不是超管用户
      if(!isAdmin){
         condition1.addAll(Arrays.asList(classifyAuths.get(0).getButtonIds().split(",")));
      }
      List<Menu> menuList = menuService.getMenuListById(condition1,"1648879284590858241");
      if(menuList.isEmpty()){
         return new HashMap<>();
      }
      Map<String, Boolean> buttonMaps = menuList.stream()
         .collect(Collectors.toMap(Menu::getCode, menu -> true));
      return buttonMaps;
   }

   /**
    * æ ¹æ®è§’色id查看有哪些分类具备查看权限
    * @param roleIds
    * @return
    */
   @Override
   public List<String> getViewClassByRoleIds(List<String> roleIds) {
      if(roleIds.isEmpty()){
         return new ArrayList<>();
      }
      return this.classifyAuthMapper.getViewClassByRoleIds(roleIds);
   }

}