com.github.saphyra.randwo.common.CustomExceptionHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rand-wo Show documentation

There is a newer version: 1.1.0

package com.github.saphyra.randwo.common;

import static java.util.Objects.isNull;

import java.io.IOException;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.web.HttpRequestMethodNotSupportedException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;

import com.github.saphyra.exceptionhandling.domain.ErrorResponse;

@ControllerAdvice
public class CustomExceptionHandler {

    @ExceptionHandler(MethodArgumentTypeMismatchException.class)
    ResponseEntity handleTypeMismatch(HttpServletRequest request, HttpServletResponse response) throws IOException {
        Object requestType = request.getAttribute("Request-Type");
        if (!isNull(requestType) && requestType.equals("rest")) {
            return ResponseEntity.badRequest().body(errorResponse(HttpStatus.BAD_REQUEST, ErrorCode.ARGUMENT_MISMATCH));
        }

        response.sendRedirect("/");
        return null;
    }

    @ExceptionHandler(HttpMessageNotReadableException.class)
    ResponseEntity handleEmptyRequestBody() {
        return ResponseEntity.badRequest().body(errorResponse(HttpStatus.BAD_REQUEST, ErrorCode.REQUEST_BODY_MISSING));
    }

    @ExceptionHandler(HttpRequestMethodNotSupportedException.class)
    ResponseEntity handleMethodNotSupported() {
        return new ResponseEntity(errorResponse(HttpStatus.METHOD_NOT_ALLOWED, ErrorCode.GENERAL_ERROR), HttpStatus.METHOD_NOT_ALLOWED);
    }

    private ErrorResponse errorResponse(HttpStatus httpStatus, ErrorCode errorCode) {
        return ErrorResponse.builder()
            .httpStatus(httpStatus.value())
            .errorCode(errorCode.getErrorCode())
            .build();
    }
}