一、自定义SpringApplication

默认的我们都是直接通过SpringApplication的run方法来直接启动Spring Boot,其实我们可以通过一些API来调整某些行为。

1.1 通过SpringApplication API调整

我们新建一个SpringBoot项目,Spring Boot版本为2.1.0.RELEASE,artifactId为SpringApplication,并引入spring-boot-starter-web依赖。项目结构如下所示:

我们将入口类的代码改为:

1
2
3
4
5
SpringApplication application = new SpringApplication(DemoApplication.class);
application.setBannerMode(Banner.Mode.OFF);
application.setWebApplicationType(WebApplicationType.SERVLET);
application.setAdditionalProfiles("dev");
application.run(args);

通过调用SpringApplication的方法,我们关闭了Banner的打印,设置应用环境为非WEB应用,profiles指定为dev。除此之外,SpringApplication还包含了许多别的方法,具体可以查看源码或者官方文档:

1.2 通过SpringApplicationBuilder API调整

SpringApplicationBuilder提供了Fluent API,可以实现链式调用,下面的代码和上面的效果一致,但在编写上较为方便:

1
2
3
4
5
6
new SpringApplicationBuilder(Application.class)
.bannerMode(Banner.Mode.OFF)
.web(WebApplicationType.SERVLET)
.profiles("dev")
.addCommandLineProperties(false)
.run(args);

二、SpringApplication准备阶段

SpringApplication的生命周期阶段大致可以分为准备阶段和运行阶段。

我们通过源码来查看SpringApplication的有参构造器:

通过有参构造器里的代码我们可以将SpringApplication的准备阶段分为以下几个步骤:

2.1 配置源

构造器中this.primarySources = new LinkedHashSet<>(Arrays.asList(primarySources));这行代码用于加载我们配置的Spring Boot Bean源。通常我们使用SpringApplication或者SpringApplicationBuilder的构造器来直接指定源。

所谓的Spring Boot Bean源指的是某个被@SpringBootApplication注解标注的类,比如入口类:

1
2
3
4
5
6
7
8
@SpringBootApplication
public class Application {

public static void main(String[] args) {
SpringApplication app = new SpringApplication(Application.class);
app.run(args);
}
}

我们也可以将上面的代码改为下面这种方式:

1
2
3
4
5
6
7
8
9
10
11
12
public class Application {

public static void main(String[] args) {
SpringApplication app = new SpringApplication(ApplicationResource.class);
app.run(args);
}

@SpringBootApplication
public static class ApplicationResource {

}
}

这样也是可行的。查看SpringApplication的单个参数构造器:

说明我们除了配置单个源外,还可以配置多个源。

2.2 推断应用类型

构造器中这行this.webApplicationType = WebApplicationType.deduceFromClasspath();代码用于推断当前Spring Boot应用类型。

Spring Boot 2.0后,应用可以分为下面三种类型:

  1. WebApplicationType.NONE:非WEB类型;
  2. WebApplicationType.REACTIVE:Web Reactive类型;
  3. WebApplicationType.SERVLET:Web Servlet类型。

WebApplicationType.deduceFromClasspath()方法根据当前应用ClassPath中是否存在相关的实现类来判断应用类型到底是哪个,deduceFromClasspath方法的源码如下所示:

我们也可以直接通过SpringApplication的setWebApplicationType方法或者SpringApplicationBuilder的web方法来指定当前应用的类型。

2.3 加载应用上下文初始器

接着下一行代码setInitializers((Collection) getSpringFactoriesInstances(ApplicationContextInitializer.class));用于加载应用上下文初始器ApplicationContextInitializer。

getSpringFactoriesInstances方法的源码如下所示:

上面代码利用Spring工厂加载机制,实例化ApplicationContextInitializer实现类,并进行排序。

所以我们可以通过实现ApplicationContextInitializer接口用于在Spring Boot应用初始化之前执行一些自定义操作。

举个例子,在com.example.demo下新建initializer包,然后创建一个HelloApplicationContextInitializer类,实现ApplicationContextInitializer接口:

1
2
3
4
5
6
7
@Order(Ordered.HIGHEST_PRECEDENCE)
public class HelloApplicationContextInitializer implements ApplicationContextInitializer {
@Override
public void initialize(ConfigurableApplicationContext applicationContext) {
System.out.println("ConfigurableApplicationContext.id - " + applicationContext.getId());
}
}

上面代码中实现了initialize方法,并且使用@Order注解指定优先级。其中Ordered.HIGHEST_PRECEDENCE等于Integer.MIN_VALUE,Ordered.LOWEST_PRECEDENCE等于Integer.MAX_VALUE。所以数值越小,优先级越高。

除了使用@Order注解来指定优先级外,我们也可以通过实现org.springframework.core.Ordered接口的getOrder方法来指定优先级。

接着我们来创建一个优先级比HelloApplicationContextInitializer低的Initializer —— AfterHelloApplicationContextInitializer:

1
2
3
4
5
6
7
8
9
10
11
public class AfterHelloApplicationContextInitializer implements ApplicationContextInitializer, Ordered {
@Override
public void initialize(ConfigurableApplicationContext applicationContext) {
System.out.println("AfterHelloApplicationContextInitializer: " + applicationContext.getId());
}

@Override
public int getOrder() {
return Ordered.LOWEST_PRECEDENCE;
}
}

上面通过getOrder方法来指定了优先级为最低优先级。

创建好后,我们还需在工厂配置文件里配置这两个实现类。在resources目录下新建META-INF目录,并创建spring.factories文件:

1
2
3
4
# Initializers
org.springframework.context.ApplicationContextInitializer=\
com.wno704.boot.HelloApplicationContextInitializer,\
com.wno704.boot.AfterHelloApplicationContextInitializer

这时候,启动Spring Boot项目,会发现控制台在打印Banner后就执行了这两个初始化器,并且HelloApplicationContextInitializer的initialize方法执行时机先于AfterHelloApplicationContextInitializer的initialize方法:

2.4 加载应用事件监听器

在加载完应用上下文初始器后,下一行的setListeners((Collection) getSpringFactoriesInstances(ApplicationListener.class));代码加载了应用事件监听器。与加载事件上下文初始器类似,Spring Boot也是通过Spring的工厂方法来实例化ApplicationListener的实现类,并进行排序。

既然是事件监听,那么其可以监听什么事件呢?其监听的是ApplicationEvent接口的实现类,我们查看一下都有哪些事件实现了这个接口:

这里我们以ContextClosedEvent为例子来编写自定义的应用事件监听器,监听Spring上下文关闭事件。
在com.wno704.boot下新建listener包,然后创建一个ContextClosedEventListener类,实现ApplicationListener接口:

1
2
3
4
5
6
7
@Order(Ordered.HIGHEST_PRECEDENCE)
public class ContextClosedEventListener implements ApplicationListener<ContextClosedEvent> {
@Override
public void onApplicationEvent(ContextClosedEvent event) {
System.out.println("ContextClosedEvent: " + event.getApplicationContext().getId());
}
}

上面代码实现了对ContextClosedEvent事件的监听,并且分配了最高优先级。

接着创建一个优先级比ContextClosedEventListener低的监听器AfterContextClosedEventListener:

1
2
3
4
5
6
7
8
9
10
11
public class AfterContextClosedEventListener implements ApplicationListener<ContextClosedEvent>, Ordered {
@Override
public int getOrder() {
return Ordered.HIGHEST_PRECEDENCE + 1;
}

@Override
public void onApplicationEvent(ContextClosedEvent event) {
System.out.println("AfterContextClosedEventr: " + event.getApplicationContext().getId());
}
}

最后,别忘了在Spring工厂配置文件里进行配置:

1
2
3
4
# Application Listeners
org.springframework.context.ApplicationListener=\
com.wno704.boot.listener.ContextClosedEventListener,\
com.wno704.boot.listener.AfterContextClosedEventListener

运行Spring Boot入口类,控制台输出如下:

2.5 推断入口类

接着构造器里的代码下一行this.mainApplicationClass = deduceMainApplicationClass();用于推断运行Spring Boot应用的入口类。查看deduceMainApplicationClass方法源码:

代码主要逻辑是根据Main线程执行堆栈判断实际的入口类。

准备阶段介绍完毕后,接下来开始介绍运行阶段。

三、SpringApplication运行阶段

SpringApplication的运行阶段对应SpringApplication的run方法,我们查看其源码:

运行阶段大致可以分为下面这几个过程:

3.1 开启时间监听

run方法开头的这两行代码用于开启时间监听:

1
2
StopWatch stopWatch = new StopWatch();
stopWatch.start();

上面代码用于开启Spring Boot应用启动时间监听,配合下面的stopWatch.stop();便可以计算出完整的启动时间。
run方法的这几行代码用于加载Spring应用运行监听器(SpringApplicationRunListener):

1
2
SpringApplicationRunListeners listeners = getRunListeners(args);
listeners.started();

getRunListeners方法源码:

上面代码通过SpringFactoriesLoader检索META-INF/spring.factories找到声明的所有SpringApplicationRunListener的实现类并将其实例化,然后装配到List运行监听器集合中。

listeners.started();用于遍历运行监听器集合中的所有SpringApplicationRunListener的实现类,并逐一调用它们的starting方法,广播Spring Boot应用要开始启动了。

在Spring Boot中SpringApplicationRunListener接口用于监听整个Spring Boot应用生命周期,其代码如下所示:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/**
* A collection of {@link SpringApplicationRunListener}.
*
* @author Phillip Webb
*/
class SpringApplicationRunListeners {

private final Log log;

private final List<SpringApplicationRunListener> listeners;

SpringApplicationRunListeners(Log log, Collection<? extends SpringApplicationRunListener> listeners) {
this.log = log;
this.listeners = new ArrayList<>(listeners);
}

void starting() {
for (SpringApplicationRunListener listener : this.listeners) {
listener.starting();
}
}

void environmentPrepared(ConfigurableEnvironment environment) {
for (SpringApplicationRunListener listener : this.listeners) {
listener.environmentPrepared(environment);
}
}

void contextPrepared(ConfigurableApplicationContext context) {
for (SpringApplicationRunListener listener : this.listeners) {
listener.contextPrepared(context);
}
}

void contextLoaded(ConfigurableApplicationContext context) {
for (SpringApplicationRunListener listener : this.listeners) {
listener.contextLoaded(context);
}
}

void started(ConfigurableApplicationContext context) {
for (SpringApplicationRunListener listener : this.listeners) {
listener.started(context);
}
}

void running(ConfigurableApplicationContext context) {
for (SpringApplicationRunListener listener : this.listeners) {
listener.running(context);
}
}

void failed(ConfigurableApplicationContext context, Throwable exception) {
for (SpringApplicationRunListener listener : this.listeners) {
callFailedListener(listener, context, exception);
}
}

private void callFailedListener(SpringApplicationRunListener listener, ConfigurableApplicationContext context,
Throwable exception) {
try {
listener.failed(context, exception);
}
catch (Throwable ex) {
if (exception == null) {
ReflectionUtils.rethrowRuntimeException(ex);
}
if (this.log.isDebugEnabled()) {
this.log.error("Error handling failed", ex);
}
else {
String message = ex.getMessage();
message = (message != null) ? message : "no error message";
this.log.warn("Error handling failed (" + message + ")");
}
}
}
}

这些方法对应着Spring Boot应用生命周期的各个阶段:

方法名称对应生命周期Spring Boot起始版本
starting()Spring应用刚启动1.0
environmentPrepared(ConfigurableEnvironment)ConfigurableEnvironment 准备完毕,允许将其调整1.0
contextPrepared(ConfigurableApplicationContext)ConfigurableApplicationContext 准备完毕,允许将其调整1.0
contextLoaded(ConfigurableApplicationContext)ConfigurableApplicationContext 已装载,但仍未启动1.0
started(ConfigurableApplicationContext)ConfigurableApplicationContext 已启动,此时 Spring Bean 已初始化完成2.0
running(ConfigurableApplicationContext)Spring 应用正在运行2.0
failed(ConfigurableApplicationContext,Throwable)Spring 应用运行失败2.0

我们在com.wno704.boot.listener下自定义一个SpringApplicationRunListener接口实现类HelloSpringApplicationRunListener:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
public class HelloSpringApplicationRunListener implements SpringApplicationRunListener {
public HelloSpringApplicationRunListener(SpringApplication application, String[] args) {
}

@Override
public void starting() {
System.out.println("HelloApplicationRunListener starting......");
}

@Override
public void environmentPrepared(ConfigurableEnvironment environment) {

}

@Override
public void contextPrepared(ConfigurableApplicationContext context) {

}

@Override
public void contextLoaded(ConfigurableApplicationContext context) {

}

@Override
public void started(ConfigurableApplicationContext context) {

}

@Override
public void running(ConfigurableApplicationContext context) {

}

@Override
public void failed(ConfigurableApplicationContext context, Throwable exception) {

}
}

通过这个实现类,我们可以在Spring Boot应用刚启动的时候在控制台输出HelloApplicationRunListener starting......。

因为其基于Spring的工厂方法来实现,所以我们需要在spring.factories文件里配置这个实现类:

1
2
org.springframework.boot.SpringApplicationRunListener=\
com.wno704.boot.listener.HelloSpringApplicationRunListener

启动Spring Boot应用便可以在控制台看到如下输出了:

3.2 创建 Environment

run方法中的这行代码用于创建并配置当前SpringBoot应用将要使用的Environment(包括配置要使用的PropertySource以及Profile):

我们已经在准备阶段里推断出了应用类型,这里只要根据相应的应用类型来创建相应的应用环境即可,类型和环境对应关系如下:

Web Reactive: StandardReactiveWebEnvironment

Web Servlet: StandardServletEnvironment

非 Web: StandardEnvironment

在prepareEnvironment方法中会执行listeners.environmentPrepared(environment);,用于遍历调用所有SpringApplicationRunListener实现类的environmentPrepared()方法,广播Environment准备完毕。

3.3 是否打印Banner

run方法中的这行代码会根据我们的配置来决定是否打印Banner:

1
Banner printedBanner = printBanner(environment);

3.4 创建Context

run方法中的这行代码用于创建ApplicationContext:

1
context = createApplicationContext();

不同的环境对应不同的ApplicationContext:

Web Reactive: AnnotationConfigReactiveWebServerApplicationContext

Web Servlet: AnnotationConfigServletWebServerApplicationContext

非 Web: AnnotationConfigApplicationContext

3.5 装配Context

run方法中的这行代码用于装配Context:

1
prepareContext(context, environment, listeners, applicationArguments, printedBanner);

3.6 方法prepareContext的源码如下所示:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
private void prepareContext(ConfigurableApplicationContext context,
ConfigurableEnvironment environment, SpringApplicationRunListeners listeners,
ApplicationArguments applicationArguments, Banner printedBanner) {
context.setEnvironment(environment);
postProcessApplicationContext(context);
applyInitializers(context);
listeners.contextPrepared(context);
if (this.logStartupInfo) {
logStartupInfo(context.getParent() == null);
logStartupProfileInfo(context);
}
// Add boot specific singleton beans
ConfigurableListableBeanFactory beanFactory = context.getBeanFactory();
beanFactory.registerSingleton("springApplicationArguments", applicationArguments);
if (printedBanner != null) {
beanFactory.registerSingleton("springBootBanner", printedBanner);
}
if (beanFactory instanceof DefaultListableBeanFactory) {
((DefaultListableBeanFactory) beanFactory)
.setAllowBeanDefinitionOverriding(this.allowBeanDefinitionOverriding);
}
// Load the sources
Set<Object> sources = getAllSources();
Assert.notEmpty(sources, "Sources must not be empty");
load(context, sources.toArray(new Object[0]));
listeners.contextLoaded(context);
}

prepareContext方法开头为ApplicationContext加载了environment,之后通过applyInitializers方法逐个执行ApplicationContextInitializer的initialize方法来进一步封装ApplicationContext,并调用所有的SpringApplicationRunListener实现类的contextPrepared方法,广播ApplicationContext已经准备完毕了。

之后初始化IOC容器,并调用SpringApplicationRunListener实现类的contextLoaded方法,广播ApplicationContext加载完成,这里就包括通过@EnableAutoConfiguration导入的各种自动配置类。

3.7 Refresh Context

run方法中的这行代码用于初始化所有自动配置类,并调用ApplicationContext的refresh方法:

1
refreshContext(context);

3.8 广播应用已启动

run方法中的这行代码用于广播Spring Boot应用已启动:

1
listeners.started(context);

started方法会调用所有的SpringApplicationRunListener的finished方法,广播SpringBoot应用已经成功启动。

3.9 执行Runner

run方法中的这行代码callRunners(context, applicationArguments);遍历所有ApplicationRunner和CommandLineRunner的实现类,并执行其run方法。我们可以实现自己的ApplicationRunner或者CommandLineRunner,来对Spring Boot的启动过程进行扩展。

我们在com.wno704.boot下新建runner包,然后创建一个ApplicationRunner的实现类HelloApplicationRunner:

1
2
3
4
5
6
7
@Component
public class HelloApplicationRunner implements ApplicationRunner {
@Override
public void run(ApplicationArguments args) {
System.out.println("HelloApplicationRunner: hello spring boot");
}
}

这里我们需要将HelloApplicationRunner使用@Component注解标注,让其注册到IOC容器中。

然后再创建一个CommandLineRunner的实现类HelloCommandLineRunner:

1
2
3
4
5
6
7
@Component
public class HelloCommandLineRunner implements CommandLineRunner {
@Override
public void run(String... args) {
System.out.println("HelloCommandLineRunner: hello spring boot");
}
}

启动Spring Boot应用,便可以在应用刚启动好后看到如下输出:

3.10 广播应用运行中

run方法中的这行代码listeners.running(context);用于调用SpringApplicationRunListener的running方法,广播Spring Boot应用正在运行中。

当run方法运行出现异常时,便会调用handleRunFailure方法来处理异常,该方法里会通过listeners.failed(context, exception);来调用SpringApplicationRunListener的failed方法,广播应用启动失败,并将异常扩散出去。

注意
上面所有的广播事件都是使用Spring的应用事件广播器接口ApplicationEventMulticaster的实现类SimpleApplicationEventMulticaster来进行广播的。