package cn.besteam.modules.system.biz;
|
|
import cn.besteam.common.constant.DefaultMenusEnum;
|
import cn.besteam.modules.system.entity.SysPermission;
|
import cn.besteam.modules.system.entity.SysRolePermission;
|
import cn.besteam.modules.system.service.ISysPermissionService;
|
import cn.besteam.modules.system.service.ISysRolePermissionService;
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author ltri
|
* @date 2020/1/11 11:20 AM
|
*/
|
@Service
|
public class SysPermissionBiz {
|
|
@Autowired
|
private ISysPermissionService sysPermissionService;
|
|
@Autowired
|
private ISysRolePermissionService sysRolePermissionService;
|
|
/**
|
* 添加默认菜单以及其子菜单
|
*/
|
public List<SysPermission> addDefaultMenu(List<SysPermission> metaList) {
|
|
Set<SysPermission> set = new HashSet<>(metaList);
|
//默认一级菜单
|
List<SysPermission> topMenus = sysPermissionService.list(Wrappers.<SysPermission>lambdaQuery().in(SysPermission::getName, DefaultMenusEnum.SET));
|
//一级菜单id集合
|
Set<Integer> topMenuIds = topMenus.stream().map(SysPermission::getId).collect(Collectors.toSet());
|
//一级菜单下子菜单
|
List<SysPermission> sonMenus = sysPermissionService.list(Wrappers.<SysPermission>lambdaQuery().in(SysPermission::getParentId, topMenuIds));
|
set.addAll(topMenus);
|
set.addAll(sonMenus);
|
metaList = new ArrayList<>(set);
|
//菜单排序
|
CollectionUtil.sort(metaList, Comparator.comparing(SysPermission::getSortNo));
|
return metaList;
|
}
|
|
/**
|
* 通过角色Id查询角色权限，（添加默认菜单后）
|
*
|
* @param roleId
|
* @return
|
*/
|
public Set<Integer> queryRolePermission(String roleId) {
|
|
Set<Integer> set = sysRolePermissionService.list(Wrappers.<SysRolePermission>lambdaQuery().eq(SysRolePermission::getRoleId, roleId)).stream().map(SysRolePermission::getPermissionId).collect(Collectors.toSet());
|
//首页菜单ID
|
set.add(6);
|
//新增运营人员菜单权限为空
|
// if (!set.isEmpty()) {
|
// set.addAll(getDefaultMenusId(set));
|
// }
|
return set;
|
}
|
|
/**
|
* 菜单及子菜单id集合
|
*
|
* @return
|
*/
|
public Set<Integer> getDefaultMenusId(Set<Integer> set) {
|
// //一级菜单
|
// List<SysPermission> topMenus = sysPermissionService.listByIds(set);
|
// //子菜单
|
// List<SysPermission> sonMenus = sysPermissionService.list(Wrappers.<SysPermission>lambdaQuery().in(SysPermission::getParentId, set));
|
// Set<SysPermission> metaList = new HashSet<>();
|
// metaList.addAll(topMenus);
|
// metaList.addAll(sonMenus);
|
// return metaList.stream().map(SysPermission::getId).collect(Collectors.toSet());
|
return set;
|
}
|
|
}
|
