1.后台登录接口
2.异常处理
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
package com.czg.config;
|
||||
|
||||
import cn.dev33.satoken.stp.StpInterface;
|
||||
import com.czg.sa.StpKit;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 自定义权限加载接口实现类
|
||||
* @author Administrator
|
||||
*/
|
||||
@Component // 保证此类被 SpringBoot 扫描,完成 Sa-Token 的自定义权限验证扩展
|
||||
public class StpInterfaceImpl implements StpInterface {
|
||||
|
||||
/**
|
||||
* 返回一个账号所拥有的权限码集合
|
||||
*/
|
||||
@Override
|
||||
public List<String> getPermissionList(Object loginId, String loginType) {
|
||||
if ("admin".equals(loginType)) {
|
||||
Object value = StpKit.ADMIN.getSaTokenDao().getObject("sa:permissionList:" + loginType + ":" + loginId);
|
||||
if (value instanceof List<?> list) {
|
||||
return (List<String>) list;
|
||||
}
|
||||
}
|
||||
|
||||
return new ArrayList<>();
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回一个账号所拥有的角色标识集合 (权限与角色可分开校验)
|
||||
*/
|
||||
@Override
|
||||
public List<String> getRoleList(Object loginId, String loginType) {
|
||||
if ("admin".equals(loginType)) {
|
||||
Object value = StpKit.ADMIN.getSaTokenDao().getObject("sa:roleList:" + loginType + ":" + loginId);
|
||||
if (value instanceof List<?> a) {
|
||||
return (List<String>) a;
|
||||
}
|
||||
}
|
||||
return new ArrayList<>();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,47 @@
|
||||
package com.czg.sa;
|
||||
|
||||
import cn.dev33.satoken.stp.StpLogic;
|
||||
import cn.hutool.core.collection.CollectionUtil;
|
||||
import cn.hutool.core.util.ArrayUtil;
|
||||
import com.alibaba.fastjson2.JSONObject;
|
||||
import jakarta.annotation.Resource;
|
||||
import org.springframework.data.redis.core.StringRedisTemplate;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* @author Administrator
|
||||
*/
|
||||
public class MyStpLogic extends StpLogic {
|
||||
@Resource
|
||||
private StringRedisTemplate redisTemplate;
|
||||
|
||||
/**
|
||||
* 初始化 StpLogic, 并指定账号类型
|
||||
*
|
||||
* @param loginType 账号类型标识
|
||||
*/
|
||||
public MyStpLogic(String loginType) {
|
||||
super(loginType);
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加当前账号的角色
|
||||
* @param roleList 角色列表
|
||||
* @return 当前实例
|
||||
*/
|
||||
public MyStpLogic addRoleList(List<String> roleList) {
|
||||
this.getSaTokenDao().setObject("sa:roleList:" + getLoginType() + ":" + getLoginId(), roleList, getTokenTimeout());
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加当前账号的权限
|
||||
* @param permissionList 角色列表
|
||||
* @return 当前实例
|
||||
*/
|
||||
public MyStpLogic addPermissionList(List<String> permissionList) {
|
||||
this.getSaTokenDao().setObject("sa:permissionList:" + getLoginType() + ":" + getLoginId(), permissionList, getTokenTimeout());
|
||||
return this;
|
||||
}
|
||||
}
|
||||
@@ -11,11 +11,11 @@ public class StpKit {
|
||||
/**
|
||||
* Admin 会话对象,管理 Admin 表所有账号的登录、权限认证
|
||||
*/
|
||||
public static final StpLogic ADMIN = new StpLogic("admin");
|
||||
public static final MyStpLogic ADMIN = new MyStpLogic("admin");
|
||||
|
||||
/**
|
||||
* User 会话对象,管理 User 表所有账号的登录、权限认证
|
||||
*/
|
||||
public static final StpLogic USER = new StpLogic("user");
|
||||
public static final MyStpLogic USER = new MyStpLogic("user");
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user