lsm/sa-token
1
0
Fork 0
mirror of https://gitee.com/dromara/sa-token.git synced 2025-04-05 17:37:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

补上注解拦截器里漏掉验证@SaCheckRole的bug

Browse Source
This commit is contained in:
shengzhang 2021-01-04 21:42:32 +08:00
parent dbba90d846
commit 7ffe6cb0e6
2 changed files with 33 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
sa-token-demo-springboot/src/main/java/com/pj/test/TestController.java
View File

@ -11,6 +11,7 @@ import com.fasterxml.jackson.databind.ObjectMapper;
import cn.dev33.satoken.annotation.SaCheckLogin; import cn.dev33.satoken.annotation.SaCheckLogin;
import cn.dev33.satoken.annotation.SaCheckPermission; import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.dev33.satoken.annotation.SaCheckRole;
import cn.dev33.satoken.annotation.SaMode; import cn.dev33.satoken.annotation.SaMode;
import cn.dev33.satoken.session.SaSessionCustomUtil; import cn.dev33.satoken.session.SaSessionCustomUtil;
import cn.dev33.satoken.stp.SaTokenInfo; import cn.dev33.satoken.stp.SaTokenInfo;
@ -151,13 +152,14 @@ public class TestController {
// 测试注解式鉴权， 浏览器访问： http://localhost:8081/test/atCheck // 测试注解式鉴权， 浏览器访问： http://localhost:8081/test/atCheck
@SaCheckLogin // 注解式鉴权：当前会话必须登录才能通过 @SaCheckLogin // 注解式鉴权：当前会话必须登录才能通过
@SaCheckRole("super-admin") // 注解式鉴权：当前会话必须具有指定角色标识才能通过
@SaCheckPermission("user-add") // 注解式鉴权：当前会话必须具有指定权限才能通过 @SaCheckPermission("user-add") // 注解式鉴权：当前会话必须具有指定权限才能通过
@RequestMapping("atCheck") @RequestMapping("atCheck")
public AjaxJson atCheck() { public AjaxJson atCheck() {
System.out.println("======================= 进入方法，测试注解鉴权接口 ========================= "); System.out.println("======================= 进入方法，测试注解鉴权接口 ========================= ");
System.out.println("只有通过注解鉴权，才能进入此方法"); System.out.println("只有通过注解鉴权，才能进入此方法");
StpUtil.checkActivityTimeout(); // StpUtil.checkActivityTimeout();
StpUtil.updateLastActivityToNow(); // StpUtil.updateLastActivityToNow();
return AjaxJson.getSuccess(); return AjaxJson.getSuccess();
} }

35
sa-token-spring-boot-starter/src/main/java/cn/dev33/satoken/interceptor/SaCheckInterceptor.java
View File

@ -8,6 +8,7 @@ import org.springframework.web.servlet.HandlerInterceptor;
import cn.dev33.satoken.annotation.SaCheckLogin; import cn.dev33.satoken.annotation.SaCheckLogin;
import cn.dev33.satoken.annotation.SaCheckPermission; import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.dev33.satoken.annotation.SaCheckRole;
import cn.dev33.satoken.annotation.SaMode; import cn.dev33.satoken.annotation.SaMode;
import cn.dev33.satoken.stp.StpLogic; import cn.dev33.satoken.stp.StpLogic;
import cn.dev33.satoken.stp.StpUtil; import cn.dev33.satoken.stp.StpUtil;
@ -58,25 +59,47 @@ public class SaCheckInterceptor implements HandlerInterceptor {
stpLogic.checkLogin(); stpLogic.checkLogin();
} }
// ----------- 验证角色
// 验证方法上的
SaCheckRole scr = method.getMethodAnnotation(SaCheckRole.class);
if(scr != null) {
String[] roleArray = scr.value();
if(scr.mode() == SaMode.AND) {
stpLogic.checkRoleAnd(roleArray); // 必须全部都有
} else {
stpLogic.checkRoleOr(roleArray); // 有一个就行了
}
}
// 验证类上的
scr = method.getBeanType().getAnnotation(SaCheckRole.class);
if(scr != null) {
String[] roleArray = scr.value();
if(scr.mode() == SaMode.AND) {
stpLogic.checkRoleAnd(roleArray); // 必须全部都有
} else {
stpLogic.checkRoleOr(roleArray); // 有一个就行了
}
}
// ----------- 验证权限 // ----------- 验证权限
// 验证方法上的 // 验证方法上的
SaCheckPermission scp = method.getMethodAnnotation(SaCheckPermission.class); SaCheckPermission scp = method.getMethodAnnotation(SaCheckPermission.class);
if(scp != null) { if(scp != null) {
String[] permissionCodeArray = scp.value(); String[] permissionArray = scp.value();
if(scp.mode() == SaMode.AND) { if(scp.mode() == SaMode.AND) {
stpLogic.checkPermissionAnd(permissionCodeArray); // 必须全部都有 stpLogic.checkPermissionAnd(permissionArray); // 必须全部都有
} else { } else {
stpLogic.checkPermissionOr(permissionCodeArray); // 有一个就行了 stpLogic.checkPermissionOr(permissionArray); // 有一个就行了
} }
} }
// 验证类上的 // 验证类上的
scp = method.getBeanType().getAnnotation(SaCheckPermission.class); scp = method.getBeanType().getAnnotation(SaCheckPermission.class);
if(scp != null) { if(scp != null) {
String[] permissionCodeArray = scp.value(); String[] permissionArray = scp.value();
if(scp.mode() == SaMode.AND) { if(scp.mode() == SaMode.AND) {
stpLogic.checkPermissionAnd(permissionCodeArray); // 必须全部都有 stpLogic.checkPermissionAnd(permissionArray); // 必须全部都有
} else { } else {
stpLogic.checkPermissionOr(permissionCodeArray); // 有一个就行了 stpLogic.checkPermissionOr(permissionArray); // 有一个就行了
} }
} }