Java SpringBoot通用异常处理类

1. package tacos.advisor;
3. import com.fasterxml.jackson.databind.exc.InvalidFormatException;
4. import org.slf4j.Logger;
5. import org.slf4j.LoggerFactory;
6. import org.springframework.context.support.DefaultMessageSourceResolvable;
7. import org.springframework.http.ResponseEntity;
8. import org.springframework.http.converter.HttpMessageNotReadableException;
9. import org.springframework.validation.BindException;
10. import org.springframework.web.bind.MethodArgumentNotValidException;
11. import org.springframework.web.bind.MissingPathVariableException;
12. import org.springframework.web.bind.MissingServletRequestParameterException;
13. import org.springframework.web.bind.annotation.ExceptionHandler;
14. import org.springframework.web.bind.annotation.RestControllerAdvice;
15. import tacos.model.APIResult;
17. import javax.validation.ConstraintViolation;
18. import javax.validation.ConstraintViolationException;
19. import java.util.Optional;
20. import java.util.regex.Matcher;
21. import java.util.regex.Pattern;
22. import java.util.stream.Collectors;
24. @RestControllerAdvice
25. @SuppressWarnings("unused")
26. public class GlobalExceptionHandler {
27.     private ResponseEntity<APIResult<Object>> logObject(int code, String msg, String category) {
28.         Logger log = LoggerFactory.getLogger(category);
29.         log.error("fail: {}", msg);
30.         return ResponseEntity.ok().body(APIResult.fail(code, msg));
31.     }
33.     @ExceptionHandler(Exception.class)
34.     public ResponseEntity<APIResult<Object>> handleException(Throwable exception) {
35.         Logger log = LoggerFactory.getLogger(exception.getStackTrace()[0].getClassName());
36.         log.error("fail: {}", "", exception);
37.         return ResponseEntity.ok().body(APIResult.unknown(exception.getMessage()));
38.     }
40.     @ExceptionHandler(MethodArgumentNotValidException.class)
41.     public ResponseEntity<APIResult<Object>> handleMethodArgumentNotValidException(
42.             MethodArgumentNotValidException exception) {
43.         String msg = Optional.of(exception.getBindingResult().getFieldErrors().stream()
44.                 .map(DefaultMessageSourceResolvable::getDefaultMessage)
45.                 .collect(Collectors.joining("\r\n"))).orElse("Validate failed");
46.         return logObject(APIResult.ERROR_PARAM, msg, exception.getStackTrace()[0].getClassName());
47.     }
49.     @ExceptionHandler(MissingPathVariableException.class)
50.     public ResponseEntity<APIResult<Object>> handleMissingPathVariableException(MissingPathVariableException exception) {
51.         String msg = String.format("The %s is required", exception.getVariableName());
52.         return logObject(APIResult.ERROR_PARAM, msg, exception.getStackTrace()[0].getClassName());
53.     }
55.     @ExceptionHandler(MissingServletRequestParameterException.class)
56.     public ResponseEntity<APIResult<Object>> handleMissingServletRequestParameterException(
57.             MissingServletRequestParameterException exception) {
58.         String msg = String.format("The %s is required", exception.getParameterName());
59.         return logObject(APIResult.ERROR_PARAM, msg, exception.getStackTrace()[0].getClassName());
60.     }
62.     @ExceptionHandler(ConstraintViolationException.class)
63.     public ResponseEntity<APIResult<Object>> handleConstraintViolationException(ConstraintViolationException exception) {
64.         String msg = Optional.of(exception.getConstraintViolations().stream()
65.                 .map(ConstraintViolation::getMessage)
66.                 .collect(Collectors.joining("\r\n"))).orElse("Validate failed");
67.         return logObject(APIResult.ERROR_PARAM, msg, exception.getStackTrace()[0].getClassName());
68.     }
70.     private static final Pattern ENUM_MSG = Pattern.compile("values accepted for Enum class: \\[(.*?)\\];(.*)from String \\\"(.*)\\\": (.*)\\n(.*)\\[\\\"(.*)\\\"\\]\\)", Pattern.MULTILINE);
72.     @ExceptionHandler(HttpMessageNotReadableException.class)
73.     public ResponseEntity<APIResult<Object>> handleHttpMessageNotReadableException(HttpMessageNotReadableException exception) {
74.         if (exception.getCause() != null && exception.getCause() instanceof InvalidFormatException) {
75.             Matcher match = ENUM_MSG.matcher(exception.getMessage());
76.             if (match.find()) {
77.                 return logObject(APIResult.ERROR_PARAM,
78.                         String.format("%s: Invalid value '%s', possible values - [%s]", match.group(6), match.group(3), match.group(1)),
79.                         exception.getStackTrace()[0].getClassName());
80.             }
81.         }
83.         return logObject(APIResult.ERROR_UNKNOWN, exception.getMessage(), exception.getStackTrace()[0].getClassName());
84.     }
86.     @ExceptionHandler(BindException.class)
87.     public ResponseEntity<APIResult<Object>> handleBindException(BindException exception) {
88.         String msg = Optional.of(exception.getBindingResult().getFieldErrors().stream().map(e -> e.getField() + ": " + e.getDefaultMessage())
89.                 .collect(Collectors.joining("\r\n"))).orElse("Validate failed");
90.         return logObject(APIResult.ERROR_PARAM, msg, exception.getStackTrace()[0].getClassName());
91.     }
92. }