SpringSecurity权限控制&实现思路
SpringSecurity权限控制&实现思路
1. 简介
权限控制是项目中必不可少的部分
2. 表结构
实体 | 表 | 说明 |
---|---|---|
SysUser | sys_user | 用户信息 |
SysRole | sys_role | 用户信息 |
SysUserRole | sys_user_role | 用户和角色关联 |
SysMenu | sys_menu | 菜单权限 |
SysRoleMenu | sys_role_menu | 角色和菜单关联 |
5 个表的关系比较简单:
- 一个 SysUser ,可以拥有多个 SysRole ,通过 SysUserRole 存储关联。
- 一个 SysRole ,可以拥有多个 SysMenu ,通过 SysRoleMenu 存储关联。
2.1 SysUser
SysUser,用户实体类。代码如下:
// SysUser.java
public class SysUser extends BaseEntity {
private static final long serialVersionUID = 1L;
@Excel(name = "用户序号", cellType = ColumnType.NUMERIC, prompt = "用户编号")
private Long userId;
@Excel(name = "部门编号", type = Type.IMPORT)
private Long deptId;
@Excel(name = "登录名称")
private String userName;
@Excel(name = "用户名称")
private String nickName;
@Excel(name = "用户邮箱")
private String email;
@Excel(name = "手机号码")
private String phonenumber;
@Excel(name = "用户性别", readConverterExp = "0=男,1=女,2=未知")
private String sex;
/** 用户头像 */
private String avatar;
/** 密码 */
private String password;
/** 盐加密 */
private String salt;
@Excel(name = "帐号状态", readConverterExp = "0=正常,1=停用")
private String status;
/** 删除标志(0代表存在 2代表删除) */
private String delFlag;
@Excel(name = "最后登录IP", type = Type.EXPORT)
private String loginIp;
@Excel(name = "最后登录时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss", type = Type.EXPORT)
private Date loginDate;
@Excels({
@Excel(name = "部门名称", targetAttr = "deptName", type = Type.EXPORT),
@Excel(name = "部门负责人", targetAttr = "leader", type = Type.EXPORT)
})
@Transient
private SysDept dept;
/** 角色对象 */
@Transient
private List<SysRole> roles;
/** 角色组 */
@Transient
private Long[] roleIds;
/** 岗位组 */
@Transient
private Long[] postIds;
// ...省略 set/get 方法
}
- 添加
@Transient
注解的字段,非存储字段。后续的实体,补充重复赘述。 - 每个字段比较简单,根据注释理解下即可。
对应表的创建 SQL 如下:
create table sys_user (
user_id bigint(20) not null auto_increment comment '用户ID',
dept_id bigint(20) default null comment '部门ID',
user_name varchar(30) not null comment '用户账号',
nick_name varchar(30) not null comment '用户昵称',
user_type varchar(2) default '00' comment '用户类型(00系统用户)',
email varchar(50) default '' comment '用户邮箱',
phonenumber varchar(11) default '' comment '手机号码',
sex char(1) default '0' comment '用户性别(0男 1女 2未知)',
avatar varchar(100) default '' comment '头像地址',
password varchar(100) default '' comment '密码',
status char(1) default '0' comment '帐号状态(0正常 1停用)',
del_flag char(1) default '0' comment '删除标志(0代表存在 2代表删除)',
login_ip varchar(50) default '' comment '最后登录IP',
login_date datetime comment '最后登录时间',
create_by varchar(64) default '' comment '创建者',
create_time datetime comment '创建时间',
update_by varchar(64) default '' comment '更新者',
update_time datetime comment '更新时间',
remark varchar(500) default null comment '备注',
primary key (user_id)
) engine=innodb auto_increment=100 comment = '用户信息表';
2.2 SysRole
SysRole,角色实体类。代码如下:
// SysRole.java
public class SysRole extends BaseEntity {
private static final long serialVersionUID = 1L;
@Excel(name = "角色序号", cellType = ColumnType.NUMERIC)
private Long roleId;
@Excel(name = "角色名称")
private String roleName;
@Excel(name = "角色权限")
private String roleKey;
@Excel(name = "角色排序")
private String roleSort;
@Excel(name = "数据范围", readConverterExp = "1=所有数据权限,2=自定义数据权限,3=本部门数据权限,4=本部门及以下数据权限")
private String dataScope;
@Excel(name = "角色状态", readConverterExp = "0=正常,1=停用")
private String status;
/** 删除标志(0代表存在 2代表删除) */
private String delFlag;
/** 用户是否存在此角色标识 默认不存在 */
@Transient
private boolean flag = false;
/** 菜单组 */
@Transient
private Long[] menuIds;
/** 部门组(数据权限) */
@Transient
private Long[] deptIds;
// ...省略 set/get 方法
}
- 每个字段比较简单,根据注释理解下即可。
对应表的创建 SQL 如下:
create table sys_role (
role_id bigint(20) not null auto_increment comment '角色ID',
role_name varchar(30) not null comment '角色名称',
role_key varchar(100) not null comment '角色权限字符串',
role_sort int(4) not null comment '显示顺序',
data_scope char(1) default '1' comment '数据范围(1:全部数据权限 2:自定数据权限 3:本部门数据权限 4:本部门及以下数据权限)',
status char(1) not null comment '角色状态(0正常 1停用)',
del_flag char(1) default '0' comment '删除标志(0代表存在 2代表删除)',
create_by varchar(64) default '' comment '创建者',
create_time datetime comment '创建时间',
update_by varchar(64) default '' comment '更新者',
update_time datetime comment '更新时间',
remark varchar(500) default null comment '备注',
primary key (role_id)
) engine=innodb auto_increment=100 comment = '角色信息表';
2.3 SysUserRole
SysUserRole,用户和角色关联实体类。代码如下:
// SysUserRole.java
public class SysUserRole {
/** 用户ID */
private Long userId;
/** 角色ID */
private Long roleId;
// ...省略 set/get 方法
}
- 每个字段比较简单,根据注释理解下即可。
roleKey
属性,对应的角色标识字符串,可以对应多个角色标识,使用逗号分隔。例如说:"admin,normal"
。
对应表的创建 SQL 如下:
create table sys_user_role (
user_id bigint(20) not null comment '用户ID',
role_id bigint(20) not null comment '角色ID',
primary key(user_id, role_id)
) engine=innodb comment = '用户和角色关联表';
2.4 SysMenu
SysMenu ,菜单权限实体类。代码如下:
// SysMenu.java
public class SysMenu extends BaseEntity {
private static final long serialVersionUID = 1L;
/** 菜单ID */
private Long menuId;
/** 菜单名称 */
private String menuName;
/** 父菜单名称 */
private String parentName;
/** 父菜单ID */
private Long parentId;
/** 显示顺序 */
private String orderNum;
/** 路由地址 */
private String path;
/** 组件路径 */
private String component;
/** 是否为外链(0是 1否) */
private String isFrame;
/** 类型(M目录 C菜单 F按钮) */
private String menuType;
/** 菜单状态:0显示,1隐藏 */
private String visible;
/** 权限字符串 */
private String perms;
/** 菜单图标 */
private String icon;
/** 子菜单 */
@Transient
private List<SysMenu> children = new ArrayList<SysMenu>();
// ...省略 set/get 方法
}
个人感觉,这个实体改成 SysResource 资源,更加合适,菜单仅仅是其中的一种。
每个字段比较简单,我们来重点看几个字段。
menuType
属性,定义了三种类型。其中,F
代表按钮,是为了做页面中的功能级的权限。perms
属性,对应的权限标识字符串。一般格式为${大模块}:${小模块}:{操作}
。示例如下:
用户查询:system:user:query 用户新增:system:user:add 用户修改:system:user:edit 用户删除:system:user:remove 用户导出:system:user:export 用户导入:system:user:import 重置密码:system:user:resetPwd
- 对于前端来说,每个按钮在展示时,可以判断用户是否有该按钮的权限。如果没有,则进行隐藏。当然,前端在首次进入系统的时候,会请求一次权限列表到本地进行缓存。
- 对于后端来说,每个接口上会添加
@PreAuthorize("@ss.hasPermi('system:user:list')")
注解。在请求接口时,会校验用户是否有该 URL 对应的权限。如果没有,则会抛出权限验证失败的异常。 - 一个
perms
属性,可以对应多个权限标识,使用逗号分隔。例如说:"system:user:query,system:user:add"
。
对应表的创建 SQL 如下:
create table sys_menu (
menu_id bigint(20) not null auto_increment comment '菜单ID',
menu_name varchar(50) not null comment '菜单名称',
parent_id bigint(20) default 0 comment '父菜单ID',
order_num int(4) default 0 comment '显示顺序',
path varchar(200) default '' comment '路由地址',
component varchar(255) default null comment '组件路径',
is_frame int(1) default 1 comment '是否为外链(0是 1否)',
menu_type char(1) default '' comment '菜单类型(M目录 C菜单 F按钮)',
visible char(1) default 0 comment '菜单状态(0显示 1隐藏)',
perms varchar(100) default null comment '权限标识',
icon varchar(100) default '#' comment '菜单图标',
create_by varchar(64) default '' comment '创建者',
create_time datetime comment '创建时间',
update_by varchar(64) default '' comment '更新者',
update_time datetime comment '更新时间',
remark varchar(500) default '' comment '备注',
primary key (menu_id)
) engine=innodb auto_increment=2000 comment = '菜单权限表';
2.5 SysRoleMenu
SysRoleMenu,菜单权限实体类。代码如下:
// SysRoleMenu.java
public class SysRoleMenu {
/** 角色ID */
private Long roleId;
/** 菜单ID */
private Long menuId;
// ...省略 set/get 方法
}
- 每个字段比较简单,友自己根据注释理解下即可。
对应表的创建 SQL 如下:
create table sys_role_menu (
role_id bigint(20) not null comment '角色ID',
menu_id bigint(20) not null comment '菜单ID',
primary key(role_id, menu_id)
) engine=innodb comment = '角色和菜单关联表';
3. SecurityConfig
在SecurityConfig 配置类,继承 WebSecurityConfigurerAdapter 抽象类,实现 Spring Security 在 Web 场景下的自定义配置。代码如下:
// SecurityConfig.java
@Configuration
public class SecurityConfig extends WebSecurityConfigurerAdapter {
// ...
}
涉及到的配置方法较多,我们逐个来看看。
3.1 实现 AuthenticationManager认证管理器
重写 #configure(AuthenticationManagerBuilder auth)
方法,实现 AuthenticationManager
认证管理器。代码如下:
// SecurityConfig.java
/**
* 自定义用户认证逻辑
*/
@Autowired
private UserDetailsService userDetailsService;
/**
* 身份认证接口
*/
@Override
protected void configure(AuthenticationManagerBuilder auth) throws Exception {
auth.userDetailsService(userDetailsService) // <X>
.passwordEncoder(bCryptPasswordEncoder()); // <Y>
}
/**
* 强散列哈希加密实现
*/
@Bean
public BCryptPasswordEncoder bCryptPasswordEncoder() {
return new BCryptPasswordEncoder();
}
<X>
处,调用AuthenticationManagerBuilder#userDetailsService(userDetailsService)
方法,使用自定义实现的UserDetailsService实现类,更加灵活且自由的实现认证的用户信息的读取。在[加载用户信息中,我们会看到 RuoYi-Vue 对 UserDetailsService 的自定义实现类。<Y>
处,调用AbstractDaoAuthenticationConfigurer#passwordEncoder(passwordEncoder)
方法,设置 PasswordEncoder 密码编码器。这里,就使用了 bCryptPasswordEncoder 强散列哈希加密实现。
3.2 配置 URL 的权限
// SecurityConfig.java
/**
* 认证失败处理类
*/
@Autowired
private AuthenticationEntryPointImpl unauthorizedHandler;
/**
* 退出处理类
*/
@Autowired
private LogoutSuccessHandlerImpl logoutSuccessHandler;
/**
* token 认证过滤器
*/
@Autowired
private JwtAuthenticationTokenFilter authenticationTokenFilter;
@Override
protected void configure(HttpSecurity httpSecurity) throws Exception {
httpSecurity
// CRSF禁用,因为不使用session
.csrf().disable()
// <X> 认证失败处理类
.exceptionHandling().authenticationEntryPoint(unauthorizedHandler).and()
// 基于token,所以不需要session
.sessionManagement().sessionCreationPolicy(SessionCreationPolicy.STATELESS).and()
// 过滤请求
.authorizeRequests()
// <Y> 对于登录login 验证码captchaImage 允许匿名访问
.antMatchers("/login", "/captchaImage").anonymous()
.antMatchers(
HttpMethod.GET,
"/*.html",
"/**/*.html",
"/**/*.css",
"/**/*.js"
).permitAll()
.antMatchers("/profile/**").anonymous()
.antMatchers("/common/download**").anonymous()
.antMatchers("/swagger-ui.html").anonymous()
.antMatchers("/swagger-resources/**").anonymous()
.antMatchers("/webjars/**").anonymous()
.antMatchers("/*/api-docs").anonymous()
.antMatchers("/druid/**").anonymous()
// 除上面外的所有请求全部需要鉴权认证
.anyRequest().authenticated()
.and()
.headers().frameOptions().disable();
httpSecurity.logout().logoutUrl("/logout").logoutSuccessHandler(logoutSuccessHandler); // <Z>
// <P> 添加 JWT filter
httpSecurity.addFilterBefore(authenticationTokenFilter, UsernamePasswordAuthenticationFilter.class);
}
<X>
处,设置认证失败时的处理器为unauthorizedHandler
。详细解析,见 AuthenticationEntryPointImpl。<Y>
处,设置用于登录的/login
接口,允许匿名访问。这样,后续我们就可以使用自定义的登录接口。详细解析,见登录 API 接口<Z>
处,设置登出成功的处理器为logoutSuccessHandler
。详细解析,见LogoutSuccessHandlerImpl<P>
处,添加 JWT 认证过滤器authenticationTokenFilter
,用于用户使用用户名与密码登录完成后,后续请求基于 JWT 来认证。 详细解析,见 JwtAuthenticationTokenFilter。
3.3 重写 #authenticationManagerBean
方法
重写 #authenticationManagerBean 方法,解决无法直接注入 AuthenticationManager 的问题。代码如下:
// SecurityConfig.java
@Bean
@Override
public AuthenticationManager authenticationManagerBean() throws Exception {
return super.authenticationManagerBean();
}
在方法上,额外添加了 @Bean 注解,保证创建出 AuthenticationManager Bean 。
4. 登录 API 接口
4.1 登录接口
SysLoginController#login(...)
在 SysLoginController 中,定义了 /login
接口,提供登录功能。代码如下:
/ SysLoginController.java
@Autowired
private SysLoginService loginService;
/**
* 登录方法
*
* @param username 用户名
* @param password 密码
* @param code 验证码
* @param uuid 唯一标识
* @return 结果
*/
@PostMapping("/login")
public AjaxResult login(String username, String password, String code, String uuid) {
AjaxResult ajax = AjaxResult.success();
// 生成令牌
String token = loginService.login(username, password, code, uuid);
ajax.put(Constants.TOKEN, token);
return ajax;
}
在内部,会调用
loginService#login(username, password, code, uuid)
方法,会进行基于用户名与密码的登录认证。认证通过后,返回身份 TOKEN 。登录成功后,该接口响应示例如下
{ "msg": "操作成功", "code": 200, "token": "eyJhbGciOiJIUzUxMiJ9.eyJsb2dpbl91c2VyX2tleSI6ImJkN2Q4OTZiLTU2NTAtNGIyZS1iNjFjLTc0MjlkYmRkNzA1YyJ9.lkU8ot4GecLHs7VAcRAo1fLMOaFryd4W5Q_a2wzPwcOL0Kiwyd4enpnGd79A_aQczXC-JB8vELNcNn7BrtJn9A" }
- 后续,前端在请求后端接口时,在请求头上带头该
token
值,作为用户身份标识。
- 后续,前端在请求后端接口时,在请求头上带头该
4.2 登录service
在 SysLoginService
中,定义了 #login(username, password, code, uuid)
方法,进行基于用户名与密码的登录认证。认证通过后,返回身份 TOKEN 。代码如下:
// SysLoginService.java
@Autowired
private TokenService tokenService;
@Resource
private AuthenticationManager authenticationManager;
@Autowired
private RedisCache redisCache;
/**
* 登录验证
*
* @param username 用户名
* @param password 密码
* @param code 验证码
* @param uuid 唯一标识
* @return 结果
*/
public String login(String username, String password, String code, String uuid) {
// <1> 验证图片验证码的正确性
String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid; // uuid 的作用,是获得对应的图片验证码
String captcha = redisCache.getCacheObject(verifyKey); // 从 Redis 中,获得图片验证码
redisCache.deleteObject(verifyKey); // 从 Redis 中,删除图片验证码
if (captcha == null) { // 图片验证码不存在
AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.jcaptcha.error")));
throw new CaptchaExpireException();
}
if (!code.equalsIgnoreCase(captcha)) { // 图片验证码不正确
AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.jcaptcha.expire")));
throw new CaptchaException();
}
// <2> 用户验证
Authentication authentication;
try {
// 该方法会去调用 UserDetailsServiceImpl.loadUserByUsername
authentication = authenticationManager
.authenticate(new UsernamePasswordAuthenticationToken(username, password));
} catch (Exception e) {
// <2.1> 发生异常,说明验证不通过,记录相应的登录失败日志
if (e instanceof BadCredentialsException) {
AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.password.not.match")));
throw new UserPasswordNotMatchException();
} else {
AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, e.getMessage()));
throw new CustomException(e.getMessage());
}
}
// <2.2> 验证通过,记录相应的登录成功日志
AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_SUCCESS, MessageUtils.message("user.login.success")));
// <3> 生成 Token
LoginUser loginUser = (LoginUser) authentication.getPrincipal();
return tokenService.createToken(loginUser);
}
<1>
处,验证图片验证码的正确性。该验证码会存储在 Redis 缓存中,通过uuid
作为对应的标识。生成的逻辑,看 CaptchaController 提供的/captchaImage
接口。<2>
处,调用 Spring Security 的AuthenticationManager#authenticate(UsernamePasswordAuthenticationToken authentication)方法,基于用户名与密码的登录认证。在其内部,会调用我们定义的 UserDetailsServiceImpl #loadUserByUsername(String username)方法,获得指定用户名对应的用户信息。<2.1>
处,发生异常,说明认证不通过,记录相应的登录失败日志。<2.2>
处,未发生异常,说明认证通过,记录相应的登录成功日志。- 关于上述日志,我们在 登录日志 来讲。
<3>
处,调用 TokenService 的#createToken(LoginUser loginUser)
方法,给认证通过的用户,生成其对应的认证 TOKEN 。这样,该用户的后续请求,就使用该 TOKEN 作为身份标识进行认证。
4.3 加载用户信息
在 UserDetailsServiceImpl 中,实现 Spring Security UserDetailsService 接口,实现了该接口定义的
// UserDetailsServiceImpl.java
private static final Logger log = LoggerFactory.getLogger(UserDetailsServiceImpl.class);
@Autowired
private ISysUserService userService;
@Autowired
private SysPermissionService permissionService;
@Override
public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
// <1> 查询指定用户名对应的 SysUser
SysUser user = userService.selectUserByUserName(username);
// <2> 各种校验
if (StringUtils.isNull(user)) {
log.info("登录用户:{} 不存在.", username);
throw new UsernameNotFoundException("登录用户:" + username + " 不存在");
} else if (UserStatus.DELETED.getCode().equals(user.getDelFlag())) {
log.info("登录用户:{} 已被删除.", username);
throw new BaseException("对不起,您的账号:" + username + " 已被删除");
} else if (UserStatus.DISABLE.getCode().equals(user.getStatus())) {
log.info("登录用户:{} 已被停用.", username);
throw new BaseException("对不起,您的账号:" + username + " 已停用");
}
// <3> 创建 Spring Security UserDetails 用户明细
return createLoginUser(user);
}
public UserDetails createLoginUser(SysUser user) {
return new LoginUser(user, permissionService.getMenuPermission(user));
}
<1>
处,调用 ISysUserService 的 #selectUserByUserName(String userName)
方法,查询指定用户名对应的 SysUser 。代码如下:
// SysUserServiceImpl.java
@Autowired
private SysUserMapper userMapper;
@Override
public SysUser selectUserByUserName(String userName) {
return userMapper.selectUserByUserName(userName);
}
// SysUserMapper.XML
<sql id="selectUserVo">
select u.user_id, u.dept_id, u.user_name, u.nick_name, u.email, u.avatar, u.phonenumber, u.password, u.sex, u.status, u.del_flag, u.login_ip, u.login_date, u.create_by, u.create_time, u.remark,
d.dept_id, d.parent_id, d.dept_name, d.order_num, d.leader, d.status as dept_status,
r.role_id, r.role_name, r.role_key, r.role_sort, r.data_scope, r.status as role_status
from sys_user u
left join sys_dept d on u.dept_id = d.dept_id
left join sys_user_role ur on u.user_id = ur.user_id
left join sys_role r on r.role_id = ur.role_id
</sql>
<select id="selectUserByUserName" parameterType="String" resultMap="SysUserResult">
<include refid="selectUserVo"/>
where u.user_name = #{userName}
</select>
通过查询
sys_user
表,同时连接sys_dept
、sys_user_role
、sys_role
表,将username
对应的 SysUser 相关信息都一次性查询出来。返回结果
SysUserResult
,实际就是 SysUser 实体类。<2>
处,各种校验。如果校验不通过,抛出 UsernameNotFoundException 或 BaseException 异常。<3>
处,调用 SysPermissionService 的#getMenuPermission(SysUser user)
方法,获得用户的 SysRoleMenu 的权限标识字符串的集合。代码如下:// SysPermissionService.java @Autowired private ISysMenuService menuService; public Set<String> getMenuPermission(SysUser user) { Set<String> roles = new HashSet<String>(); // 管理员拥有所有权限 if (user.isAdmin()) { roles.add("*:*:*"); // 所有模块 } else { // 读取 roles.addAll(menuService.selectMenuPermsByUserId(user.getUserId())); } return roles; } // SysMenuServiceImpl.java @Autowired private SysMenuMapper menuMapper; @Override public Set<String> selectMenuPermsByUserId(Long userId) { // 读取 SysMenu 的权限标识数组 List<String> perms = menuMapper.selectMenuPermsByUserId(userId); // 逐个,按照“逗号”分隔 Set<String> permsSet = new HashSet<>(); for (String perm : perms) { if (StringUtils.isNotEmpty(perm)) { permsSet.addAll(Arrays.asList(perm.trim().split(","))); } } return permsSet; } // SysMenuMapper.xml <select id="selectMenuPermsByUserId" parameterType="Long" resultType="String"> select distinct m.perms from sys_menu m left join sys_role_menu rm on m.menu_id = rm.menu_id left join sys_user_role ur on rm.role_id = ur.role_id where ur.user_id = #{userId} </select>
- 虽然代码很长,但是核心的并不多。
- 首先,如果 SysUser 是超级管理员,则其权限标识集合就是
*:*:*
,标识可以所有模块的所有操作。 - 然后,查询
sys_menu
表,同时连接sys_role_menu
、sys_user_role
表,将 SysUser 拥有的 SysMenu 的权限标识数组,然后使用","
分隔每个 SysMenu 对应的权限标识。
这里,我们看到最终返回的是 LoginUser ,实现 Spring Security UserDetails 接口,自定义的用户明细。代码如下:
// LoginUser.java
public class LoginUser implements UserDetails {
private static final long serialVersionUID = 1L;
/** 用户唯一标识 */
private String token;
/** 登录时间 */
private Long loginTime;
/** 过期时间 */
private Long expireTime;
/** 登录IP地址 */
private String ipaddr;
/** 登录地点 */
private String loginLocation;
/** 浏览器类型 */
private String browser;
/** 操作系统 */
private String os;
/** 权限列表 */
private Set<String> permissions;
/** 用户信息 */
private SysUser user;
// ...省略 set/get 方法,以及各种实现方法
}
5. 创建认证 Token
在 TokenService
中,定义了 #createToken(LoginUser loginUser)
方法,给认证通过的用户,生成其对应的认证 Token 。代码如下:
// TokenService.java
/**
* 创建令牌
*
* @param loginUser 用户信息
* @return 令牌
*/
public String createToken(LoginUser loginUser) {
// <1> 设置 LoginUser 的用户唯一标识。注意,这里虽然变量名叫 token ,其实不是身份认证的 Token
String token = IdUtils.fastUUID();
loginUser.setToken(token);
// <2> 设置用户终端相关的信息,包括 IP、城市、浏览器、操作系统
setUserAgent(loginUser);
// <3> 记录缓存
refreshToken(loginUser);
// <4> 生成 JWT 的 Token
Map<String, Object> claims = new HashMap<>();
claims.put(Constants.LOGIN_USER_KEY, token);
return createToken(claims);
}
注意,这个方法不仅仅会生成认证 Token ,还会缓存 LoginUser 到 Redis 缓存中。
<1>
处,设置 LoginUser 的用户唯一标识,即LoginUser.token
。注意,这里虽然变量名叫token
,其实不是身份认证的 Token 。<2>
处,调用#setUserAgent(LoginUser loginUser)
方法,设置用户终端相关的信息,包括 IP、城市、浏览器、操作系统。代码如下:// TokenService.java public void setUserAgent(LoginUser loginUser) { UserAgent userAgent = UserAgent.parseUserAgentString(ServletUtils.getRequest().getHeader("User-Agent")); String ip = IpUtils.getIpAddr(ServletUtils.getRequest()); loginUser.setIpaddr(ip); loginUser.setLoginLocation(AddressUtils.getRealAddressByIP(ip)); loginUser.setBrowser(userAgent.getBrowser().getName()); loginUser.setOs(userAgent.getOperatingSystem().getName()); }
<3>
处,调用#refreshToken(LoginUser loginUser)
方法,缓存 LoginUser 到 Redis 缓存中。代码如下:// application.yaml # token配置 token: # 令牌有效期(默认30分钟) expireTime: 30 // Constants.java /** * 登录用户 redis key */ public static final String LOGIN_TOKEN_KEY = "login_tokens:"; // TokenService.java // 令牌有效期(默认30分钟) @Value("${token.expireTime}") private int expireTime; @Autowired private RedisCache redisCache; public void refreshToken(LoginUser loginUser) { loginUser.setLoginTime(System.currentTimeMillis()); loginUser.setExpireTime(loginUser.getLoginTime() + expireTime * MILLIS_MINUTE); // 根据 uuid 将 loginUser 缓存 String userKey = getTokenKey(loginUser.getToken()); redisCache.setCacheObject(userKey, loginUser, expireTime, TimeUnit.MINUTES); } private String getTokenKey(String uuid) { return Constants.LOGIN_TOKEN_KEY + uuid; }
- 缓存的 Redis Key 的统一前缀为
"login_tokens:"
,使用 Login 的用户唯一标识(LoginUser.token
)作为后缀。
- 缓存的 Redis Key 的统一前缀为
<4>
处,调用#createToken(Map<String, Object> claims)
方法,生成 JWT 的 Token 。代码如下:// application.yaml # token配置 token: # 令牌秘钥 secret: abcdefghijklmnopqrstuvwxyz // TokenService.java // 令牌秘钥 @Value("${token.secret}") private String secret; private String createToken(Map<String, Object> claims) { return Jwts.builder() .setClaims(claims) .signWith(SignatureAlgorithm.HS512, secret).compact(); }
不要把这里生成的 JWT 的 Token ,和我们上面的
LoginUser.token
混淆在一起。- 因为
LoginUser.token
添加到claims
中,最终生成了 JWT 的 Token 。所以,后续我们可以通过解码该 JWT 的 Token ,从而获得claims
,最终获得到对应的LoginUser.token
。 - 在 JWT 的 Token 中,使用
LoginUser.token
而不是userId
的好处,可以带来更好的安全性,避免万一secret
秘钥泄露之后,黑客可以顺序生成userId
从而生成对应的 JWT 的 Token ,后续直接可以操作该用户的数据。不过,这么做之后就不是纯粹的 JWT ,解析出来的LoginUser.token
需要查询对应的 LoginUser 的 Redis 缓存。
- 因为
5.1 JwtAuthenticationTokenFilter
JwtAuthenticationTokenFilter 中,继承 OncePerRequestFilter 过滤器,实现了基于 Token 的认证。代码如下:
// JwtAuthenticationTokenFilter.java
@Component
public class JwtAuthenticationTokenFilter extends OncePerRequestFilter {
@Autowired
private TokenService tokenService;
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain)
throws ServletException, IOException {
// <1> 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(request);
// 如果存在 LoginUser ,并且未认证过
if (StringUtils.isNotNull(loginUser) && StringUtils.isNull(SecurityUtils.getAuthentication())) {
// <2> 校验 Token 有效性
tokenService.verifyToken(loginUser);
// <3> 创建 UsernamePasswordAuthenticationToken 对象,设置到 SecurityContextHolder 中
UsernamePasswordAuthenticationToken authenticationToken = new UsernamePasswordAuthenticationToken(loginUser, null, loginUser.getAuthorities());
authenticationToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request));
SecurityContextHolder.getContext().setAuthentication(authenticationToken);
}
// <4> 继续过滤器
chain.doFilter(request, response);
}
}
<1>
处,调用 TokenService 的#getLoginUser(request)
方法,获得当前 LoginUser 。代码如下:// application.yaml # token配置 token: # 令牌自定义标识 header: Authorization // TokenService.jav // 令牌自定义标识 @Value("${token.header}") private String header; /** * 获取用户身份信息 * * @return 用户信息 */ public LoginUser getLoginUser(HttpServletRequest request) { // <1.1> 获取请求携带的令牌 String token = getToken(request); if (StringUtils.isNotEmpty(token)) { // <1.2> 解析 JWT 的 Token Claims claims = parseToken(token); // <1.3> 从 Redis 缓存中,获得对应的 LoginUser String uuid = (String) claims.get(Constants.LOGIN_USER_KEY); String userKey = getTokenKey(uuid); return redisCache.getCacheObject(userKey); } return null; } private String getToken(HttpServletRequest request) { String token = request.getHeader(header); if (StringUtils.isNotEmpty(token) && token.startsWith(Constants.TOKEN_PREFIX)) { token = token.replace(Constants.TOKEN_PREFIX, ""); } return token; } private Claims parseToken(String token) { return Jwts.parser() .setSigningKey(secret) .parseClaimsJws(token) .getBody(); }
<1.1>
处,调用#getToken(request)
方法,从请求头"Authorization"
中,获得身份认证的 Token 。<1.2>
处,调用#parseToken(token)
方法,解析 JWT 的 Token ,获得 Claims 对象,从而获得用户唯一标识(LoginUser.token
)。<1.3>
处,从 Redis 缓存中,获得对应的 LoginUser 。
<2>
处,调用 TokenService 的#verifyToken(LoginUser loginUser)
方法,验证令牌有效期。代码如下:// TokenService.java protected static final long MILLIS_SECOND = 1000; protected static final long MILLIS_MINUTE = 60 * MILLIS_SECOND; private static final Long MILLIS_MINUTE_TEN = 20 * 60 * 1000L; /** * 验证令牌有效期,相差不足 20 分钟,自动刷新缓存 * * @param loginUser 用户 */ public void verifyToken(LoginUser loginUser) { long expireTime = loginUser.getExpireTime(); long currentTime = System.currentTimeMillis(); // 相差不足 20 分钟,自动刷新缓存 if (expireTime - currentTime <= MILLIS_MINUTE_TEN) { String token = loginUser.getToken(); loginUser.setToken(token); refreshToken(loginUser); } }
- 实际上,这个方法的目的不是验证 Token 的有效性,而是刷新对应的 LoginUser 的缓存的过期时间。
- 考虑到避免每次请求都去刷新缓存的过期时间,所以过期时间不足 20 分钟,才会去刷新。
<3>
处,手动创建 UsernamePasswordAuthenticationToken 对象,设置到 SecurityContextHolder 中。😈 因为,我们已经通过 Token 来完成认证了。<4>
处,继续过滤器。
严格来说,RuoYi-Vue 并不是采用的无状态的 JWT ,而是使用 JWT 的 Token 的生成方式
6. 权限验证
我们看到可以通过 Spring Security 提供的 @PreAuthorize
注解,实现基于 Spring EL 表达式的执行结果为 true
时,可以访问,从而实现灵活的权限校验。
在 RuoYi-Vue 中,通过 @PreAuthorize
注解的特性,使用其 PermissionService 提供的权限验证的方法。使用示例如下:
// SysDictDataController.java
@PreAuthorize("@ss.hasPermi('system:dict:list')")
@GetMapping("/list")
- 请求
/system/dict/data/list
接口,会调用 PermissionService 的#hasPermi(String permission)
方法,校验用户是否有指定的权限。 - 为什么这里会有一个
@ss
呢?在 Spring EL 表达式中,调用指定 Bean 名字的方法时,使用@
+ Bean 的名字。在 RuoYi-Vue 中,声明 PermissionService 的 Bean 名字为ss
。
6.1 判断是否有权限
在 PermissionService 中,定义了 #hasPermi(String permission)
方法,判断当前用户是否有指定的权限。代码如下:
// PermissionService.java
/**
* 所有权限标识
*/
private static final String ALL_PERMISSION = "*:*:*";
@Autowired
private TokenService tokenService;
/**
* 验证用户是否具备某权限
*
* @param permission 权限字符串
* @return 用户是否具备某权限
*/
public boolean hasPermi(String permission) {
// 如果未设置需要的权限,强制不具备。
if (StringUtils.isEmpty(permission)) {
return false;
}
// 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
// 如果不存在,或者没有任何权限,说明权限验证不通过
if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions())) {
return false;
}
// 判断是否包含权限
return hasPermissions(loginUser.getPermissions(), permission);
}
/**
* 判断是否包含权限
*
* @param permissions 权限列表
* @param permission 权限字符串
* @return 用户是否具备某权限
*/
private boolean hasPermissions(Set<String> permissions, String permission) {
return permissions.contains(ALL_PERMISSION) || permissions.contains(StringUtils.trim(permission));
}
在 PermissionService 中,定义了 #lacksPermi(String permission)
方法,判断当前用户是否没有指定的权限。代码如下:
// PermissionService.java
/**
* 验证用户是否不具备某权限,与 hasPermi逻辑相反
*
* @param permission 权限字符串
* @return 用户是否不具备某权限
*/
public boolean lacksPermi(String permission) {
return !hasPermi(permission);
}
在 PermissionService 中,定义了 #hasAnyPermi(String permissions)
方法,判断当前用户是否有指定的任一权限。代码如下:
// PermissionService.java
private static final String PERMISSION_DELIMETER = ",";
/**
* 验证用户是否具有以下任意一个权限
*
* @param permissions 以 PERMISSION_NAMES_DELIMETER 为分隔符的权限列表
* @return 用户是否具有以下任意一个权限
*/
public boolean hasAnyPermi(String permissions) {
// 如果未设置需要的权限,强制不具备。
if (StringUtils.isEmpty(permissions)) {
return false;
}
// 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
// 如果不存在,或者没有任何权限,说明权限验证不通过
if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions())) {
return false;
}
// 判断是否包含指定的任一权限
Set<String> authorities = loginUser.getPermissions();
for (String permission : permissions.split(PERMISSION_DELIMETER)) {
if (permission != null && hasPermissions(authorities, permission)) {
return true;
}
}
return false;
}
6.2 判断是否有角色
在 PermissionService 中,定义了 #hasRole(String role)
方法,判断当前用户是否有指定的角色。代码如下:
// PermissionService.java
/**
* 判断用户是否拥有某个角色
*
* @param role 角色字符串
* @return 用户是否具备某角色
*/
public boolean hasRole(String role) {
// 如果未设置需要的角色,强制不具备。
if (StringUtils.isEmpty(role)) {
return false;
}
// 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
// 如果不存在,或者没有任何角色,说明权限验证不通过
if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles())) {
return false;
}
// 判断是否包含指定角色
for (SysRole sysRole : loginUser.getUser().getRoles()) {
String roleKey = sysRole.getRoleKey();
if (SUPER_ADMIN.contains(roleKey) // 超级管理员的特殊处理
|| roleKey.contains(StringUtils.trim(role))) {
return true;
}
}
return false;
}
在 PermissionService 中,定义了 #lacksRole(String role)
方法,判断当前用户是否没有指定的角色。代码如下:
// PermissionService.java
/**
* 验证用户是否不具备某角色,与 isRole逻辑相反。
*
* @param role 角色名称
* @return 用户是否不具备某角色
*/
public boolean lacksRole(String role) {
return !hasRole(role);
}
在 PermissionService 中,定义了 #hasAnyRoles(String roles)
方法,判断当前用户是否有指定的任一角色。代码如下:
// PermissionService.java
private static final String ROLE_DELIMETER = ",";
/**
* 验证用户是否具有以下任意一个角色
*
* @param roles 以 ROLE_NAMES_DELIMETER 为分隔符的角色列表
* @return 用户是否具有以下任意一个角色
*/
public boolean hasAnyRoles(String roles) {
// 如果未设置需要的角色,强制不具备。
if (StringUtils.isEmpty(roles)) {
return false;
}
// 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
// 如果不存在,或者没有任何角色,说明权限验证不通过
if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles())) {
return false;
}
// 判断是否包含指定的任一角色
for (String role : roles.split(ROLE_DELIMETER)) {
if (hasRole(role)) { // 这里实现有点问题,会循环调用 hasRole 方法,重复从 Redis 中读取当前 LoginUser
return true;
}
}
return false;
}
7. 各种处理器
在 Ruoyi-Vue 中,提供了各种处理器,处理各种情况
7.1 AuthenticationEntryPointImpl
在 AuthenticationEntryPointImpl 中,实现 Spring Security AuthenticationEntryPoint 接口,处理认失败的 AuthenticationException 异常。代码如下:
// AuthenticationEntryPointImpl.java
// 认证失败处理类 返回未授权
@Component
public class AuthenticationEntryPointImpl implements AuthenticationEntryPoint, Serializable {
private static final long serialVersionUID = -8970718410437077606L;
@Override
public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException e) {
// 响应认证不通过
int code = HttpStatus.UNAUTHORIZED;
String msg = StringUtils.format("请求访问:{},认证失败,无法访问系统资源", request.getRequestURI());
ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(code, msg)));
}
}
- 响应认证不通过的 JSON 字符串。
6.2 GlobalExceptionHandler
在 GlobalExceptionHandler 中,定义了对 Spring Security 的异常处理。代码如下:
// GlobalExceptionHandler.java
@RestControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler(AccessDeniedException.class) // 没有访问权限。使用 @PreAuthorize 校验权限不通过时,就会抛出 AccessDeniedException 异常
public AjaxResult handleAuthorizationException(AccessDeniedException e) {
log.error(e.getMessage());
return AjaxResult.error(HttpStatus.FORBIDDEN, "没有权限,请联系管理员授权");
}
@ExceptionHandler(AccountExpiredException.class) // 账号已过期
public AjaxResult handleAccountExpiredException(AccountExpiredException e) {
log.error(e.getMessage(), e);
return AjaxResult.error(e.getMessage());
}
@ExceptionHandler(UsernameNotFoundException.class) // 用户名不存在
public AjaxResult handleUsernameNotFoundException(UsernameNotFoundException e) {
log.error(e.getMessage(), e);
return AjaxResult.error(e.getMessage());
}
// ... 省略对其它的异常类的处理的方法
}
- 基于 Spring MVC 提供的
@RestControllerAdvice
+@ExceptionHandler
注解,实现全局异常的处理。
7.3 LogoutSuccessHandlerImpl
在 LogoutSuccessHandlerImpl 中,实现 Spring Security LogoutSuccessHandler 接口,自定义退出的处理,主动删除 LoginUser 在 Redis 中的缓存。代码如下:
// LogoutSuccessHandlerImpl.java
// 自定义退出处理类 返回成功
@Configuration
public class LogoutSuccessHandlerImpl implements LogoutSuccessHandler {
@Autowired
private TokenService tokenService;
/**
* 退出处理
*/
@Override
public void onLogoutSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication) {
// <1> 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(request);
// 如果有登录的情况下
if (StringUtils.isNotNull(loginUser)) {
String userName = loginUser.getUsername();
// <2> 删除用户缓存记录
tokenService.delLoginUser(loginUser.getToken());
// <3> 记录用户退出日志
AsyncManager.me().execute(AsyncFactory.recordLogininfor(userName, Constants.LOGOUT, "退出成功"));
}
// <4> 响应退出成功
ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(HttpStatus.SUCCESS, "退出成功")));
}
}
<1>
处,调用 TokenService 的#getLoginUser(request)
方法,获得当前 LoginUser 。<2>
处,调用 TokenService 的#delLoginUser(String token)
方法,删除 LoginUser 的 Redis 缓存。代码如下:// TokenService.java public void delLoginUser(String token) { if (StringUtils.isNotEmpty(token)) { String userKey = getTokenKey(token); // 删除缓存 redisCache.deleteObject(userKey); } }
<3>
处,记录相应的退出成功日志。<4>
处,响应退出成功的 JSON 字符串。
8. 获得用户信息 API 接口
在 SysLoginController 中,定义了 /getInfo
接口,获取登录的用户信息。代码如下:
// SysLoginController.java
/**
* 获取用户信息
*
* @return 用户信息
*/
@GetMapping("getInfo")
public AjaxResult getInfo() {
// <1> 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
SysUser user = loginUser.getUser();
// <2> 角色标识的集合
Set<String> roles = permissionService.getRolePermission(user);
// <3> 权限集合
Set<String> permissions = permissionService.getMenuPermission(user);
// <4> 返回结果
AjaxResult ajax = AjaxResult.success();
ajax.put("user", user);
ajax.put("roles", roles);
ajax.put("permissions", permissions);
return ajax;
}
<1>
处,调用 TokenService 的#getLoginUser(request)
方法,获得当前 LoginUser 。<2>
处,调用 PermissionService 的#getRolePermission(SysUser user)
方法,获得 LoginUser 拥有的角色标识的集合。代码如下:// SysPermissionService.java @Autowired private ISysRoleService roleService; /** * 获取角色数据权限 * * @param user 用户信息 * @return 角色权限信息 */ public Set<String> getRolePermission(SysUser user) { Set<String> roles = new HashSet<String>(); // 管理员拥有所有权限 if (user.isAdmin()) { // 如果是管理员,强制添加 admin 角色 roles.add("admin"); } else { // 如果非管理员,进行查询 roles.addAll(roleService.selectRolePermissionByUserId(user.getUserId())); } return roles; } // SysRoleServiceImpl.java @Autowired private SysRoleMapper roleMapper; /** * 根据用户ID查询权限 * * @param userId 用户ID * @return 权限列表 */ @Override public Set<String> selectRolePermissionByUserId(Long userId) { // 获得 userId 拥有的 SysRole 数组 List<SysRole> perms = roleMapper.selectRolePermissionByUserId(userId); // 遍历 SysRole 数组,生成角色标识数组 Set<String> permsSet = new HashSet<>(); for (SysRole perm : perms) { if (StringUtils.isNotNull(perm)) { permsSet.addAll(Arrays.asList(perm.getRoleKey().trim().split(","))); } } return permsSet; } // SysRoleMapper.xml <sql id="selectRoleVo"> select distinct r.role_id, r.role_name, r.role_key, r.role_sort, r.data_scope, r.status, r.del_flag, r.create_time, r.remark from sys_role r left join sys_user_role ur on ur.role_id = r.role_id left join sys_user u on u.user_id = ur.user_id left join sys_dept d on u.dept_id = d.dept_id </sql> <select id="selectRolePermissionByUserId" parameterType="Long" resultMap="SysRoleResult"> <include refid="selectRoleVo"/> WHERE r.del_flag = '0' and ur.user_id = #{userId} </select>
- 通过查询
sys_role
表,同时连接sys_user_role
、sys_user
、sys_dept
表,将userId
对应的 SysRole 相关信息都一次性查询出来。 - 返回结果 SysRoleResult 的具体定义,点击 传送门 查看,实际就是 SysRole 实体类。
- 通过查询
<3>
处,调用 SysPermissionService 的#getMenuPermission(SysUser user)
方法,获得用户的 SysRoleMenu 的权限标识字符串的集合。<4>
处,返回用户信息的 AjaxResult 结果。
通过调用该 /getInfo
接口,前端就可以根据角色标识、又或者权限标识,实现对页面级别的按钮实现权限控制,进行有权限时显示,无权限时隐藏。
9. 获取路由信息
在 SysLoginController 中,定义了 /getRouters
接口,获取获取路由信息。代码如下:
// SysLoginController.java
@GetMapping("getRouters")
public AjaxResult getRouters() {
// 获得当前 LoginUser
LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
// 获得用户的 SysMenu 数组
SysUser user = loginUser.getUser();
List<SysMenu> menus = menuService.selectMenuTreeByUserId(user.getUserId());
// 构建路由 RouterVo 数组。可用于 Vue 构建管理后台的左边菜单
return AjaxResult.success(menuService.buildMenus(menus));
}