io.github.mjyoun.spring.web.error.CustomErrorController Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-web Show documentation

A project that organizes frequently used utilities in Spring Web Service

There is a newer version: 0.3.1

package io.github.mjyoun.spring.web.error;

import java.util.Set;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;

import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.validation.BindException;
import org.springframework.validation.BindingResult;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import io.github.mjyoun.core.data.Result;
import lombok.extern.slf4j.Slf4j;

/**
 * 에러 핸들 controller
 * 
 * @author MJ Youn
 * @since 2022. 01. 04.
 */
@RestControllerAdvice
@Slf4j
public class CustomErrorController {

    /**
     * validate 오류시 동작하는 화면
     * 
     * @param e
     *            exception 정보
     * @return {@link ResponseEntity}
     * 
     * @author MJ Youn
     * @since 2022. 01. 04.
     */
    @ExceptionHandler({ BindException.class, ConstraintViolationException.class })
    public ResponseEntity> argumentCheck(Exception e) {
        Result result = null;

        if (e instanceof BindException) {
            BindingResult bindingResult = ((BindException) e).getBindingResult();
            String[] violateMsgs = !bindingResult.hasErrors() ? new String[0] //
                    : bindingResult.getAllErrors() //
                            .stream() //
                            .map(ObjectError::getDefaultMessage) //
                            .toArray(String[]::new);

            result = Result.error(String.join(",", violateMsgs));
        } else if (e instanceof ConstraintViolationException) {
            Set> constraintViolations = ((ConstraintViolationException) e).getConstraintViolations();

            String[] violateMsgs = constraintViolations //
                    .stream() //
                    .map(ConstraintViolation::getMessage) //
                    .toArray(String[]::new);

            result = Result.error(String.join(",", violateMsgs));
        } else {
            throw new RuntimeException("존재하지 않는 오류입니다.");
        }

        log.error("[CustomErrorController] 유효성 검사 오류 [msg: {}]", result.getMessage());
        return new ResponseEntity>(result, HttpStatus.BAD_REQUEST);
    }

    /**
     * 모든 exception 발생시 동작하는 함수
     * 
     * @param e
     *            exception 정보
     * @return {@link ResponseEntity}
     * 
     * @author MJ Youn
     * @since 2022. 01. 04.
     */
    @ExceptionHandler({ Exception.class })
    public ResponseEntity> error(Exception e) {
        Result result = Result.error(e.getMessage());

        log.error("[CustomErrorController] 서버 오류 [msg: {}]", result.getMessage());
        return new ResponseEntity<>(result, HttpStatus.INTERNAL_SERVER_ERROR);
    }

}