引言

本文將作為Spring系列教程中源碼版塊的第一篇，整個(gè)源碼系列將分為兩部分進(jìn)行介紹；單純的源碼解析，大概率是個(gè)吃力沒人看的事情，因此我們將結(jié)合源碼解析，一個(gè)是學(xué)習(xí)下別人的優(yōu)秀設(shè)計(jì)，一個(gè)是站在源碼的角度看一下我們除了日常的CURD之外，還可以干些啥

在Spring的啟動(dòng)過程中，一系列的操作步驟中，提供了很多的擴(kuò)展點(diǎn)，供我們來增強(qiáng)；簡單來說就是提供了很多的鉤子，這樣當(dāng)我們在某個(gè)節(jié)點(diǎn)執(zhí)行前后，想干點(diǎn)其他的事情時(shí)，可以很簡單的支持；

本文介紹的ApplicationContextInitializer，spring容器在刷新之前會(huì)回調(diào)這個(gè)接口，從而實(shí)現(xiàn)在spring容器未初始化前，干一些用戶希望做的事情

I. 項(xiàng)目準(zhǔn)備

本文創(chuàng)建的實(shí)例工程采用SpringBoot 2.2.1.RELEASE + maven 3.5.3 + idea進(jìn)行開發(fā)

具體的SpringBoot項(xiàng)目工程創(chuàng)建就不贅述了，核心的pom文件，無需額外的依賴

配置文件 application.yml， 也沒有什么特殊的配置

源碼工程參考文末的源碼

II. 容器刷新前擴(kuò)展點(diǎn)實(shí)例

1. 自定義ApplicationContextInitializer

當(dāng)我們希望實(shí)現(xiàn)一個(gè)自定義的上下文初始化時(shí)，非常簡單，實(shí)現(xiàn)上面這個(gè)接口就行了，如

public class ApplicationContextInitializer01 implements ApplicationContextInitializer {
    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
        System.out.println("ApplicationContextInitializer01");
    }
}

2. 擴(kuò)展點(diǎn)注冊

上面自定義一個(gè)擴(kuò)展點(diǎn)，如何使它生效呢？

官方提供了三種方式，如在啟動(dòng)時(shí)，直接進(jìn)行注冊: springApplication.addInitializers(new ApplicationContextInitializer01());

@SpringBootApplication
public class Application {
    public static void main(String[] args) {
        SpringApplication springApplication = new SpringApplication(Application.class);
        springApplication.addInitializers(new ApplicationContextInitializer01());
        try (ConfigurableApplicationContext context = springApplication.run(args)) {
        }
    }
}

當(dāng)我們的擴(kuò)展點(diǎn)是放在一個(gè)jar包中對外提供時(shí)，使用上面的啟動(dòng)注冊方式顯然是不可行的，此時(shí)更推薦的做法就是通過Spring的SPI機(jī)制進(jìn)行注冊

在資源目錄下的META-INF/spring.factories文件中進(jìn)行注冊

org.springframework.context.ApplicationContextInitializer=com.git.hui.extention.context.ApplicationContextInitializer02

說明

• 上面SPI的機(jī)制非常推薦大家使用，在之前的文章中，AutoConfiguration的注冊通常也是使用這種方式

除了上面的兩種注冊方式之外，另外還有一個(gè)配置文件的方式，在配置文件application.properties 或 application.yml中，如下配置

context:
  initializer:
    classes: com.git.hui.extention.context.ApplicationContextInitializer03

啟動(dòng)測試

上面三種注冊方式，我們實(shí)現(xiàn)三個(gè)自定義的擴(kuò)展點(diǎn)，然后啟動(dòng)之后，看一下實(shí)際輸出

上面的輸出，可以簡單的得出一個(gè)結(jié)論，不同注冊方式的優(yōu)先級（為了更合理的驗(yàn)證下面的觀點(diǎn)，推薦大家修改下上面三個(gè)自定義擴(kuò)展點(diǎn)名，排除掉是因?yàn)閿U(kuò)展名導(dǎo)致的排序問題）

• 配置文件注冊 > SPI注冊 > 啟動(dòng)時(shí)注冊

3. 執(zhí)行順序指定

對于自定義的擴(kuò)展點(diǎn)實(shí)現(xiàn)，當(dāng)存在順序關(guān)系時(shí)，我們可以通過@Order注解來實(shí)現(xiàn)， 如當(dāng)上面的三個(gè)擴(kuò)展點(diǎn)都是通過啟動(dòng)方式注冊時(shí)

@Order(5)
public class ApplicationContextInitializer01 implements ApplicationContextInitializer {
    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
        System.out.println("ApplicationContextInitializer01");
    }
}
@Order(2)
public class ApplicationContextInitializer02 implements ApplicationContextInitializer {
    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
        System.out.println("ApplicationContextInitializer02");
    }
}
@Order(10)
public class ApplicationContextInitializer03 implements ApplicationContextInitializer {
    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
        System.out.println("ApplicationContextInitializer03");
    }
}
@SpringBootApplication
public class Application {
    public static void main(String[] args) {
        SpringApplication springApplication = new SpringApplication(Application.class);
        springApplication.addInitializers(new ApplicationContextInitializer01(), new ApplicationContextInitializer02(), new ApplicationContextInitializer03());
        try (ConfigurableApplicationContext context = springApplication.run(args)) {
        }
    }
}

輸出實(shí)例如下

接著重點(diǎn)來了

• 若上面的三個(gè)自定義實(shí)現(xiàn)，不是相同的注冊方式，如將03采用配置文件方式進(jìn)行注冊，那么01, 02 依然是啟動(dòng)注冊
• 則順序是 03 > 02 > 01
• 即 @Order注解修飾的順序，并不能打破 配置文件 > SPI > 啟動(dòng)方式注冊的順序

關(guān)于自定義實(shí)現(xiàn)類的執(zhí)行順序，規(guī)則如下

• 配置文件 > SPI > 啟動(dòng)方式
• 相同的注冊方式，可以通過 @Order 注解進(jìn)行修飾，值越小則優(yōu)先級越高

4. 使用場景示例

最后我們再來看一下，這個(gè)擴(kuò)展點(diǎn)到底有什么用，我們再什么場景下會(huì)用到這個(gè)呢？

一個(gè)經(jīng)?？梢钥吹降膽?yīng)用場景如通過它來指定需要激活的配置文件

public class ApplicationContextInitializer03 implements ApplicationContextInitializer {
    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
        // 指定激活prod對應(yīng)的配置文件
        configurableApplicationContext.getEnvironment().setActiveProfiles("prod");
    }
}

但是一般也很少見到有人這么干，因?yàn)橹苯邮褂门渲脜?shù)就行了，那么有場景需要這么做么？

答案當(dāng)然是有的，比如現(xiàn)在廣為流行的docker容器部署，當(dāng)我們希望每次都是打同一個(gè)鏡像，然后在實(shí)際運(yùn)行的時(shí)候，根據(jù)不同的環(huán)境來決定當(dāng)前鏡像到底啟用哪些配置文件，這時(shí)就有用了

比如我們通過容器的環(huán)境參數(shù) app.env 來獲取當(dāng)前運(yùn)行的環(huán)境，如果是prod，則激活application-prod.yml; 如果是test，則激活application-test.yml

那么此時(shí)可以這么干

public class EenvActiveApplicationContextInitializer implements ApplicationContextInitializer {
    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
        String env = System.getenv("app.env");
        if ("prod".equalsIgnoreCase(env)) {
            configurableApplicationContext.getEnvironment().setActiveProfiles("prod");
        } else if ("test".equalsIgnoreCase(env)) {
            configurableApplicationContext.getEnvironment().setActiveProfiles("test");
        } else {
            throw new RuntimeException("非法的環(huán)境參數(shù)：" + env);
        }
    }
}

5. 小結(jié)

本文作為擴(kuò)展點(diǎn)的第一篇，通過實(shí)現(xiàn)ApplicationContextInitializer接口，從而達(dá)到在spring容器刷新之前做某些事情的目的

通常自定義的ApplicationContextInitializer有三種注冊方式，按照優(yōu)先級如下

• 配置文件 > SPI方式 > 啟動(dòng)方式注冊
• 相同的注冊方式中，可以使用@Order注解來指定優(yōu)先級，值越小優(yōu)先級越高

最后還給出了一個(gè)可以應(yīng)用的實(shí)例場景，即如何實(shí)現(xiàn)一個(gè)鏡像在不同的環(huán)境中啟動(dòng)運(yùn)行

下一個(gè)擴(kuò)展點(diǎn)我們將介紹如何通過BeanDefinitionRegistryPostProcessor來實(shí)現(xiàn)非Spring生態(tài)的Bean加載使用

III. 不能錯(cuò)過的源碼和相關(guān)知識點(diǎn)

項(xiàng)目

工程：github.com/liuyueyi/sp…

源碼：github.com/liuyueyi/sp…

以上就是SpringBoot 容器刷新前回調(diào)ApplicationContextInitializer的詳細(xì)內(nèi)容，更多關(guān)于SpringBoot ApplicationContextInitializer的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論