springboot中如何注冊servlet和filter

法1

通過@web*開頭的注解

    @WebServlet
    public class HalloServlet extends HttpServlet{}

---

    @WebFilter("/hello/*")
    public class HelloFilter implements Filter {}

---    

    //啟動類中需要通過@ServletComponentScan掃描這些注解
    @SpringBootApplication
    @ServletComponentScan
    public class SpringBootServletApplication {
       public static void main(String[] args) {
           SpringApplication.run(SpringBootServletApplication.class, args);
       }
    }

法2

通過RegistrationBean

    @Bean
    public ServletRegistrationBean helloServlet() {
        ServletRegistrationBean helloServlet = new ServletRegistrationBean();
        helloServlet.addUrlMappings("/hello");
        helloServlet.setServlet(new HelloServlet());
        return helloServlet;
    }

    @Bean
    public FilterRegistrationBean helloFilter() {
        FilterRegistrationBean helloFilter = new FilterRegistrationBean();
        helloFilter.addUrlPatterns("/hello/*");
        helloFilter.setFilter(new HelloFilter());
        return helloWorldFilter;
    }

springboot如何加載DispatcherServlet？

我們只討論springboot環(huán)境下使用內(nèi)嵌tomcat容器的情況。
springboot版本 2.0.2.RELEASE

DispatcherServlet初始化的位置

// DispatcherServletAutoConfiguration.class


/*
 * The bean name for a DispatcherServlet that will be mapped to the root URL "/"
 */
public static final String DEFAULT_DISPATCHER_SERVLET_BEAN_NAME = "dispatcherServlet";

@Bean(name = DEFAULT_DISPATCHER_SERVLET_BEAN_NAME)
public DispatcherServlet dispatcherServlet() {
    DispatcherServlet dispatcherServlet = new DispatcherServlet();
    dispatcherServlet.setDispatchOptionsRequest(
            this.webMvcProperties.isDispatchOptionsRequest());
    dispatcherServlet.setDispatchTraceRequest(
            this.webMvcProperties.isDispatchTraceRequest());
    dispatcherServlet.setThrowExceptionIfNoHandlerFound(
            this.webMvcProperties.isThrowExceptionIfNoHandlerFound());
    return dispatcherServlet;
}

同時(shí)也需要關(guān)注下DispatcherServlet的RigistrationBean

// DispatcherServletAutoConfiguration.class

/*
 * The bean name for a DispatcherServlet that will be mapped to the root URL "/"
 */
public static final String DEFAULT_DISPATCHER_SERVLET_BEAN_NAME = "dispatcherServlet";

/*
 * The bean name for a ServletRegistrationBean for the DispatcherServlet "/"
 */
public static final String DEFAULT_DISPATCHER_SERVLET_REGISTRATION_BEAN_NAME = "dispatcherServletRegistration";

@Bean(name = DEFAULT_DISPATCHER_SERVLET_REGISTRATION_BEAN_NAME)
@ConditionalOnBean(value = DispatcherServlet.class, name = DEFAULT_DISPATCHER_SERVLET_BEAN_NAME)
public ServletRegistrationBean<DispatcherServlet> dispatcherServletRegistration(
        DispatcherServlet dispatcherServlet) {
    ServletRegistrationBean<DispatcherServlet> registration = new ServletRegistrationBean<>(
            dispatcherServlet,
            this.serverProperties.getServlet().getServletMapping());
    registration.setName(DEFAULT_DISPATCHER_SERVLET_BEAN_NAME);
    registration.setLoadOnStartup(
            this.webMvcProperties.getServlet().getLoadOnStartup());
    if (this.multipartConfig != null) {
        registration.setMultipartConfig(this.multipartConfig);
    }
    return registration;
}

可以看出，ServletRegistrationBean中的servlet屬性即為上面創(chuàng)建的DispatcherServlet，現(xiàn)在 RegistrationBean準(zhǔn)備好了，那么是在什么地方被使用的呢？

DispatcherServlet的加載流程

TomcatStarter
首先在tomcatstarter中 會有整體對ServletContextInitializer的onStartup方法的調(diào)用。

// TomcatStarter.class

@Override
public void onStartup(Set<Class<?>> classes, ServletContext servletContext)
        throws ServletException {
    try {
        for (ServletContextInitializer initializer : this.initializers) {
            initializer.onStartup(servletContext);
        }
    }
    catch (Exception ex) {
        this.startUpException = ex;
        // Prevent Tomcat from logging and re-throwing when we know we can
        // deal with it in the main thread, but log for information here.
        if (logger.isErrorEnabled()) {
            logger.error("Error starting Tomcat context. Exception: "
                    + ex.getClass().getName() + ". Message: " + ex.getMessage());
        }
    }
}

image.png

通過斷點(diǎn)信息，并沒有在其中發(fā)現(xiàn)RegisterBean的相關(guān)信息
為了真正理解springboot如何加載servlet，還得研究下ServletWebServerApplicationContext，來看看ServletWebServerApplicationContext的邏輯流程：

//ServletWebServerApplicationContext.class

// 首先是onRefresh方法
@Override
protected void onRefresh() {
    super.onRefresh();
    try {
        createWebServer();
    }
    catch (Throwable ex) {
        throw new ApplicationContextException("Unable to start web server", ex);
    }
}

// onRefresh方法中 調(diào)用了createWebServer()
private void createWebServer() {
    WebServer webServer = this.webServer;
    ServletContext servletContext = getServletContext();
    if (webServer == null && servletContext == null) {
        ServletWebServerFactory factory = getWebServerFactory();
        //下一層的入口 
        this.webServer = factory.getWebServer(getSelfInitializer());
    }
    else if (servletContext != null) {
        try {
            getSelfInitializer().onStartup(servletContext);
        }
        catch (ServletException ex) {
            throw new ApplicationContextException("Cannot initialize servlet context",
                    ex);
        }
    }
    initPropertySources();
}

// 然后是下一層的入口getSelfInitializer()，我們在TomcatStarter中看到的其實(shí)就是這里返回的匿名類
private org.springframework.boot.web.servlet.ServletContextInitializer getSelfInitializer() {
    return this::selfInitialize;
}

private void selfInitialize(ServletContext servletContext) throws ServletException {
    prepareWebApplicationContext(servletContext);
    ConfigurableListableBeanFactory beanFactory = getBeanFactory();
    ExistingWebApplicationScopes existingScopes = new ExistingWebApplicationScopes(
            beanFactory);
    WebApplicationContextUtils.registerWebApplicationScopes(beanFactory,
            getServletContext());
    existingScopes.restore();
    WebApplicationContextUtils.registerEnvironmentBeans(beanFactory,
            getServletContext());
    for (ServletContextInitializer beans : getServletContextInitializerBeans()) {
        beans.onStartup(servletContext);
    }
}

// 在selfInitialize()中，轉(zhuǎn)而調(diào)用getServletContextInitializerBeans()
protected Collection<ServletContextInitializer> getServletContextInitializerBeans() {
    return new ServletContextInitializerBeans(getBeanFactory());
}

現(xiàn)在流程轉(zhuǎn)到了ServletContextInitializerBeans中，那么ServletContextInitializerBeans做了什么呢

//ServletContextInitializerBeans.class

// 在構(gòu)造方法中，調(diào)用了addServletContextInitializerBeans()方法，看起來離我們的目標(biāo)不遠(yuǎn)了
public ServletContextInitializerBeans(ListableBeanFactory beanFactory) {
    this.initializers = new LinkedMultiValueMap<>();
    addServletContextInitializerBeans(beanFactory);
    addAdaptableBeans(beanFactory);
    List<ServletContextInitializer> sortedInitializers = new ArrayList<>();
    this.initializers.values().forEach((contextInitializers) -> {
        AnnotationAwareOrderComparator.sort(contextInitializers);
        sortedInitializers.addAll(contextInitializers);
    });
    this.sortedList = Collections.unmodifiableList(sortedInitializers);
}

//addServletContextInitializerBeans()方法會去容器中尋找注冊過的ServletContextInitializer
private void addServletContextInitializerBeans(ListableBeanFactory beanFactory) {
    for (Entry<String, ServletContextInitializer> initializerBean : getOrderedBeansOfType(
            beanFactory, ServletContextInitializer.class)) {
        addServletContextInitializerBean(initializerBean.getKey(),
                initializerBean.getValue(), beanFactory);
    }
}

image.png

到此，整個(gè)流程就串起來了。

• 首先，以ServletWebServerApplicationContext中的onRefresh()方法為起點(diǎn)觸發(fā)配置了一個(gè)匿名的ServletContextInitializer。
• 這個(gè)匿名的ServletContextInitializer的onStart方法會去容器中找到所有的RegisterBean并加載到ServletContext中，ServletContainer 其實(shí)就是 servlet filter listener 的匯總。
• 這個(gè)匿名的ServletContextInitializer最終傳遞給TomcatStarter，由TomcatStarter的onStartup方法觸發(fā)ServletContextInitializer 的 onStartup 方法，完成裝配。

后記

是哪里調(diào)用的TomcatStarter的onStart方法呢

在TomcatServletWebServerFactory中可以看到TomcatStart是被new出來的

// TomcatServletWebServerFactory.class

protected void configureContext(Context context,
        ServletContextInitializer[] initializers) {
    TomcatStarter starter = new TomcatStarter(initializers);
    if (context instanceof TomcatEmbeddedContext) {
        // Should be true
        ((TomcatEmbeddedContext) context).setStarter(starter);
    }
    context.addServletContainerInitializer(starter, NO_CLASSES);
    ......
}

那么 TomcatServletWebServerFactory 又是如何被創(chuàng)建的呢，經(jīng)過尋找，是在ServletWebServerFactoryConfiguration中

//ServletWebServerFactoryConfiguration.class

@Configuration
@ConditionalOnClass({ Servlet.class, Tomcat.class, UpgradeProtocol.class })
@ConditionalOnMissingBean(value = ServletWebServerFactory.class, search = SearchStrategy.CURRENT)
public static class EmbeddedTomcat {

    @Bean
    public TomcatServletWebServerFactory tomcatServletWebServerFactory() {
        return new TomcatServletWebServerFactory();
    }

}

滿足上面的條件，就會觸發(fā)springboot的自動配置。

后后記

ServletRegistrationBean的onStract方法干了什么呢？

image.png

可以看到ServletRegistrationBean是繼承自RegistartionBean的,在RegistartionBean的onStart方法中，調(diào)用了register方法。

// RegistartionBean.class

@Override
public final void onStartup(ServletContext servletContext) throws ServletException {
    String description = getDescription();
    if (!isEnabled()) {
        logger.info(StringUtils.capitalize(description)
                + " was not registered (disabled)");
        return;
    }
    register(description, servletContext);
}

register方法被DynamicRegistrationBean實(shí)現(xiàn),在register方法中，轉(zhuǎn)而調(diào)用了addRegistration方法。

// DynamicRegistrationBean.class

@Override
protected final void register(String description, ServletContext servletContext) {
    D registration = addRegistration(description, servletContext);
    if (registration == null) {
        logger.info(StringUtils.capitalize(description) + " was not registered "
                + "(possibly already registered?)");
        return;
    }
    configure(registration);
}

該方法最終被ServletRegistrationBean實(shí)現(xiàn)。

//ServletRegistrationBean.class

protected ServletRegistration.Dynamic addRegistration(String description,
        ServletContext servletContext) {
    String name = getServletName();
    logger.info("Servlet " + name + " mapped to " + this.urlMappings);
    return servletContext.addServlet(name, this.servlet);
}

可以看到 ，最終其實(shí)就是將servlet放入servletContext中。