Java中Spring擴充套件點詳解

2022-06-16 10:00:46

如何在所有Bean建立完後做擴充套件

方式一

Spring在容器重新整理完成後會註冊ContextRefreshedEvent。

所以可以自定義事件監聽器監聽該事件進行擴充套件。

監聽器實現：

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

方式二

Spring在所有bean註冊完成後，會檢查bean是否實現了SmartInitializingSingleton介面，如果實現了，會回撥改類的afterSingletonsInstantiated()方法，我們可以在方法裡實現擴充套件。

實現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擴充套件方法設定環境設定

Spring的prepareRefresh()方法中有initPropertySources()方法，但是預設容器是未實現這個方法的。我們可以實現該方法進行擴充套件。

實現自定義擴充套件容器：

/**
 * @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() {
        //設定一些必須的環境變數
        getEnvironment().setRequiredProperties("appName");
    }
}

設定環境變數：

測試類：

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

結果：