init
liu
12 months ago
parent
edb6297ff4
commit
a2e5eac4dc
@ -0,0 +1,91 @@
|
|||||||
|
package com.supervision.knowsub.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 jakarta.servlet.http.HttpServletRequest;
|
||||||
|
import jakarta.servlet.http.HttpServletResponse;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.jetbrains.annotations.NotNull;
|
||||||
|
import org.springframework.http.HttpStatus;
|
||||||
|
import org.springframework.lang.Nullable;
|
||||||
|
import org.springframework.web.servlet.HandlerInterceptor;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
@Slf4j
|
||||||
|
public class JwtInterceptor implements HandlerInterceptor {
|
||||||
|
|
||||||
|
public JwtInterceptor() {
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
|
||||||
|
//请求消息头获取用户ID
|
||||||
|
String token = request.getHeader("token");
|
||||||
|
if (StrUtil.isBlank(token)) {
|
||||||
|
// 如果是swagger来的接口,说明这里是测试的,会伪造一个用户
|
||||||
|
String referer = request.getHeader("Referer");
|
||||||
|
if (StrUtil.isNotBlank(referer) && StrUtil.contains(referer, "doc.html")) {
|
||||||
|
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(@NotNull HttpServletRequest request, @NotNull HttpServletResponse response, @NotNull 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,36 @@
|
|||||||
|
/*
|
||||||
|
* 文 件 名: MybatisPlusConfig
|
||||||
|
* 版 权:
|
||||||
|
* 描 述: <描述>
|
||||||
|
* 修 改 人: RedName
|
||||||
|
* 修改时间: 2022/8/9
|
||||||
|
* 跟踪单号: <跟踪单号>
|
||||||
|
* 修改单号: <修改单号>
|
||||||
|
* 修改内容: <修改内容>
|
||||||
|
*/
|
||||||
|
package com.supervision.knowsub.config;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.annotation.DbType;
|
||||||
|
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
||||||
|
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <功能详细描述>
|
||||||
|
*
|
||||||
|
* @author RedName
|
||||||
|
* @version [版本号, 2022/8/9]
|
||||||
|
* @see [相关类/方法]
|
||||||
|
* @since [产品/模块版本]
|
||||||
|
*/
|
||||||
|
@Configuration
|
||||||
|
public class MybatisPlusConfig {
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
public MybatisPlusInterceptor mybatisPlusInterceptor() {
|
||||||
|
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
||||||
|
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
|
||||||
|
return interceptor;
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,138 @@
|
|||||||
|
package com.supervision.knowsub.config;
|
||||||
|
|
||||||
|
import cn.hutool.core.util.StrUtil;
|
||||||
|
import cn.hutool.json.JSONUtil;
|
||||||
|
import com.supervision.domain.GlobalResult;
|
||||||
|
import com.supervision.exception.BusinessException;
|
||||||
|
import com.supervision.exception.HumanException;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.core.MethodParameter;
|
||||||
|
import org.springframework.http.HttpStatus;
|
||||||
|
import org.springframework.http.MediaType;
|
||||||
|
import org.springframework.http.converter.HttpMessageConverter;
|
||||||
|
import org.springframework.http.server.ServerHttpRequest;
|
||||||
|
import org.springframework.http.server.ServerHttpResponse;
|
||||||
|
import org.springframework.lang.Nullable;
|
||||||
|
import org.springframework.validation.BindException;
|
||||||
|
import org.springframework.web.bind.MethodArgumentNotValidException;
|
||||||
|
import org.springframework.web.bind.annotation.ExceptionHandler;
|
||||||
|
import org.springframework.web.bind.annotation.RestController;
|
||||||
|
import org.springframework.web.bind.annotation.RestControllerAdvice;
|
||||||
|
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
|
||||||
|
|
||||||
|
import java.util.Objects;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 统一返回
|
||||||
|
*
|
||||||
|
* @author wb
|
||||||
|
* @date 2022/3/10 13:24
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
@RestControllerAdvice(annotations = RestController.class, basePackages = {"com.supervision.**.controller"})
|
||||||
|
public class ResponseConfig implements ResponseBodyAdvice<Object> {
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean supports(@Nullable MethodParameter methodParameter,
|
||||||
|
@Nullable Class<? extends HttpMessageConverter<?>> aClass) {
|
||||||
|
assert methodParameter != null;
|
||||||
|
String lowerCase = methodParameter.getDeclaringClass().getName().toLowerCase();
|
||||||
|
log.info(lowerCase);
|
||||||
|
if (lowerCase.contains("swagger") || lowerCase.contains("springdoc")) {
|
||||||
|
return false;
|
||||||
|
} else {
|
||||||
|
log.info("--------{}---------", lowerCase);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@SuppressWarnings("all")
|
||||||
|
public Object beforeBodyWrite(Object o, @Nullable MethodParameter methodParameter, @Nullable MediaType mediaType,
|
||||||
|
@Nullable Class<? extends HttpMessageConverter<?>> aClass, @Nullable ServerHttpRequest serverHttpRequest,
|
||||||
|
@Nullable ServerHttpResponse serverHttpResponse) {
|
||||||
|
serverHttpResponse.getHeaders().set("Content-Type", "application/json");
|
||||||
|
|
||||||
|
if (Objects.isNull(o)) {
|
||||||
|
return GlobalResult.ok(null, "success");
|
||||||
|
}
|
||||||
|
if (o instanceof GlobalResult) {
|
||||||
|
return o;
|
||||||
|
}
|
||||||
|
// 对于String类型的返回值需要进行特殊处理
|
||||||
|
if (o instanceof String) {
|
||||||
|
return JSONUtil.toJsonStr(GlobalResult.ok(o, "success"));
|
||||||
|
}
|
||||||
|
return GlobalResult.ok(o, "success");
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 业务异常处理
|
||||||
|
*
|
||||||
|
* @param exception 业务异常
|
||||||
|
* @return 通用返回值
|
||||||
|
*/
|
||||||
|
@ExceptionHandler(BusinessException.class)
|
||||||
|
public GlobalResult<?> businessExceptionResponse(BusinessException exception) {
|
||||||
|
log.error(exception.getMessage(), exception);
|
||||||
|
Integer code = null == exception.getCode() ? HttpStatus.INTERNAL_SERVER_ERROR.value(): exception.getCode();
|
||||||
|
String message = StrUtil.isNotEmpty(exception.getMessage()) ? exception.getMessage() : "业务异常";
|
||||||
|
return GlobalResult.error( code, exception.getMessage(), message);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 数字人异常处理
|
||||||
|
*
|
||||||
|
* @param exception 业务异常
|
||||||
|
* @return 通用返回值
|
||||||
|
*/
|
||||||
|
@ExceptionHandler(HumanException.class)
|
||||||
|
public GlobalResult<?> humanExceptionResponse(HumanException exception) {
|
||||||
|
log.error(exception.getMessage(), exception);
|
||||||
|
return GlobalResult.error(exception.getCode(), exception.getMessage(), "数字人异常");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 参数验证异常处理
|
||||||
|
*
|
||||||
|
* @param exception 参数验证异常
|
||||||
|
* @return 通用返回值
|
||||||
|
*/
|
||||||
|
@ExceptionHandler({MethodArgumentNotValidException.class, BindException.class})
|
||||||
|
public GlobalResult<?> validationExceptionResponse(MethodArgumentNotValidException exception) {
|
||||||
|
log.error(exception.getMessage(), exception);
|
||||||
|
// 格式化错误信息
|
||||||
|
String errorMsg = exception.getBindingResult().getFieldErrors().stream()
|
||||||
|
.map(e -> e.getField() + ":" + e.getDefaultMessage()).collect(Collectors.joining("、"));
|
||||||
|
return GlobalResult.error(HttpStatus.INTERNAL_SERVER_ERROR.value(), errorMsg, "参数验证异常");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 添加手动校验参数的异常处理
|
||||||
|
* @param exception 参数验证异常
|
||||||
|
* @return 通用返回值
|
||||||
|
*/
|
||||||
|
@ExceptionHandler(IllegalArgumentException.class)
|
||||||
|
public GlobalResult<?> manualValidationExceptionResponse(IllegalArgumentException exception) {
|
||||||
|
log.error(exception.getMessage(), exception);
|
||||||
|
return GlobalResult.error(HttpStatus.INTERNAL_SERVER_ERROR.value(), exception.getMessage(), "参数验证异常");
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 未知异常处理
|
||||||
|
*
|
||||||
|
* @param exception 未知异常
|
||||||
|
* @return 通用返回值
|
||||||
|
*/
|
||||||
|
@ExceptionHandler(Exception.class)
|
||||||
|
public GlobalResult<?> validationExceptionResponse(Exception exception) {
|
||||||
|
log.error(exception.getMessage(), exception);
|
||||||
|
return GlobalResult.error(HttpStatus.INTERNAL_SERVER_ERROR.value(), "未知错误", "未知错误");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,71 @@
|
|||||||
|
package com.supervision.knowsub.config;
|
||||||
|
|
||||||
|
import com.fasterxml.jackson.databind.DeserializationFeature;
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import com.fasterxml.jackson.databind.SerializationFeature;
|
||||||
|
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
|
||||||
|
import com.fasterxml.jackson.datatype.jsr310.deser.LocalDateTimeDeserializer;
|
||||||
|
import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
|
||||||
|
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
import java.time.format.DateTimeFormatter;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
@Configuration
|
||||||
|
public class WebConfig implements WebMvcConfigurer {
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void addInterceptors(InterceptorRegistry registry) {
|
||||||
|
// 添加权限拦截器
|
||||||
|
registry.addInterceptor(new JwtInterceptor())
|
||||||
|
.addPathPatterns("/**")
|
||||||
|
.excludePathPatterns(ignorePathPatterns());
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<String> ignorePathPatterns() {
|
||||||
|
List<String> paths = new ArrayList<>();
|
||||||
|
paths.add("/swagger-resources/**");
|
||||||
|
paths.add("/webjars/**");
|
||||||
|
paths.add("/v3/**");
|
||||||
|
paths.add("/swagger-ui.html/**");
|
||||||
|
paths.add("/swagger-ui/**");
|
||||||
|
paths.add("/webjars/");
|
||||||
|
paths.add("/doc.html/**");
|
||||||
|
paths.add("/error");
|
||||||
|
paths.add("/favicon.ico");
|
||||||
|
paths.add("/user/login");
|
||||||
|
paths.add("/user/register");
|
||||||
|
paths.add("/user/changePassWord");
|
||||||
|
paths.add("/user/checkAccount");
|
||||||
|
paths.add("/webSocket/**");
|
||||||
|
paths.add("/ask/downloadTalkVideo");
|
||||||
|
paths.add("/fileManage/downloadFile");
|
||||||
|
paths.add("/aqLibrary/downloadQuestionLibraryTemplate");
|
||||||
|
paths.add("/medicalRecManage/downloadMedicalAnswerTemplate");
|
||||||
|
paths.add("/qaKnowledge/**");
|
||||||
|
// 开发环境,放开不校验token.每次修改这里需要重启(热部署不行)
|
||||||
|
// paths.add("/**");
|
||||||
|
return paths;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
public ObjectMapper objectMapper() {
|
||||||
|
ObjectMapper objectMapper = new ObjectMapper();
|
||||||
|
objectMapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
|
||||||
|
|
||||||
|
JavaTimeModule javaTimeModule = new JavaTimeModule();
|
||||||
|
javaTimeModule.addSerializer(LocalDateTime.class, new LocalDateTimeSerializer(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
|
||||||
|
javaTimeModule.addDeserializer(LocalDateTime.class, new LocalDateTimeDeserializer(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
|
||||||
|
|
||||||
|
objectMapper.registerModule(javaTimeModule);
|
||||||
|
objectMapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS);
|
||||||
|
|
||||||
|
return objectMapper;
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,76 @@
|
|||||||
|
/*
|
||||||
|
* 文 件 名: CustomException
|
||||||
|
* 版 权:
|
||||||
|
* 描 述: <描述>
|
||||||
|
* 修 改 人: RedName
|
||||||
|
* 修改时间: 2022/8/5
|
||||||
|
* 跟踪单号: <跟踪单号>
|
||||||
|
* 修改单号: <修改单号>
|
||||||
|
* 修改内容: <修改内容>
|
||||||
|
*/
|
||||||
|
package com.supervision.knowsub.exception;
|
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.http.HttpStatus;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <功能详细描述>
|
||||||
|
* 自定义异常
|
||||||
|
*
|
||||||
|
* @author ljt
|
||||||
|
* @version [版本号, 2022/8/5]
|
||||||
|
* @see [相关类/方法]
|
||||||
|
* @since [产品/模块版本]
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
public class BusinessException extends RuntimeException {
|
||||||
|
/**
|
||||||
|
* 异常编码
|
||||||
|
*/
|
||||||
|
private final Integer code;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 异常信息
|
||||||
|
*/
|
||||||
|
private final String message;
|
||||||
|
|
||||||
|
public BusinessException(Throwable cause) {
|
||||||
|
super(cause);
|
||||||
|
this.code = HttpStatus.INTERNAL_SERVER_ERROR.value();
|
||||||
|
this.message = null;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public BusinessException(Throwable cause, String message) {
|
||||||
|
super(cause);
|
||||||
|
this.code = HttpStatus.INTERNAL_SERVER_ERROR.value();
|
||||||
|
this.message = message;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public BusinessException(String message) {
|
||||||
|
this.code = HttpStatus.INTERNAL_SERVER_ERROR.value();
|
||||||
|
this.message = message;
|
||||||
|
}
|
||||||
|
|
||||||
|
public BusinessException(String message, Integer code) {
|
||||||
|
this.message = message;
|
||||||
|
this.code = code;
|
||||||
|
}
|
||||||
|
|
||||||
|
public BusinessException(String message, Throwable e) {
|
||||||
|
super(message, e);
|
||||||
|
log.error(message, e);
|
||||||
|
this.code = HttpStatus.INTERNAL_SERVER_ERROR.value();
|
||||||
|
this.message = message;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getMessage() {
|
||||||
|
return message;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Integer getCode() {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,19 @@
|
|||||||
|
package com.supervision.knowsub.util;
|
||||||
|
|
||||||
|
import cn.hutool.json.JSONObject;
|
||||||
|
import cn.hutool.json.JSONUtil;
|
||||||
|
import cn.hutool.jwt.JWTUtil;
|
||||||
|
import cn.hutool.jwt.signers.JWTSigner;
|
||||||
|
import cn.hutool.jwt.signers.JWTSignerUtil;
|
||||||
|
|
||||||
|
public class TokenUtil {
|
||||||
|
|
||||||
|
public static String creatToken(String userInfo){
|
||||||
|
final JWTSigner signer = JWTSignerUtil.hs256("123456".getBytes());
|
||||||
|
JSONObject info = JSONUtil.parseObj(userInfo);
|
||||||
|
// 过期时间一天,同时这个字段也作为单点登录使用
|
||||||
|
info.putOnce("expireTime",System.currentTimeMillis() + 1000 * 60 * 60 * 24);
|
||||||
|
info.putOnce("issueTime",System.currentTimeMillis());
|
||||||
|
return JWTUtil.createToken(info, signer);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue