Initial project

src/main/java/com/imyeyu/spring/util/GlobalExceptionHandler.java

@@ -0,0 +1,103 @@
+package com.imyeyu.spring.util;
+
+import jakarta.servlet.ServletException;
+import jakarta.validation.ValidationException;
+import com.imyeyu.java.bean.timi.TimiCode;
+import com.imyeyu.java.bean.timi.TimiException;
+import com.imyeyu.java.bean.timi.TimiResponse;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.TypeMismatchException;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.converter.HttpMessageConversionException;
+import org.springframework.validation.BindException;
+import org.springframework.validation.FieldError;
+import org.springframework.web.bind.MethodArgumentNotValidException;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.RestControllerAdvice;
+
+/**
+ * 全局异常处理器
+ *
+ * @author 夜雨
+ * @version 2023-05-06 16:28
+ */
+@RestControllerAdvice
+public class GlobalExceptionHandler {
+
+	private static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);
+	private static final String DEV_LANG_CONFIG = "dev.lang";
+
+	@Value("${spring.profiles.active}")
+	private String env;
+
+	/**
+	 * @param e
+	 * @return
+	 */
+	@ExceptionHandler(HttpMessageConversionException.class)
+	public TimiResponse<?> conversionException(HttpMessageConversionException e) {
+		log.warn(e.getMessage());
+		if (env.contains("dev") || log.isDebugEnabled()) {
+			log.error("conversion error", e);
+		}
+		return new TimiResponse<>(TimiCode.ARG_BAD);
+	}
+
+	/**
+	 * 请求异常
+	 *
+	 * @param e 异常
+	 * @return 异常返回
+	 */
+	@ExceptionHandler(ServletException.class)
+	public TimiResponse<?> headerException(ServletException e) {
+		log.warn(e.getMessage());
+		if (env.contains("dev") || log.isDebugEnabled()) {
+			log.error("header error", e);
+		}
+		return new TimiResponse<>(TimiCode.REQUEST_BAD);
+	}
+
+	/**
+	 * 接口入参基本校验异常
+	 *
+	 * @param e 异常
+	 * @return 异常返回
+	 */
+	@ExceptionHandler({BindException.class, ValidationException.class, MethodArgumentNotValidException.class, TypeMismatchException.class})
+	public TimiResponse<?> paramsException(Exception e) {
+		if (e instanceof MethodArgumentNotValidException subE) {
+			log.warn("request error", e);
+			FieldError error = subE.getBindingResult().getFieldError();
+			if (error != null) {
+				return new TimiResponse<>(TimiCode.ARG_BAD, error.getDefaultMessage());
+			}
+		}
+		if (env.startsWith("dev") || log.isDebugEnabled()) {
+			log.error("request error", e);
+		}
+		return new TimiResponse<>(TimiCode.REQUEST_BAD);
+	}
+
+	/**
+	 * 全局异常
+	 *
+	 * @param e 异常
+	 * @return 异常返回
+	 */
+	@ExceptionHandler(Throwable.class)
+	public TimiResponse<?> error(Throwable e) {
+		if (e instanceof TimiException timiE) {
+			// TODO 400 以下即使是开发环境也不算异常
+			if (env.startsWith("dev") || log.isDebugEnabled()) {
+				log.error(timiE.getMessage(), e);
+			}
+			// 一般异常
+			return timiE.toResponse();
+		}
+		// 致命异常
+		log.error("fatal error", e);
+		return new TimiResponse<>(TimiCode.ERROR);
+	}
+}