@ControllerAdvice注解原理

1. 啟動時掃描 @ControllerAdvice 注解

在 Spring Boot 應(yīng)用程序啟動過程中，Spring 會掃描所有的類，尋找?guī)в?@ControllerAdvice 注解的類。

這是通過 ConfigurationClassPostProcessor 和 ComponentScanAnnotationParser 來完成的。

這些類會被實例化并作為 Spring Bean 注入到 Spring 容器中。

2. ControllerAdviceBeanPostProcessor

ControllerAdviceBeanPostProcessor 是一個實現(xiàn)了 BeanPostProcessor 接口的類。

在應(yīng)用程序啟動過程中，所有 Bean 初始化之后，ControllerAdviceBeanPostProcessor 的 postProcessAfterInitialization 方法會被調(diào)用。

該方法負責(zé)處理帶有 @ControllerAdvice 注解的類；

postProcessAfterInitialization：

@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
    if (bean instanceof ControllerAdviceBean) {
        ControllerAdviceBean adviceBean = (ControllerAdviceBean) bean;
        Class<?> beanType = adviceBean.getBeanType();
        if (beanType != null) {
            ExceptionHandlerMethodResolver resolver = new ExceptionHandlerMethodResolver(beanType);
            if (resolver.hasExceptionMappings()) {
                this.exceptionHandlerAdviceCache.put(adviceBean, resolver);
            }
        }
    }
    return bean;
}

3. ExceptionHandlerMethodResolver

在上述 ControllerAdviceBeanPostProcessor 的方法中，會創(chuàng)建 ExceptionHandlerMethodResolver 實例來掃描帶有 @ControllerAdvice 注解的類中的所有方法，找到使用了 @ExceptionHandler 注解的異常處理方法。這些方法會被添加到一個映射表中，以便后續(xù)處理異常時能找到對應(yīng)的處理方法。

4. 注冊異常處理方法

在 ControllerAdviceBeanPostProcessor 中，異常處理方法會被注冊到一個全局的異常處理器映射表（exceptionHandlerAdviceCache）中。

映射表將異常類型與相應(yīng)的處理方法關(guān)聯(lián)起來，以便在發(fā)生異常時能夠找到正確的處理方法。

5. 請求處理和異常捕獲

當(dāng) Spring Boot 應(yīng)用程序接收到一個請求時，它會將請求分派給相應(yīng)的 Controller 方法。

在執(zhí)行 Controller 方法時，如果拋出了異常，DispatcherServlet 會將異常傳遞給 HandlerExceptionResolver 來處理。

在這里，ExceptionHandlerExceptionResolver 會被調(diào)用。

6. ExceptionHandlerExceptionResolver

ExceptionHandlerExceptionResolver 類負責(zé)在全局異常處理器映射表（exceptionHandlerAdviceCache）中查找與異常類型關(guān)聯(lián)的處理方法，并執(zhí)行這個方法來處理異常。

具體實現(xiàn)在 doResolveHandlerMethodException 方法中：

protected ModelAndView doResolveHandlerMethodException(HttpServletRequest request, HttpServletResponse response, HandlerMethod handlerMethod, Exception exception) {
    // ...
    ServletInvocableHandlerMethod exceptionHandlerMethod = getExceptionHandlerMethod(handlerMethod, exception);
    if (exceptionHandlerMethod == null) {
       return null;
   }
   // ...
   try {
       exceptionHandlerMethod.invokeAndHandle(webRequest, mavContainer, exceptionResult);
   } catch (Exception invocationEx) {
       // ...
   }
   // ...
   return ModelAndViewResolver.UNRESOLVED;
}

在這個方法中，`ExceptionHandlerExceptionResolver` 會根據(jù)異常類型查找對應(yīng)的處理方法。

如果找到了適合的處理方法，它將使用 `ServletInvocableHandlerMethod` 類的實例來執(zhí)行該方法，并將異常處理結(jié)果（通常是一個 ModelAndView 對象或 ResponseEntity 對象）返回給客戶端；

總結(jié)

`@ControllerAdvice` 的工作原理在 Spring Boot 應(yīng)用程序啟動時涉及到多個組件和步驟；

在啟動過程中，`ControllerAdviceBeanPostProcessor` 會掃描帶有 `@ControllerAdvice` 注解的類，`ExceptionHandlerMethodResolver` 會掃描這些類中的異常處理方法并將它們注冊到全局異常處理器映射表中。當(dāng)請求處理過程中發(fā)生異常時，`ExceptionHandlerExceptionResolver` 會從映射表中查找并執(zhí)行相應(yīng)的異常處理方法；

到此這篇關(guān)于SpringBoot中的@ControllerAdvice注解原理詳解的文章就介紹到這了,更多相關(guān)@ControllerAdvice注解原理內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論