Java Spring扩展点
hi wei 人气:0如何在所有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);
}
}
结果：
@Import进行扩展
方式一：实现ImportBeanDefinitionRegistrar接口
例如开启AOP注解，使用AspectJAutoProxyRegistrar.class
AspectJAutoProxyRegistrar实现了ImportBeanDefinitionRegistrar接口，进行BeadDefinition的注册：
方式二：实现ImportSelector接口
ImportSelector接口的selectImports方法返回的是要注入类的全类名数组。spring会根据全类名注册bean。
例如：开启事务管理功能就是使用实现ImportSelector接口进行扩展。
看下TransactionManagementConfigurationSelector.class：
加载全部内容