servlet-exception-handler

ServletExceptionHandler.java

@Slf4j
@ControllerAdvice
@Profile("servlet")
@RequiredArgsConstructor
@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.SERVLET)
public class ServletExceptionHandler {
    private final Map<Class<? extends Exception>, HttpStatus> exceptionToStatusCode;
    private final HttpStatus defaultStatus;

    @ExceptionHandler(CustomExceptionInFilter.class)
    public ResponseEntity<ErrorResponse> handleCustomException(CustomExceptionInFilter ex) {
        return this.handleException(ex);
    }

    @ExceptionHandler(Exception.class)
    public ResponseEntity<ErrorResponse> handleException(Exception ex) {
        HttpStatus status = exceptionToStatusCode.getOrDefault(ex.getClass(), defaultStatus);
        ErrorResponse errorResponse = ErrorResponse
                .builder()
                .message(ex.getMessage())
                .code(status.value())
                .build();
        log.error("Exception has been occurred", ex);
        return new ResponseEntity<>(errorResponse, status);
    }
}