인프런 커뮤니티 질문&답변

지하철에서까지 빠른 답변 감사합니다.

저도 말씀하신 것처럼 예상이 됩니다. 그렇다면 ControllerAdvice에서 RuntimeExeption을 잡고자한다면 필연적으로 AccessDeniedExption은 AccessDeniedHandler 구현체가 아닌 (HttpSecurity에 끼워넣는 것이 아닌) ControllerAdvice에서 따로 잡아야 할까요?

호돌맨님이라면 어떻게 하실지가 궁금합니다!

더불어 MethodSecurity가 아닌

auth.requestMatchers("/").permitAll()
    .requestMatchers("/user").hasRole("USER")
    .requestMatchers("/admin").hasRole("ADMIN")

형태로 직접 작성해주면 ControllerAdvice가 활성화 되어도

정상적으로 AccessDeniedHandler(Http403Handler)로직이 동작합니다.

다 찾아내시다니.. 정말 멋지시네요!

그러게요 이게 좀 애매합니다.

AccessDeniedException, AuthenticationException 등의 스프링 시큐리티 Exception들이 어떤 특정 Class를 상속받고 있다면 문제 해결이 쉬운데

모두 RuntimeException을 상속받고 있습니다.

그래서 현재는 아래와같이 분기문을 통해 status Code를 결정하는게 최선 같아보입니다.

아 이러면 코드가 안예쁜데..

    @ResponseBody
    @ExceptionHandler(Exception.class)
    public ResponseEntity<ErrorResponse> exception(Exception e) {
        log.error("예외발생", e);

        int errorCode = 500;
        if (e instanceof AccessDeniedException) {
            errorCode = 403;
        } else if (...) { }
        
        ErrorResponse body = ErrorResponse.builder()
                .code(String.valueOf(errorCode))
                .message(e.getMessage())
                .build();

        ResponseEntity<ErrorResponse> response = ResponseEntity.status(errorCode)
                .body(body);

        return response;
    }

아니면 아래와같이 하는 방법도 있을것 같습니다.

enum 형태의 SpringSecurityErrorCode.java 를 만들고요

@RequiredArgsConstructor
public enum SpringSecurityErrorCode {

    ACCESS_DENIED(AccessDeniedException.class, 403),
    UNAUTHORIZED(AuthenticationException.class, 401),
    // 기타등등 ...
    ETC(RuntimeException.class, 500);

    private final Class<? extends RuntimeException> exceptionClass;
    private final int statusCode;

    public static int getStatusCode(RuntimeException springException) {
        return Arrays.stream(values())
                .filter(e -> e.exceptionClass.isAssignableFrom(springException.getClass()))
                .findAny()
                .map(e -> e.statusCode)
                .orElse(500);
    }
}

아래처럼 예외에 맞는 status를 찾아 응답하는겁니다.

@ResponseBody
@ExceptionHandler(RuntimeException.class)
public ResponseEntity<ErrorResponse> exception(RuntimeException e) {
    log.error("예외발생", e);

    var statusCode = SpringSecurityErrorCode.getStatusCode(e);

    ErrorResponse body = ErrorResponse.builder()
            .code(String.valueOf(statusCode))
            .message(e.getMessage())
            .build();

    ResponseEntity<ErrorResponse> response = ResponseEntity.status(statusCode)
            .body(body);

    return response;
}

멋지십니다!

감사합니다.