init
liu
12 months ago
parent
d4b4cb637e
commit
51966631fa
@ -1,84 +0,0 @@
|
||||
package com.supervision.config;
|
||||
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import cn.hutool.json.JSONObject;
|
||||
import cn.hutool.json.JSONUtil;
|
||||
import cn.hutool.jwt.JWT;
|
||||
import cn.hutool.jwt.JWTUtil;
|
||||
import com.supervision.exception.BusinessException;
|
||||
import com.supervision.util.TokenUtil;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.lang.Nullable;
|
||||
import org.springframework.web.servlet.HandlerInterceptor;
|
||||
|
||||
import jakarta.servlet.http.HttpServletRequest;
|
||||
import jakarta.servlet.http.HttpServletResponse;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
@Slf4j
|
||||
public class JwtInterceptor implements HandlerInterceptor {
|
||||
|
||||
@Override
|
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
|
||||
//请求消息头获取用户ID
|
||||
String token = request.getHeader("token");
|
||||
if (StrUtil.isBlank(token)) {
|
||||
// 如果是swagger来的接口,说明这里是测试的,会伪造一个用户
|
||||
if (StrUtil.isNotBlank(request.getHeader("Knife4j-Gateway-Code"))) {
|
||||
cacheAuth(JWTUtil.parseToken(devActiveUser()));
|
||||
return true;
|
||||
} else {
|
||||
throw new BusinessException("当前用户未登录", HttpStatus.UNAUTHORIZED.value());
|
||||
}
|
||||
}
|
||||
JWT jwt = JWTUtil.parseToken(token);
|
||||
// 校验token是否过期,如果过期了,需要提示过期重新登录
|
||||
checkTokenExpire(jwt);
|
||||
cacheAuth(jwt);
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler,
|
||||
@Nullable Exception ex) throws Exception {
|
||||
// 请求结束,将用户信息从thread中移除
|
||||
clearAuth();
|
||||
HandlerInterceptor.super.afterCompletion(request, response, handler, ex);
|
||||
}
|
||||
|
||||
|
||||
private void checkTokenExpire(JWT jwt) {
|
||||
Object expireTime = jwt.getPayload("expireTime");
|
||||
long l = Long.parseLong(String.valueOf(expireTime));
|
||||
// 校验是否比当前时间大
|
||||
long currentTimeMillis = System.currentTimeMillis();
|
||||
if (currentTimeMillis > l) {
|
||||
throw new BusinessException("用户登录已过期,请重新登录", HttpStatus.UNAUTHORIZED.value());
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private void cacheAuth(JWT jwt) {
|
||||
try {
|
||||
JSONObject claimsJson = jwt.getPayload().getClaimsJson();
|
||||
ThreadCache.USER.set(claimsJson.toString());
|
||||
} catch (Exception e) {
|
||||
log.error("用户信息异常", e);
|
||||
}
|
||||
}
|
||||
|
||||
private String devActiveUser() {
|
||||
Map<String, Object> map = new HashMap<>();
|
||||
map.put("id", "1");
|
||||
map.put("account", "test");
|
||||
map.put("name", "测试账户");
|
||||
return TokenUtil.creatToken(JSONUtil.toJsonStr(map));
|
||||
}
|
||||
|
||||
private void clearAuth() {
|
||||
ThreadCache.USER.remove();
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,62 @@
|
||||
package com.supervision.config;
|
||||
|
||||
import cn.dev33.satoken.context.SaHolder;
|
||||
import cn.dev33.satoken.filter.SaServletFilter;
|
||||
import cn.dev33.satoken.router.SaRouter;
|
||||
import cn.dev33.satoken.stp.StpUtil;
|
||||
import cn.dev33.satoken.util.SaResult;
|
||||
import cn.hutool.core.collection.CollUtil;
|
||||
import cn.hutool.core.util.ArrayUtil;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
|
||||
@Configuration
|
||||
public class SaTokenConfigure {
|
||||
|
||||
public String[] ignorePathPatterns() {
|
||||
List<String> paths = new ArrayList<>();
|
||||
paths.add("/swagger-resources/**");
|
||||
paths.add("/webjars/**");
|
||||
paths.add("/v2/**");
|
||||
paths.add("/swagger-ui.html/**");
|
||||
paths.add("/doc.html/**");
|
||||
paths.add("/error");
|
||||
paths.add("/favicon.ico");
|
||||
paths.add("/user/login");
|
||||
return ArrayUtil.toArray(paths,String.class);
|
||||
}
|
||||
|
||||
/**
|
||||
* 注册 [Sa-Token全局过滤器]
|
||||
*/
|
||||
@Bean
|
||||
public SaServletFilter getSaServletFilter() {
|
||||
return new SaServletFilter()
|
||||
|
||||
// 指定 拦截路由 与 放行路由
|
||||
.addInclude("/**")
|
||||
// 排除掉部分不需要登录的路由
|
||||
.addExclude(ignorePathPatterns())
|
||||
|
||||
// 认证函数: 每次请求执行
|
||||
.setAuth(obj -> {
|
||||
System.out.println("---------- 进入Sa-Token全局认证 -----------");
|
||||
// 登录认证 -- 拦截所有路由，并排除/user/doLogin 用于开放登录
|
||||
SaRouter.match("/**", "/user/doLogin", StpUtil::checkLogin);
|
||||
|
||||
// 更多拦截处理方式，请参考“路由拦截式鉴权”章节 */
|
||||
})
|
||||
|
||||
// 异常处理函数：每次认证函数发生异常时执行此函数
|
||||
.setError(e -> {
|
||||
System.out.println("---------- 进入Sa-Token异常处理 -----------");
|
||||
return SaResult.error(e.getMessage());
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
@ -1,58 +0,0 @@
|
||||
/*
|
||||
* 文 件 名: CustomException
|
||||
* 版 权:
|
||||
* 描 述: <描述>
|
||||
* 修 改 人: RedName
|
||||
* 修改时间: 2022/8/5
|
||||
* 跟踪单号: <跟踪单号>
|
||||
* 修改单号: <修改单号>
|
||||
* 修改内容: <修改内容>
|
||||
*/
|
||||
package com.supervision.exception;
|
||||
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
|
||||
/**
|
||||
* <功能详细描述>
|
||||
* 自定义异常
|
||||
*
|
||||
* @author ljt
|
||||
* @version [版本号, 2022/8/5]
|
||||
* @see [相关类/方法]
|
||||
* @since [产品/模块版本]
|
||||
*/
|
||||
@Slf4j
|
||||
public class HumanException extends RuntimeException {
|
||||
|
||||
/**
|
||||
* 房间状态获取异常时,前端刷新房间,重新进入
|
||||
*/
|
||||
private static final Integer HUMAN_ERROR = 6001;
|
||||
/**
|
||||
* 异常编码
|
||||
*/
|
||||
private final Integer code;
|
||||
|
||||
/**
|
||||
* 异常信息
|
||||
*/
|
||||
private final String message;
|
||||
|
||||
private HumanException(Integer code, String message) {
|
||||
this.code = code;
|
||||
this.message = message;
|
||||
}
|
||||
|
||||
public static HumanException humanError(String message) {
|
||||
return new HumanException(HUMAN_ERROR, message);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getMessage() {
|
||||
return message;
|
||||
}
|
||||
|
||||
public Integer getCode() {
|
||||
return code;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue