API 예외 처리

LeeKyoungChang·2022년 2월 4일
0
post-thumbnail

'스프링 MVC 2편 - 백엔드 웹 개발 활용 기술' 수업을 듣고 정리한 내용입니다.

 

📚 1. API 예외처리 - 시작

📣 HTML 오류 페이지 vs API 오류 메시지

  • HTML 페이지는 각 오류에 맞는 (400, 4xx 등) 오류 페이지를 만들어서 고객에게 오류 화면을 보여주면 된다.
  • API는 각 오류 상황에 맞는 오류 응답 스펙을 정하고, JSON으로 데이터를 내려줘야 한다.

 

지금부터 API의 경우 어떻게 예외 처리를 하면 좋은지 알아보자!
서블릿 오류 페이지로 방식을 사용해보자!

 

WebServerCustomizer 다시 동작

package hello.exception;

import org.springframework.boot.web.server.ConfigurableWebServerFactory;
import org.springframework.boot.web.server.ErrorPage;
import org.springframework.boot.web.server.WebServerFactoryCustomizer;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;

@Component
public class WebServerCustomizer implements WebServerFactoryCustomizer<ConfigurableWebServerFactory> {
    @Override
    public void customize(ConfigurableWebServerFactory factory) {

        ErrorPage errorPage404 = new ErrorPage(HttpStatus.NOT_FOUND, "/error-page/404");
        ErrorPage errorPage500 = new ErrorPage(HttpStatus.INTERNAL_SERVER_ERROR, "/error-page/500");

        ErrorPage errorPageEx = new ErrorPage(RuntimeException.class, "/error-page/500");

        factory.addErrorPages(errorPage404, errorPage500, errorPageEx);
    }
}
  • WAS에 예외가 전달되거나, response.sendError() 가 호출되면 위에 등록한 예외 페이지 경로가 호출된다.

 

ApiExceptionController - API 예외 컨트롤러

package hello.exception.api;


import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RestController;

@Slf4j
@RestController
public class ApiExceptionController {

    @GetMapping("/api/members/{id}")
    public MemberDto getMember(@PathVariable("id") String id) {

        if (id.equals("ex")) {
            throw new RuntimeException("잘못된 사용자");
        }

        return new MemberDto(id,"hello" +id);
    }

    @Data
    @AllArgsConstructor
    static class MemberDto{
        private String memberId;
        private String name;
    }
}
  • 단순히 회원을 조회 하는 기능이 추가되었다.
  • 예외 테스트를 위해 URL에 전달 된 id의 값이 ex이면 예외가 발생하도록 코드를 심어두었다.

 

실행 결과

HTTP Header에 Acceptapplication/json 인 것을 꼭 확인하자!

(1) 정상 호출

  • JSON 형식으로 데이터가 반환된다.
스크린샷 2022-02-03 오전 10 28 02

 

(2) 예외 발생 호출
스크린샷 2022-02-03 오전 10 29 07

JSON이 아닌, HTML 오류 페이지가 반환된다.
➡️ 우리가 원하는 것은 정상 요청이든, 오류 요청이든 JSON이 반환되는 것이다!

 

ErrorPageController - API 응답 추가

오류 페이지 컨트롤러도 JSON 응답을 할 수 있도록 수정해야 한다!

 

    @RequestMapping(value = "/error-page/500", produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Map<String, Object>> errorPage500Api(HttpServletRequest request, HttpServletResponse response) {
        log.info("API eerorPage 500");

        Map<String, Object> result = new HashMap<>();
        Exception ex = (Exception) request.getAttribute(ERROR_EXCEPTION);
        result.put("status", request.getAttribute(ERROR_STATUS_CODE));
        result.put("message", ex.getMessage());

        Integer statusCode = (Integer) request.getAttribute(RequestDispatcher.ERROR_STATUS_CODE);
        return new ResponseEntity<>(result, HttpStatus.valueOf(statusCode));
    }
  • produces = MediaType.APPLICATION_JSON_VALUE : 클라이언트가 요청하는 HTTP Header의 Accept 의 값이 application/json 일 때 해당 메서드가 호출된다. 결국 클라이언트가 받고 싶은 미디어타입이 json이면 이 컨트롤러의 메서드가 호출된다.

  • 응답 데이터를 위해서 Map 을 만들고 status , message 키에 값을 할당했다. Jackson 라이브러리는 Map 을 JSON 구조로 변환할 수 있다.

  • ResponseEntity 를 사용해서 응답하기 때문에 메시지 컨버터가 동작하면서 클라이언트에 JSON이 반환된다.

 

실행 결과

HTTP Header에 Acceptapplication/json 인 것을 꼭 확인하자!

스크린샷 2022-02-03 오전 10 37 16

이제 예외가 발생했을 경우도 (요청 HTTP Header의 Accept 값이 JSON이면), JSON 형식으로 반환되는 것을 알 수 있다!

 

📚 2. API 예외 처리 - 스프링 부트 기본 오류 처리

API 예외 처리도 스프링 부트가 제공하는 기본 오류 방식을 사용할 수 있다.

 

BasicErrorController 코드

  @RequestMapping(produces = MediaType.TEXT_HTML_VALUE)
  public ModelAndView errorHtml(HttpServletRequest request, HttpServletResponse response) {}
  @RequestMapping
  public ResponseEntity<Map<String, Object>> error(HttpServletRequest request) {}

/error 동일한 경로를 처리하는 errorHtml() , error() 두 메서드를 확인할 수 있다.

  • errorHtml() : produces = MediaType.TEXT_HTML_VALUE : 클라이언트 요청의 Accept 해더 값이 text/html 인 경우에는 errorHtml() 을 호출해서 view를 제공한다.
  • error() : 그외 경우에 호출되고 ResponseEntity 로 HTTP Body에 JSON 데이터를 반환한다.

 

스프링 부트의 예외 처리

  • 스프링 부트의 기본 설정은 오류 발생시 /error 를 오류 페이지로 요청한다.
  • BasicErrorController 는 이 경로를 기본으로 받는다. ( server.error.path 로 수정 가능, 기본 경로 /error )

 

실행 결과

BasicErrorController 를 사용하도록 WebServerCustomizer@Component 를 주석처리 하자!

 

스크린샷 2022-02-03 오전 11 13 22
  • 스프링 부트는 BasicErrorController 가 제공하는 기본 정보들을 활용해서 오류 API를 생성해준다.

 

다음 옵션들을 설정하면 더 자세한 오류 정보를 추가할 수 있다.

  • server.error.include-binding-errors=always
  • server.error.include-exception=true
  • server.error.include-message=always
  • server.error.include-stacktrace=always

➡️ 오류 메시지는 이렇게 막 추가하면 보안상 위험할 수 있다. 간결한 메시지만 노출하고, 로그를 통해서 확인하자.

 

📖 A. Html 페이지 vs API 오류

BasicErrorController 를 확장하면 JSON 메시지도 변경할 수 있다.

  • HTML 페이지 : 4xx, 5xx 등등 모두 잘 처리해준다.
  • API 오류 : API 마다, 각각의 컨트롤러나 예외마다 서로 다른 응답 결과를 출력해야 할 수도 있다. 결과적으로 BasicErrorController을 사용할 경우 매우 세밀하고 복잡하다.

➡️ 따라서 BasicErrorController는 HTML 화면을 처리할 때 사용하고, API 오류 처리는 @ExceptionHandler를 사용하자!

 

📚 3. API 예외 처리 - HandlerExceptionResolver 시작

복잡한 API 오류는 어떻게 처리해야하는지 지금부터 하나씩 알아보자!

 

🔔 목표

  • 예외가 발생해서 서블릿을 넘어 WAS까지 예외가 전달되면 HTTP 상태코드가 500으로 처리된다.
  • 발생하는 예외에 따라서 400, 404 등등 다른 상태코드도 처리하고 싶다.
  • 오류 메시지, 형식등을 API마다 다르게 처리하고 싶다.

 

📖 A. 상태코드 변환

  • IllegalArgumentException 을 처리하지 못해서 컨트롤러 밖으로 넘어가는 일이 발생하면 HTTP 상태코드를 400으로 처리

 

ApiExceptionController - 수정

    @GetMapping("/api/members/{id}")
    public MemberDto getMember(@PathVariable("id") String id) {

        if (id.equals("ex")) {
            throw new RuntimeException("잘못된 사용자");
        }
        if(id.equals("bad")){
            throw new IllegalArgumentException("잘못된 입력 값");
        }

        return new MemberDto(id,"hello" +id);
    }
  • http://localhost:8080/api/members/bad 라고 호출하면 IllegalArgumentException 이 발생하도록 했다.

 

실행
스크린샷 2022-02-03 오후 12 15 51

  • 실행해보면 상태 코드가 500인 것을 확인할 수 있다.

 

✓ HandlerExceptionResolver

  • 스프링 MVC는 컨트롤러(핸들러) 밖으로 예외가 던져진 경우 예외를 해결하고, 동작을 새로 정의할 수 있는 방법을 제공한다.
  • 컨트롤러 밖으로 던져진 예외를 해결하고, 동작 방식을 변경하고 싶으면 HandlerExceptionResolver 를 사용하면 된다.
  • 줄여서 ExceptionResolver 라 한다.

 

✓ ExceptionResolver 적용 전
스크린샷 2022-02-03 오후 12 49 29

  • 예외가 발생하면 postHandler를 호출하지 않는다!

 

✓ ExceptionResolver 적용 후
스크린샷 2022-02-03 오후 12 49 50

  • 중간에 HandlerExceptionResolver가 추가되어 발생한 예외를 처리하고, WAS로 전달할 방식을 제어하는 과정이 추가되었다.
  • ExceptionResolver 로 예외를 해결해도 postHandle()은 호출되지 않는다.

 

HandlerExceptionResolver - 인터페이스

public interface HandlerExceptionResolver {

    ModelAndView resolveException(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex);
}
  • handler : 핸들러(컨트롤러) 정보
  • Exception ex : 핸들러(컨트롤러)에서 발생한 발생한 예외

 

MyHandlerExceptionResolver

package hello.exception.resolver;

import lombok.extern.slf4j.Slf4j;
import org.springframework.ui.Model;
import org.springframework.web.servlet.HandlerExceptionResolver;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

@Slf4j
public class MyHandlerExceptionResolver implements HandlerExceptionResolver {
    @Override
    public ModelAndView resolveException(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {

        log.info("call resolver", ex);

        try {
            if (ex instanceof IllegalArgumentException) {
                log.info("IllegalArgumentException resolver to 400");

                response.sendError(HttpServletResponse.SC_BAD_REQUEST, ex.getMessage());
                return new ModelAndView();
            }
        } catch (IOException e) {
            log.error("resolver ex", e);
        }

        return null;
    }
}
  • ExceptionResolverModelAndView 를 반환하는 이유는 마치 try, catch를 하듯이, Exception 을 처리해서 정상 흐름 처럼 변경하는 것이 목적이다.
    • 이름 그대로 Exception 을 Resolver(해결)하는 것이 목적이다.
  • IllegalArgumentException 이 발생하면 response.sendError(400) 를 호출해서 HTTP 상태 코드를 400으로 지정하고, 빈 ModelAndView 를 반환한다.

 

✔ HandlerExceptionResolver 반환 값에 따른 DispatcherServlet의 동작 방식

  • ModelAndView : new ModelAndView() 처럼 빈 ModelAndView 를 반환하면 뷰를 렌더링 하지 않고, 정상 흐름으로 서블릿이 리턴된다.
  • ModelAndView 지정 : ModelAndViewView , Model 등의 정보를 지정해서 반환하면 뷰를 렌더링 한다.
  • null : null 을 반환하면, 다음 ExceptionResolver 를 찾아서 실행한다. 만약 처리할 수 있는 ExceptionResolver 가 없으면 예외 처리가 안되고, 기존에 발생한 예외를 서블릿 밖으로 던진다.

 

✔ ExceptionResolver 활용

  • 예외 상태 코드 변환
    • 예외를 response.sendError(xxx) 호출로 변경해서 서블릿에서 상태 코드에 따른 오류를 처리하도록 위임
    • 이후 WAS는 서블릿 오류 페이지를 찾아서 내부 호출, 예를 들어서 스프링 부트가 기본으로 설정한 /error 가 호출됨
  • 뷰 템플릿 처리
    • ModelAndView 에 값을 채워서 예외에 따른 새로운 오류 화면 뷰 렌더링 해서 고객에게 제공
  • API 응답 처리
    • response.getWriter().println("hello"); 처럼 HTTP 응답 바디에 직접 데이터를 넣어주는 것도 가능하다. 여기에 JSON 으로 응답하면 API 응답 처리를 할 수 있다.

 

WebConfig - 수정

WebMvcConfigurer 를 통해 등록

 

    /**
     * 기본 설정을 유지하면서 추가
     */
    @Override
    public void extendHandlerExceptionResolvers(List<HandlerExceptionResolver> resolvers) {
        resolvers.add(new MyHandlerExceptionResolver());
    }
  • configureHandlerExceptionResolvers(..) 를 사용하면 스프링이 기본으로 등록하는 ExceptionResolver 가 제거되므로 주의하며, extendHandlerExceptionResolvers 를 사용하자!

 

실행 결과

  • /api/members/ex
스크린샷 2022-02-03 오후 12 27 59

 

  • /api/members/bad
스크린샷 2022-02-03 오후 12 28 07

 

📚 4. API 예외 처리 - HandlerExceptionResolver 활용

🔔 예외를 여기서 마무리하기

  • 예외가 발생하면 WAS까지 예외가 던져지고, WAS에서 오류 페이지 정보를 찾아서 다시 /error 를 호출하는 과정은 생각해보면 너무 복잡하다.
  • ExceptionResolver 를 활용하면 예외가 발생했을 때 이런 복잡한 과정 없이 여기에서 문제를 깔끔하게 해결할 수 있다.

 

UserException

  • 사용자 정의 예외를 하나 추가

 

package hello.exception.exception;

public class UserException extends RuntimeException{

    public UserException() {
        super();
    }

    public UserException(String message) {
        super(message);
    }

    public UserException(String message, Throwable cause) {
        super(message, cause);
    }

    public UserException(Throwable cause) {
        super(cause);
    }

    protected UserException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
        super(message, cause, enableSuppression, writableStackTrace);
    }
}

 

ApiExceptionController - 예외 추가

package hello.exception.api;


import hello.exception.exception.UserException;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RestController;

@Slf4j
@RestController
public class ApiExceptionController {

    @GetMapping("/api/members/{id}")
    public MemberDto getMember(@PathVariable("id") String id) {

        if (id.equals("ex")) {
            throw new RuntimeException("잘못된 사용자");
        }
        if(id.equals("bad")){
            throw new IllegalArgumentException("잘못된 입력 값");
        }
        if (id.equals("user-ex")) {
            throw new UserException("사용자 오류");
        }

        return new MemberDto(id,"hello" +id);
    }

    @Data
    @AllArgsConstructor
    static class MemberDto{
        private String memberId;
        private String name;
    }
}

 

실행 결과

  • http://localhost:8080/api/members/user-ex
  • UserException 이 발생

 

UserHandlerExceptionResolver

  • 위 예외를 처리하는 UserHandlerExceptionResolver

 

package hello.exception.resolver;

import com.fasterxml.jackson.databind.ObjectMapper;
import hello.exception.exception.UserException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.servlet.HandlerExceptionResolver;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

@Slf4j
public class UserHandlerExceptionResolver implements HandlerExceptionResolver {

    private final ObjectMapper objectMapper = new ObjectMapper();

    @Override
    public ModelAndView resolveException(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
        try {
            if (ex instanceof UserException) {
                log.info("UserException resolver to 400");
                String acceptHeader = request.getHeader("accept");
                response.setStatus(HttpServletResponse.SC_BAD_REQUEST);

                if ("application/json".equals(acceptHeader)) {
                    Map<String, Object> errorResult = new HashMap<>();
                    errorResult.put("ex", ex.getClass());
                    errorResult.put("message", ex.getMessage());

                    String result = objectMapper.writeValueAsString(errorResult);

                    response.setContentType("application/json");
                    response.setCharacterEncoding("utf-8");
                    response.getWriter().write(result);

                    return new ModelAndView();

                }else{
                    // TEXT/HTML
                    return new ModelAndView("error/500");
                }
            }
        } catch (IOException e) {
            log.error("resolver ex",e);
        }

        return null;
    }
}
  • HTTP 요청 해더의 ACCEPT 값이 application/json 이면 JSON으로 오류를 내려주고, 그 외 경우에는 error/500에 있는 HTML 오류 페이지를 보여준다.

 

WebConfig에 UserHandlerExceptionResolver 추가

    /**
     * 기본 설정을 유지하면서 추가
     */
    @Override
    public void extendHandlerExceptionResolvers(List<HandlerExceptionResolver> resolvers) {
        resolvers.add(new MyHandlerExceptionResolver());
        resolvers.add(new UserHandlerExceptionResolver());
    }

 

실행 결과
스크린샷 2022-02-03 오후 1 31 18

스크린샷 2022-02-03 오후 1 32 47

 

📌 정리

  • ExceptionResolver 를 사용하면 컨트롤러에서 예외가 발생해도 ExceptionResolver 에서 예외를 처리해버린다.
  • 이렇게 예외를 이곳에서 모두 처리할 수 있다는 것이 핵심이다.
  • 서블릿 컨테이너까지 예외가 올라가면 복잡하고 지저분하게 추가 프로세스가 실행된다. 반면에 ExceptionResolver 를 사용하면 예외처리가 상당히 깔끔해진다.

 

직접 ExceptionResolver 를 구현하려고 하니 상당히 복잡하다. 지금부터 스프링이 제공하는 ExceptionResolver 들을 알아보자!

 

📚 5. API 예외 처리 - 스프링이 제공하는 ExceptionResolver1

  • ExceptionResolver를 사용하면 모든 예외를 이 곳에서 처리할 수 있어서 좋다.
  • 다만, ExceptionResolver를 직접 구현하는 과정은 복잡하다.
  • 스프링이 제공하는 ExceptionResolver에 대해 알아보자!

 

📖 A. 스프링 부트가 기본으로 제공하는 ExceptionResovler

🔔 HandlerExceptionResolverComposite
1. ExceptionHandlerExceptionResolver
➡️ @ExceptionHandler 을 처리한다.
➡️ API 예외 처리는 대부분 이 기능으로 해결한다.

  1. ResponseStatusExceptionResolver
    ➡️ HTTP 상태 코드를 지정해준다.
    ➡️ 예) @ResponseStatus(value = HttpStatus.NOT_FOUND)
  2. DefaultHandlerExceptionResolver → 우선 순위가 가장 낮다.
    ➡️ 스프링 내부 기본 예외를 처리한다.

 

📖 B. ResponseStatusExceptionResolver

ResponseStatusExceptionResolver예외에 따라서 HTTP 상태 코드를 지정해주는 역할을 한다.

 

📣 두 가지 경우를 처리한다.

  • @ResponseStatus 가 달려있는 예외
  • ResponseStatusException 예외

 

✔️ @ResponseStatus 가 달려있는 예외

@ResponseStatus 애노테이션을 적용하면 HTTP 상태 코드를 변경해준다.

@ResponseStatus(code = HttpStatus.BAD_REQUEST, reason = "잘못된 오류 요청")
public class BadRequestException extends RuntimeException{
}
  • BadRequestException 예외가 컨트롤러 밖으로 넘어가면 ResponseStatusExceptionResolver 예외가 해당 애노테이션을 확인해서 오류 코드를 HttpStatus.BAD_REQUEST (400)으로 변경하고, 메시지도 담는다.
  • ResponseStatusExceptionResolver 코드를 확인해보면 결국 response.sendError(statusCode, resolvedReason) 를 호출한다.

 

ApiExceptionController - 추가

  @GetMapping("/api/response-status-ex1")
  public String responseStatusEx1() {
      throw new BadRequestException();
  }

실행 결과
스크린샷 2022-02-03 오후 4 07 18

  • sendError(400) 를 호출했기 때문에 WAS에서 다시 오류 페이지( /error )를 내부 요청한다.

 

메시지 기능
reasonMessageSource 에서 찾는 기능도 제공한다. reason = "error.bad"

@ResponseStatus(code = HttpStatus.NOT_FOUND, reason = "error.bad")
public class BadRequestException extends RuntimeException{
}

 

messages.properties

error.bad=잘못된 요청 오류입니다. 메시지 사용
  • messages.properties에서 error.bad를 찾는다.

 

실행 결과
스크린샷 2022-02-03 오후 4 07 37

 

✔️ ResponseStatusException

  • @ResponseStatus 는 개발자가 직접 변경할 수 없는 예외에는 적용할 수 없다. → 애노테이션을 직접 넣어야 하는데, 내가 코드를 수정할 수 없는 라이브러리의 예외 코드 같은 곳에는 적용할 수 없다.
  • 애노테이션을 사용하기 때문에 조건에 따라 동적으로 변경하는 것도 어렵다. → 이때는 ResponseStatusException 예외를 사용하면 된다.

 

ApiExceptionController - 추가

@GetMapping("/api/response-status-ex2")
public String responseStatusEx2() {
    throw new ResponseStatusException(HttpStatus.NOT_FOUND, "error.bad", new IllegalArgumentException());
}

 

실행 결과

스크린샷 2022-02-03 오후 4 07 52

 

📚 6. API 예외 처리 - 스프링이 제공하는 ExceptionResolver2

🔔 DefaultHandlerExceptionResolver

  • 스프링 내부에서 발생하는 스프링 예외를 해결한다.

대표적으로 파라미터 바인딩 시점에 타입이 맞지 않으면 내부에서 TypeMismatchException 이 발생한다. 이 경우를 살펴보자

  • 이 경우, 예외가 발생했기 때문에 그냥 두면 서블릿 컨테이너까지 오류가 올라가고, 결과적으로 500 오류가 발생한다.
  • 그런데 파라미터 바인딩은 대부분 클라이언트가 HTTP 요청 정보를 잘못 호출해서 발생하는 문제이다.
    ➡️ HTTP 에서는 이런 경우 HTTP 상태 코드 400을 사용하도록 되어 있다.
  • DefaultHandlerExceptionResolver는 이것을 500 오류가 아니라 HTTP 상태 코드 400 오류로 변경한다.

이처럼 DefaultHandlerExceptionResolver은 스프링 내부 오류를 어떻게 처리할지 수많은 내용이 정의되어 있으며, 이에 맞는 동작을 해준다.

 

DefaultHandlerExceptionResolver.handleTypeMismatch 내부를 보면

스크린샷 2022-02-03 오후 5 00 49

이와 같이 구현되어 있다.

response.sendError(HttpServletResponse.SC_BAD_REQUEST) (400)

➡️ 결국 response.sendError() 를 통해서 문제를 해결한다.
➡️ sendError(400) 를 호출했기 때문에 WAS에서 다시 오류 페이지( /error )를 내부 요청한다.

 

ApiExceptionController - 추가

@GetMapping("/api/default-handler-ex")
public String defaultException(@RequestParam Integer data) {
    return "ok";
}
  • Integer data 에 문자를 입력하면 내부에서 TypeMismatchException 이 발생한다.

 

실행 결과
스크린샷 2022-02-03 오후 5 06 59

스크린샷 2022-02-03 오후 4 43 17 스크린샷 2022-02-03 오후 4 44 42

 

📌 정리
HandlerExceptionResolver 는 ?

  • 직접 사용하기는 복잡하다.
  • API 오류 응답의 경우 response 에 직접 데이터를 넣어야 해서 매우 불편하고 번거롭다.
  • ModelAndView 를 반환해야 하는 것도 API에는 잘 맞지 않는다.

 

스프링은 이 문제를 해결하기 위해 @ExceptionHandler 라는 매우 혁신적인 예외 처리 기능을 제공한다.
그것은 ExceptionHandlerExceptionResolver 이다.

 

📚 7. API 예외 처리 - @ExceptionHandler

HTML 화면 오류 vs API 오류
HTML 페이지

  • 오류가 발생하면 BasicErrorController를 사용하는 것이 좋다.
  • 4xx, 5xx 관련된 오류 페이지를 만들어서 고객에게 오류 화면을 보여주면 된다. → BasicErrorController가 이런 매커니즘을 모두 구현해놓았다.

API 오류

  • 매우 세밀한 제어가 필요하다.
    • 각 시스템 마다 응답의 모양도 다르고, 스펙도 모두 다르다.
    • 예외에 따라서 각각 다른 데이터를 출력해야 할 수도 있다.
    • 같은 예외라고 해도 어떤 컨트롤러에서 발생했는가에 따라서 다른 예외 응답을 내려주어야 할 수 있다.
    • 예를 들어서 상품 API와 주문 API는 오류가 발생했을 때 응답의 모양이 완전히 다를 수 있다.
  • 결국 지금까지 살펴본 BasicErrorController 를 사용하거나 HandlerExceptionResolver 를 직접 구현하는 방식으로 API 예외를 다루기는 쉽지 않다.

 

✔️ API 예외처리의 어려운 점

  • HandlerExceptionResolver 를 떠올려 보면 ModelAndView 를 반환해야 했다. 이것은 API 응답에는 필요하지 않다.
  • API 응답을 위해서 HttpServletResponse 에 직접 응답 데이터를 넣어주었다. 이것은 매우 불편하다.
    • 스프링 컨트롤러에 비유하면 마치 과거 서블릿을 사용하던 시절로 돌아간 것 같다.
  • 특정 컨트롤러에서만 발생하는 예외를 별도로 처리하기 어렵다.
    • 예를 들어서 회원을 처리하는 컨트롤러에서 발생하는 RuntimeException 예외와 상품을 관리하는 컨트롤러에서 발생하는 동일한 RuntimeException 예외를 서로 다른 방식으로 처리하고 싶다면 어떻게 해야할까?

 

📖 A. @ExceptionHandler

스프링에서 ExceptionHandlerExceptionResolver 사용시 API 예외 처리 문제를 해결하기 위해 @ExceptionHandler 라는 애노테이션을 사용하는 매우 편리한 예외 처리 기능을 제공한다.

 

🔔 @ExceptionHandler
매우 편리한 예외 처리 기능을 제공한다.

  • 스프링은 ExceptionHandlerExceptionResolver 를 기본으로 제공하고, 기본으로 제공하는 ExceptionResolver 중에 우선순위도 가장 높다.
  • 실무에서 API 예외 처리는 대부분 이 기능을 사용한다.

 

ErrorResult

package hello.exception.exhandler;

import lombok.AllArgsConstructor;
import lombok.Data;

@Data
@AllArgsConstructor
public class ErrorResult {

    private String code;
    private String message;
}
  • 예외가 발생했을 때 API 응답으로 사용하는 객체를 정의했다.

 

ApiExceptionV2Controller

package hello.exception.api;

import hello.exception.exception.UserException;
import hello.exception.exhandler.ErrorResult;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

@Slf4j
@RestController
public class ApiExceptionV2Controller {

    @ResponseStatus(HttpStatus.BAD_REQUEST)
    @ExceptionHandler(IllegalArgumentException.class)
    public ErrorResult illegalExHandler(IllegalArgumentException e) {
        log.error("[exceptionHandler] ex", e);
        return new ErrorResult("BAD", e.getMessage());
    }

    @ExceptionHandler
    public ResponseEntity<ErrorResult> userHandler(UserException e) {
        log.error("[exceptionHandler] ex", e);
        ErrorResult errorResult = new ErrorResult("USER-EX", e.getMessage());
        return new ResponseEntity(errorResult, HttpStatus.BAD_REQUEST);
    }

    @ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
    @ExceptionHandler
    public ErrorResult exHandler(Exception e) {
        log.error("[exceptionHandler] ex", e);
        return new ErrorResult("EX", "내부 오류");
    }

    @GetMapping("/api2/members/{id}")
    public MemberDto getMember(@PathVariable("id") String id) {

        if (id.equals("ex")) {
            throw new RuntimeException("잘못된 사용자");
        }
        if(id.equals("bad")){
            throw new IllegalArgumentException("잘못된 입력 값");
        }
        if (id.equals("user-ex")) {
            throw new UserException("사용자 오류");
        }

        return new MemberDto(id,"hello" +id);
    }


    @Data
    @AllArgsConstructor
    static class MemberDto{
        private String memberId;
        private String name;
    }

}

 

@ExceptionHandler 예외 처리 방법

  • @ExceptionHandler 애노테이션을 선언하고, 해당 컨트롤러에서 처리하고 싶은 예외를 지정해주면 된다.
  • 해당 컨트롤러에서 예외가 발생하면 이 메서드가 호출된다.
  • 참고로 지정한 예외 또는 그 예외의 자식 클래스는 모두 잡을 수 있다.

 

지정한 예외 또는 그 예외의 자식 클래스는 모두 잡기 위한 예시

  • 예시에서는 IllegalArgumentException 또는 그 하위 자식 클래스를 모두 처리할 수 있다.
@ExceptionHandler(IllegalArgumentException.class)
public ErrorResult illegalExHandle(IllegalArgumentException e) {
    log.error("[exceptionHandle] ex", e);
    return new ErrorResult("BAD", e.getMessage());
}

 

✔️ 우선순위

  • 스프링의 우선순위는 항상 자세한 것이 우선권을 가진다.
  • 예를 들어서 부모, 자식 클래스가 있고 다음과 같이 예외가 처리된다.
@ExceptionHandler(부모예외.class)
public String 부모예외처리()(부모예외 e) {}

@ExceptionHandler(자식예외.class)
public String 자식예외처리()(자식예외 e) {}
  • @ExceptionHandler 에 지정한 부모 클래스는 자식 클래스까지 처리할 수 있다.
  • 자식예외 발생 → 부모예외처리(), 자식예외처리() 모두 호출 대상 → (자세한 것이 더 우선권을 가진다.) 자식예외처리() 호출
  • 부모예외 호출 → 부모예외처리() 만 호출 대상 → 부모예외처리() 호출

 

✔️ 다양한 예외

  • 다음과 같이 다양한 예외를 한번에 처리할 수 있다.
@ExceptionHandler({AException.class, BException.class})
public String ex(Exception e) {
    log.info("exception e", e);
}

 

✔️ 예외 생략

  • @ExceptionHandler 에 예외를 생략할 수 있다. 생략하면 메서드 파라미터의 예외가 지정된다.
@ExceptionHandler
public ResponseEntity<ErrorResult> userExHandle(UserException e) {}

 

✔️ 파라미터와 응답

 

실행
스크린샷 2022-02-03 오후 5 27 15

 

IllegalArgumentException 처리

@ResponseStatus(HttpStatus.BAD_REQUEST)
@ExceptionHandler(IllegalArgumentException.class)
public ErrorResult illegalExHandle(IllegalArgumentException e) {
    log.error("[exceptionHandle] ex", e);
    return new ErrorResult("BAD", e.getMessage());
}

 

실행 흐름
1. 컨트롤러를 호출한 결과 IllegalArgumentException 예외가 컨트롤러 밖으로 던져진다.
2. 예외가 발생했으로 ExceptionResolver 가 작동한다.

  • 가장 우선순위가 높은 ExceptionHandlerExceptionResolver 가 실행된다.
  1. ExceptionHandlerExceptionResolver 는 해당 컨트롤러에 IllegalArgumentException 을 처리할 수 있는 @ExceptionHandler 가 있는지 확인한다.
  2. illegalExHandle() 를 실행한다.
    • @RestController 이므로 illegalExHandle() 에도 @ResponseBody 가 적용된다.
    • 따라서 HTTP 컨버터가 사용되고, 응답이 JSON으로 반환된다.
  3. @ResponseStatus(HttpStatus.BAD_REQUEST) 를 지정했으므로 HTTP 상태 코드 400으로 응답한다.

 

실행 결과
스크린샷 2022-02-03 오후 5 31 16

 

UserException 처리

  @ExceptionHandler
  public ResponseEntity<ErrorResult> userExHandle(UserException e) {
      log.error("[exceptionHandle] ex", e);
      ErrorResult errorResult = new ErrorResult("USER-EX", e.getMessage());
      return new ResponseEntity<>(errorResult, HttpStatus.BAD_REQUEST);
  }
  1. @ExceptionHandler 에 예외를 지정하지 않으면 해당 메서드 파라미터 예외를 사용한다. 여기서는 UserException 을 사용한다.
  2. ResponseEntity 를 사용해서 HTTP 메시지 바디에 직접 응답한다. 물론 HTTP 컨버터가 사용된다.
    • ResponseEntity 를 사용하면 HTTP 응답 코드를 프로그래밍해서 동적으로 변경할 수 있다.
    • 앞서 살펴본 @ResponseStatus 는 애노테이션이므로 HTTP 응답 코드를 동적으로 변경할 수 없다.

 

실행 결과
스크린샷 2022-02-03 오후 5 35 17

 

Exception

  @ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
  @ExceptionHandler
  public ErrorResult exHandle(Exception e) {
	  log.error("[exceptionHandle] ex", e);
	  return new ErrorResult("EX", "내부 오류");
  }
  1. thrownewRuntimeException("잘못된 사용자")이코드가실행되면서,컨트롤러밖으로 RuntimeException 이 던져진다.
  2. RuntimeExceptionException 의 자식 클래스이다. 따라서 이 메서드가 호출된다.
  3. @ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR) 로 HTTP 상태 코드를 500으로 응답한다.

 

실행 결과

스크린샷 2022-02-03 오후 5 38 59

 

기타 - HTML 오류 화면

  • 다음과 같이 ModelAndView 를 사용해서 오류 화면(HTML)을 응답하는데 사용할 수도 있다.
  • 거의 사용하지 않는다.
  @ExceptionHandler(ViewException.class)
  public ModelAndView ex(ViewException e) {
      log.info("exception e", e);
      return new ModelAndView("error");
  }

 

📚 8. API 예외 처리 - @ControllerAdvice

스프링이 제공하는 편리한 예외 처리 기능인 @ExceptionHandler을 이용해서 예외를 깔끔하게 처리할 수 있었지만, 정상코드와 예외 처리 코드가 하나의 컨트롤러에 섞여 있다.
➡️ @ControllerAdvice 또는 @RestControllerAdvice 를 사용하면 둘을 분리할 수 있다.

 

ExControllerAdvice

package hello.exception.exhandler.advice;

import hello.exception.exception.UserException;
import hello.exception.exhandler.ErrorResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseStatus;
import org.springframework.web.bind.annotation.RestControllerAdvice;

@Slf4j
@RestControllerAdvice(basePackages =  "hello.exception.api")
public class ExControllerAdvice {


    @ResponseStatus(HttpStatus.BAD_REQUEST)
    @ExceptionHandler(IllegalArgumentException.class)
    public ErrorResult illegalExHandler(IllegalArgumentException e) {
        log.error("[exceptionHandler] ex", e);
        return new ErrorResult("BAD", e.getMessage());
    }

    @ExceptionHandler
    public ResponseEntity<ErrorResult> userHandler(UserException e) {
        log.error("[exceptionHandler] ex", e);
        ErrorResult errorResult = new ErrorResult("USER-EX", e.getMessage());
        return new ResponseEntity(errorResult, HttpStatus.BAD_REQUEST);
    }

    @ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
    @ExceptionHandler
    public ErrorResult exHandler(Exception e) {
        log.error("[exceptionHandler] ex", e);
        return new ErrorResult("EX", "내부 오류");
    }
}

 

ApiExceptionV2Controller 코드에 있는 @ExceptionHandler 모두 제거

package hello.exception.api;

import hello.exception.exception.UserException;
import hello.exception.exhandler.ErrorResult;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

@Slf4j
@RestController
public class ApiExceptionV2Controller {


    @GetMapping("/api2/members/{id}")
    public MemberDto getMember(@PathVariable("id") String id) {

        if (id.equals("ex")) {
            throw new RuntimeException("잘못된 사용자");
        }
        if(id.equals("bad")){
            throw new IllegalArgumentException("잘못된 입력 값");
        }
        if (id.equals("user-ex")) {
            throw new UserException("사용자 오류");
        }

        return new MemberDto(id,"hello" +id);
    }


    @Data
    @AllArgsConstructor
    static class MemberDto{
        private String memberId;
        private String name;
    }

}

 

@ControllerAdvice

  • @ControllerAdvice 는 대상으로 지정한 여러 컨트롤러에 @ExceptionHandler , @InitBinder 기능을 부여해주는 역할을 한다.
  • @ControllerAdvice 에 대상을 지정하지 않으면 모든 컨트롤러에 적용된다. (글로벌 적용)
  • @RestControllerAdvice@ControllerAdvice 와 같고, @ResponseBody 가 추가되어 있다. (@Controller , @RestController 의 차이와 같다.)

 

대상 컨트롤러 지정 방법

// Target all Controllers annotated with @RestController
@ControllerAdvice(annotations = RestController.class)
public class ExampleAdvice1 {}

// Target all Controllers within specific packages
@ControllerAdvice("org.example.controllers")
public class ExampleAdvice2 {}

// Target all Controllers assignable to specific classes
@ControllerAdvice(assignableTypes = {ControllerInterface.class, AbstractController.class})
public class ExampleAdvice3 {}

 

  • 이처럼 특정 애노테이션이 있는 컨트롤러를 지정할 수 있고, 특정 패키지를 직접 지정할 수도 있다.
  • 패키지 지정의 경우 해당 패키지와 그 하위에 있는 컨트롤러가 대상이 된다. 그리고 특정 클래스를 지정할 수도 있다.
  • 대상 컨트롤러 지정을 생략하면 모든 컨트롤러에 적용된다.

 

실행 결과
스크린샷 2022-02-03 오후 6 54 35
스크린샷 2022-02-03 오후 6 54 44

 

📌 정리

  • @ExceptionHandler@ControllerAdvice 를 조합하면 예외를 깔끔하게 해결할 수 있다.

 


참고

profile
"야, (오류 만났어?) 너두 (해결) 할 수 있어"

0개의 댓글