58 lines
1.4 KiB
Java
58 lines
1.4 KiB
Java
package com.ruoyi.system.service;
|
|
|
|
import com.ruoyi.common.core.domain.entity.SysUser;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
import java.util.HashSet;
|
|
import java.util.Set;
|
|
|
|
/**
|
|
* 用户权限处理
|
|
*
|
|
* @author ruoyi
|
|
*/
|
|
@Service
|
|
public class SysPermissionService {
|
|
|
|
@Autowired
|
|
private ISysRoleService roleService;
|
|
|
|
@Autowired
|
|
private ISysMenuService menuService;
|
|
|
|
/**
|
|
* 获取角色数据权限
|
|
*
|
|
* @param user 用户信息
|
|
* @return 角色权限信息
|
|
*/
|
|
public Set<String> getRolePermission(SysUser user) {
|
|
Set<String> roles = new HashSet<String>();
|
|
// 管理员拥有所有权限
|
|
if (user.isAdmin()) {
|
|
roles.add("admin");
|
|
} else {
|
|
roles.addAll(roleService.selectRolePermissionByUserId(user.getUserId()));
|
|
}
|
|
return roles;
|
|
}
|
|
|
|
/**
|
|
* 获取菜单数据权限
|
|
*
|
|
* @param user 用户信息
|
|
* @return 菜单权限信息
|
|
*/
|
|
public Set<String> getMenuPermission(SysUser user) {
|
|
Set<String> perms = new HashSet<String>();
|
|
// 管理员拥有所有权限
|
|
if (user.isAdmin()) {
|
|
perms.add("*:*:*");
|
|
} else {
|
|
perms.addAll(menuService.selectMenuPermsByUserId(user.getUserId()));
|
|
}
|
|
return perms;
|
|
}
|
|
}
|