Java中Spring擴展點詳解

更新時間：2022年06月16日 09:15:00 作者：hi wei

這篇文章主要介紹了Java中Spring技巧之擴展點的應用，下文Spring容器的啟動流程圖展開其內容的相關資料，具有一定的參考價值，需要的小伙伴可以參考一下

如何在所有Bean創(chuàng)建完后做擴展

方式一

Spring在容器刷新完成后會注冊ContextRefreshedEvent。

所以可以自定義事件監(jiān)聽器監(jiān)聽該事件進行擴展。

監(jiān)聽器實現(xiàn)：

@Component
public class ContextRefreshedEventListener implements ApplicationListener<ContextRefreshedEvent> {
    @Override
    public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {
        System.out.println("容器初始化完成,開始進行擴展！");
    }
}

方式二

Spring在所有bean注冊完成后，會檢查bean是否實現(xiàn)了SmartInitializingSingleton接口，如果實現(xiàn)了，會回調改類的afterSingletonsInstantiated()方法，我們可以在方法里實現(xiàn)擴展。

實現(xiàn)SmartInitializingSingleton接口：

/**
 * @author zhw
 * @description
 * @date 2021-09-29 15:28
 */
@Component
public class SmartInitializingSingletonTest implements SmartInitializingSingleton {
    @Override
    public void afterSingletonsInstantiated() {
        System.out.println("所有單例bean注冊完成，開始擴展！");
    }
}

Spring通過initPropertySources擴展方法設置環(huán)境配置

Spring的prepareRefresh()方法中有initPropertySources()方法，但是默認容器是未實現(xiàn)這個方法的。我們可以實現(xiàn)該方法進行擴展。

實現(xiàn)自定義擴展容器：

/**
 * @author zhw
 * @description
 * @date 2021-09-29 16:05
 */
public class ExtensionContext extends AnnotationConfigApplicationContext {
    public ExtensionContext(Class<MainConfig> mainConfigClass) {
        super(mainConfigClass);
    }
    @Override
    protected void initPropertySources() {
        //設置一些必須的環(huán)境變量
        getEnvironment().setRequiredProperties("appName");
    }
}

設置環(huán)境變量：

測試類：

public class MyContextTest {
    public static void main(String[] args) {
        AnnotationConfigApplicationContext annotationConfigApplicationContext = new ExtensionContext(MainConfig.class);
    }
}

結果：