Spring WebFlux请求处理流程

public class HttpHandlerAutoConfiguration {
    @Configuration(proxyBeanMethods = false)
    public static class AnnotationConfig {
        private final ApplicationContext applicationContext;
        public AnnotationConfig(ApplicationContext applicationContext) {
            this.applicationContext = applicationContext;
        }
        @Bean
        public HttpHandler httpHandler(ObjectProvider<WebFluxProperties> propsProvider) {
            HttpHandler httpHandler = WebHttpHandlerBuilder.applicationContext(this.applicationContext).build();
            WebFluxProperties properties = propsProvider.getIfAvailable();
            if (properties != null && StringUtils.hasText(properties.getBasePath())) {
                Map<String, HttpHandler> handlersMap = Collections.singletonMap(properties.getBasePath(), httpHandler);
                return new ContextPathCompositeHandler(handlersMap);
            }
            return httpHandler;
        }
    }
}

public class HttpWebHandlerAdapter extends WebHandlerDecorator implements HttpHandler {
    // 请求入口
    public Mono<Void> handle(ServerHttpRequest request, ServerHttpResponse response) {
        if (this.forwardedHeaderTransformer != null) {
            try {
                request = this.forwardedHeaderTransformer.apply(request);
            } catch (Throwable ex) {
                response.setStatusCode(HttpStatus.BAD_REQUEST);
                return response.setComplete();
            }
        }
        ServerWebExchange exchange = createExchange(request, response);
        // 获取委托类，最终是由委托的WebHandler进行处理
        // 1.1
        return getDelegate().handle(exchange)
                .doOnSuccess(aVoid -> logResponse(exchange))
                .onErrorResume(ex -> handleUnresolvedError(exchange, ex))
                .then(Mono.defer(response::setComplete));
    }
}

public abstract class AbstractApplicationContext {
    public void refresh() throws BeansException, IllegalStateException {
        // ...
        // 执行子类方法
        onRefresh();
        // ...
    }
}
public class ReactiveWebServerApplicationContext {
    protected void onRefresh() {
        // ...
        createWebServer();
        // ...
    }
    private void createWebServer() {
        WebServerManager serverManager = this.serverManager;
        if (serverManager == null) {
            // ...
            // this::getHttpHandler方法获取上面的HttpHandler对象
            this.serverManager = new WebServerManager(this, webServerFactory, this::getHttpHandler, lazyInit);
            // ...
        }
        initPropertySources();
    }
    protected HttpHandler getHttpHandler() {
        String[] beanNames = getBeanFactory().getBeanNamesForType(HttpHandler.class);
        // ...
        return getBeanFactory().getBean(beanNames[0], HttpHandler.class);
    }
}

public class WebHandlerDecorator implements WebHandler {
    private final WebHandler delegate;
    public WebHandlerDecorator(WebHandler delegate) {\
        this.delegate = delegate;
    }
    public WebHandler getDelegate() {
        return this.delegate;
    }
}

public class HttpHandlerAutoConfiguration {
    @Configuration(proxyBeanMethods = false)
    public static class AnnotationConfig {
        private final ApplicationContext applicationContext;
        public AnnotationConfig(ApplicationContext applicationContext) {
            this.applicationContext = applicationContext;
        }
        @Bean
        public HttpHandler httpHandler(ObjectProvider<WebFluxProperties> propsProvider) {
            // 这里通过WebHttpHandlerBuilder对象创建HttpWebHandlerAdapter对象
            HttpHandler httpHandler = WebHttpHandlerBuilder.applicationContext(this.applicationContext).build();
            // ...
            return httpHandler;
        }

    }
}
public final class WebHttpHandlerBuilder {
    private final List<WebFilter> filters = new ArrayList<>();
    // 获取WebHttpHandlerBuilder对象
    public static WebHttpHandlerBuilder applicationContext(ApplicationContext context) {
        // 构造WebHttpHandlerBuilder对象，在参数上会获取
        // 容器中的webHandler名称的WebHandler对象，
        // 而这里获取的就是DispatcherHandler对象
        // 这个对象是如何创建的？
        // 1.2.1
        WebHttpHandlerBuilder builder = new WebHttpHandlerBuilder(
                context.getBean(WEB_HANDLER_BEAN_NAME, WebHandler.class), context);

        // 获取容器中所有的WebFilter类型的Bean对象
        List<WebFilter> webFilters = context
                .getBeanProvider(WebFilter.class)
                .orderedStream()
                .collect(Collectors.toList());
        // 添加到了filters集合中
        builder.filters(filters -> filters.addAll(webFilters));
        // 获取容器中所有的WebExceptionHandler类型的异常处理句柄
        List<WebExceptionHandler> exceptionHandlers = context
                .getBeanProvider(WebExceptionHandler.class)
                .orderedStream()
                .collect(Collectors.toList());
        builder.exceptionHandlers(handlers -> handlers.addAll(exceptionHandlers));

        context.getBeanProvider(HttpHandlerDecoratorFactory.class)
                .orderedStream()
                .forEach(builder::httpHandlerDecorator);:
        try {
            builder.sessionManager(
                    context.getBean(WEB_SESSION_MANAGER_BEAN_NAME, WebSessionManager.class));
        }catch (NoSuchBeanDefinitionException ex) {
        }
        try {
            builder.codecConfigurer(
                    context.getBean(SERVER_CODEC_CONFIGURER_BEAN_NAME, ServerCodecConfigurer.class));
        }catch (NoSuchBeanDefinitionException ex) {
        }
        try {
            builder.localeContextResolver(
                    context.getBean(LOCALE_CONTEXT_RESOLVER_BEAN_NAME, LocaleContextResolver.class));
        } catch (NoSuchBeanDefinitionException ex) {
        }
        try {
            builder.forwardedHeaderTransformer(
                    context.getBean(FORWARDED_HEADER_TRANSFORMER_BEAN_NAME, ForwardedHeaderTransformer.class));
        } catch (NoSuchBeanDefinitionException ex) {
        }
        return builder;
    }
    public HttpHandler build() {
        // 注意这里将webHandler = DispatcherHandler保存到了
        // FilteringWebHandler对象中
        WebHandler decorated = new FilteringWebHandler(this.webHandler, this.filters);
        // 查看下面的类信息
        decorated = new ExceptionHandlingWebHandler(decorated,  this.exceptionHandlers);

        HttpWebHandlerAdapter adapted = new HttpWebHandlerAdapter(decorated);
        // ...
        return (this.httpHandlerDecorator != null ? this.httpHandlerDecorator.apply(adapted) : adapted);
    }
}
// --------------------------------------------------------------------------------
// 上面获取了WebHttpHandlerBuilder对象后调用build开始创建
// HttpWebHandlerAdapter对象
public class HttpWebHandlerAdapter extends WebHandlerDecorator implements HttpHandler {
    public HttpWebHandlerAdapter(WebHandler delegate) {
        super(delegate);
    }
}
public class WebHandlerDecorator implements WebHandler {
    // 通过上面的源码，知道了这个
    // delegate = ExceptionHandlingWebHandler
    private final WebHandler delegate;
    public WebHandlerDecorator(WebHandler delegate) {
        this.delegate = delegate;
    }
    public WebHandler getDelegate() {
        return this.delegate;
    }
}
// -------------------------------------------------------------------------------
public class FilteringWebHandler extends WebHandlerDecorator {
    private final DefaultWebFilterChain chain;
    // 这里传递过来的handler = DispatcherHandler
    public FilteringWebHandler(WebHandler handler, List<WebFilter> filters) {
        super(handler);
        // 创建了过滤器链
        this.chain = new DefaultWebFilterChain(handler, filters);
    }
}
public class ExceptionHandlingWebHandler extends WebHandlerDecorator {
    private final List<WebExceptionHandler> exceptionHandlers;
    public ExceptionHandlingWebHandler(WebHandler delegate, List<WebExceptionHandler> handlers) {
        // 这里的delegate = FilteringWebHandler
        super(delegate);
        List<WebExceptionHandler> handlersToUse = new ArrayList<>();
        handlersToUse.add(new CheckpointInsertingHandler());
        handlersToUse.addAll(handlers);
        this.exceptionHandlers = Collections.unmodifiableList(handlersToUse);
    }
}

public class WebFluxAutoConfiguration {
    @Configuration(proxyBeanMethods = false)
    @EnableConfigurationProperties(WebProperties.class)
    public static class EnableWebFluxConfiguration extends DelegatingWebFluxConfiguration {
    }
}

public class WebFluxConfigurationSupport implements ApplicationContextAware {
    @Bean
    public DispatcherHandler webHandler() {
        return new DispatcherHandler();
    }
}

public class ExceptionHandlingWebHandler extends WebHandlerDecorator {
    private final List<WebExceptionHandler> exceptionHandlers;
    public Mono<Void> handle(ServerWebExchange exchange) {
        Mono<Void> completion;
        try {
            completion = super.handle(exchange);
        } catch (Throwable ex) {
            completion = Mono.error(ex);
        }
        for (WebExceptionHandler handler : this.exceptionHandlers) {
            completion = completion.onErrorResume(ex -> handler.handle(exchange, ex));
        }
        return completion;
    }
}
public class WebHandlerDecorator implements WebHandler {
    // 通过在上面的build知道该deletgate = FilteringWebHandler
    private final WebHandler delegate;
    public Mono<Void> handle(ServerWebExchange exchange) {
        return this.delegate.handle(exchange);
    }
}
public class FilteringWebHandler extends WebHandlerDecorator {
    private final DefaultWebFilterChain chain;
    public FilteringWebHandler(WebHandler handler, List<WebFilter> filters) {
        // 这个handler = DispatcherHandler对象
        super(handler);
        this.chain = new DefaultWebFilterChain(handler, filters);
    }
    @Override
    public Mono<Void> handle(ServerWebExchange exchange) {
        // 执行所有的过滤器
        return this.chain.filter(exchange);
    }
}
public class DefaultWebFilterChain implements WebFilterChain {
    private final List<WebFilter> allFilters;
    @Nullable
    private final WebFilter currentFilter;
    // handler = DispatcherHandler
    private final WebHandler handler;
    public DefaultWebFilterChain(WebHandler handler, List<WebFilter> filters) {
        Assert.notNull(handler, "WebHandler is required");
        this.allFilters = Collections.unmodifiableList(filters);
        this.handler = handler;
        // 这里就是为了让所有的WebFilter串联起来，按照顺序执行
        DefaultWebFilterChain chain = initChain(filters, handler);
        this.currentFilter = chain.currentFilter;
        this.chain = chain.chain;
    }
    private static DefaultWebFilterChain initChain(List<WebFilter> filters, WebHandler handler) {
        // 这里就相当于是最后执行的一个Chain了，因为第三个参数（currentFilter） = null
        // 在羡慕的filter方法中执行的时候判断了currentFilter如果为空就执行DispatcherHandler#handle方法
        // 这里大致的流程如下：
        DefaultWebFilterChain chain = new DefaultWebFilterChain(filters, handler, null, null);
        ListIterator<? extends WebFilter> iterator = filters.listIterator(filters.size());
        while (iterator.hasPrevious()) {
            chain = new DefaultWebFilterChain(filters, handler, iterator.previous(), chain);
        }
        return chain;
    }
    public Mono<Void> filter(ServerWebExchange exchange) {
        return Mono.defer(() ->
                this.currentFilter != null && this.chain != null ?
                        invokeFilter(this.currentFilter, this.chain, exchange) :
                        // 当过滤器都执行完了以后会执行DispatcherHandler调用
                        this.handler.handle(exchange));
    }
    private Mono<Void> invokeFilter(WebFilter current, DefaultWebFilterChain chain, ServerWebExchange exchange) {
        String currentName = current.getClass().getName();
        return current.filter(exchange, chain).checkpoint(currentName + " [DefaultWebFilterChain]");
    }
}

public class DispatcherHandler implements WebHandler {
    public Mono<Void> handle(ServerWebExchange exchange) {
        if (this.handlerMappings == null) {
            return createNotFoundError();
        }
        if (CorsUtils.isPreFlightRequest(exchange.getRequest())) {
            return handlePreFlight(exchange);
        }
        return Flux.fromIterable(this.handlerMappings)
                .concatMap(mapping -> mapping.getHandler(exchange))
                .next()
                .switchIfEmpty(createNotFoundError())
                .flatMap(handler -> invokeHandler(exchange, handler))
                .flatMap(result -> handleResult(exchange, result));
    }
}