package com.tcm.training.admin.common;
|
|
import cn.dev33.satoken.exception.NotLoginException;
|
import com.fasterxml.jackson.databind.exc.InvalidFormatException;
|
import com.tcm.training.admin.app.auth.exception.InviteCodeRequiredException;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.validation.BindException;
|
import org.springframework.validation.FieldError;
|
import org.springframework.http.converter.HttpMessageNotReadableException;
|
import org.springframework.web.bind.MethodArgumentNotValidException;
|
import org.springframework.web.bind.annotation.ExceptionHandler;
|
import org.springframework.web.bind.annotation.RestControllerAdvice;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.io.IOException;
|
import java.time.LocalDate;
|
import java.time.LocalDateTime;
|
import java.time.LocalTime;
|
import java.time.format.DateTimeParseException;
|
|
@RestControllerAdvice
|
public class GlobalExceptionHandler {
|
|
private static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);
|
private static final String DEFAULT_SERVER_ERROR_MESSAGE = "服务器开小差啦，请稍后重试";
|
|
@ExceptionHandler(NotLoginException.class)
|
public ApiResponse<Void> handleNotLogin(NotLoginException e, HttpServletRequest request) {
|
log.warn("未登录访问 [{} {}]: {}", request.getMethod(), request.getRequestURI(), e.getMessage());
|
return ApiResponse.fail(401, "未登录");
|
}
|
|
@ExceptionHandler({MethodArgumentNotValidException.class, BindException.class})
|
public ApiResponse<Void> handleValidate(Exception e, HttpServletRequest request) {
|
log.warn("请求参数校验失败 [{} {}]", request.getMethod(), request.getRequestURI(), e);
|
if (e instanceof MethodArgumentNotValidException) {
|
FieldError fieldError = ((MethodArgumentNotValidException) e).getBindingResult().getFieldError();
|
if (fieldError != null && fieldError.getDefaultMessage() != null) {
|
return ApiResponse.fail(400, fieldError.getDefaultMessage());
|
}
|
}
|
if (e instanceof BindException) {
|
FieldError fieldError = ((BindException) e).getBindingResult().getFieldError();
|
if (fieldError != null && fieldError.getDefaultMessage() != null) {
|
return ApiResponse.fail(400, fieldError.getDefaultMessage());
|
}
|
}
|
return ApiResponse.fail(400, "参数错误");
|
}
|
|
@ExceptionHandler(InviteCodeRequiredException.class)
|
public ApiResponse<Void> handleInviteCodeRequired(InviteCodeRequiredException e, HttpServletRequest request) {
|
log.warn("第三方授权缺少邀请码 [{} {}]: {}", request.getMethod(), request.getRequestURI(), e.getMessage());
|
return ApiResponse.fail(506, e.getMessage());
|
}
|
|
@ExceptionHandler(IllegalArgumentException.class)
|
public ApiResponse<Void> handleIllegalArgument(IllegalArgumentException e, HttpServletRequest request) {
|
log.warn("业务参数异常 [{} {}]: {}", request.getMethod(), request.getRequestURI(), e.getMessage(), e);
|
return ApiResponse.fail(400, e.getMessage());
|
}
|
|
@ExceptionHandler(IllegalStateException.class)
|
public ApiResponse<Void> handleIllegalState(IllegalStateException e, HttpServletRequest request) {
|
log.warn("业务状态异常 [{} {}]: {}", request.getMethod(), request.getRequestURI(), e.getMessage(), e);
|
return ApiResponse.fail(500, resolveClientMessage(e.getMessage()));
|
}
|
|
@ExceptionHandler(HttpMessageNotReadableException.class)
|
public ApiResponse<Void> handleHttpMessageNotReadable(HttpMessageNotReadableException e, HttpServletRequest request) {
|
log.warn("请求体解析失败 [{} {}]", request.getMethod(), request.getRequestURI(), e);
|
return ApiResponse.fail(400, resolveReadableMessage(e));
|
}
|
|
@ExceptionHandler(IOException.class)
|
public ApiResponse<Void> handleIO(IOException e, HttpServletRequest request) {
|
log.error("IO异常 [{} {}]", request.getMethod(), request.getRequestURI(), e);
|
return ApiResponse.fail(500, "文件操作失败: " + e.getMessage());
|
}
|
|
@ExceptionHandler(Exception.class)
|
public ApiResponse<Void> handleException(Exception e, HttpServletRequest request) {
|
log.error("未捕获异常 [{} {}]", request.getMethod(), request.getRequestURI(), e);
|
return ApiResponse.fail(500, DEFAULT_SERVER_ERROR_MESSAGE);
|
}
|
|
private String resolveClientMessage(String message) {
|
if (message == null || message.trim().isEmpty()) {
|
return DEFAULT_SERVER_ERROR_MESSAGE;
|
}
|
return message;
|
}
|
|
private String resolveReadableMessage(Throwable throwable) {
|
Throwable current = throwable;
|
while (current != null) {
|
if (current instanceof InvalidFormatException) {
|
String message = resolveTimeFormatMessage(((InvalidFormatException) current).getTargetType());
|
if (message != null) {
|
return message;
|
}
|
}
|
if (current instanceof DateTimeParseException) {
|
return "时间格式错误，请使用 "
|
+ DateTimes.LOCAL_DATE_TIME_PATTERN + " / "
|
+ DateTimes.LOCAL_DATE_PATTERN + " / "
|
+ DateTimes.LOCAL_TIME_PATTERN;
|
}
|
current = current.getCause();
|
}
|
return "请求参数格式错误";
|
}
|
|
private String resolveTimeFormatMessage(Class<?> targetType) {
|
if (targetType == null) {
|
return null;
|
}
|
if (LocalDateTime.class.isAssignableFrom(targetType)) {
|
return "时间格式错误，需 " + DateTimes.LOCAL_DATE_TIME_PATTERN;
|
}
|
if (LocalDate.class.isAssignableFrom(targetType)) {
|
return "日期格式错误，需 " + DateTimes.LOCAL_DATE_PATTERN;
|
}
|
if (LocalTime.class.isAssignableFrom(targetType)) {
|
return "时间格式错误，需 " + DateTimes.LOCAL_TIME_PATTERN;
|
}
|
return null;
|
}
|
}
|
