目录
  • 如何在所有Bean创建完后做扩展
    • 方式一
    • 方式二
  • Spring通过initPropertySources扩展方法设置环境配置
    • @Import进行扩展

      如何在所有Bean创建完后做扩展

      方式一

      Spring在容器刷新完成后会注册ContextRefreshedEvent。

      Java中Spring扩展点详解

      所以可以自定义事件监听器监听该事件进行扩展。

      监听器实现：

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

      方式二

      Spring在所有bean注册完成后，会检查bean是否实现了SmartInitializingSingleton接口，如果实现了，会回调改类的afterSingletonsInstantiated()方法，我们可以在方法里实现扩展。

      Java中Spring扩展点详解

      实现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()方法，但是默认容器是未实现这个方法的。我们可以实现该方法进行扩展。

      Java中Spring扩展点详解

      实现自定义扩展容器：

      /**
 * @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");
    }
}

      设置环境变量：

      Java中Spring扩展点详解

      测试类：

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

      结果：

      Java中Spring扩展点详解

      @Import进行扩展

      方式一：实现ImportBeanDefinitionRegistrar接口

      例如开启AOP注解，使用AspectJAutoProxyRegistrar.class

      Java中Spring扩展点详解

      AspectJAutoProxyRegistrar实现了ImportBeanDefinitionRegistrar接口，进行BeadDefinition的注册：

      Java中Spring扩展点详解

      方式二：实现ImportSelector接口

      Java中Spring扩展点详解

      ImportSelector接口的selectImports方法返回的是要注入类的全类名数组。spring会根据全类名注册bean。

      例如：开启事务管理功能就是使用实现ImportSelector接口进行扩展。

      Java中Spring扩展点详解

      看下TransactionManagementConfigurationSelector.class：

      Java中Spring扩展点详解

      声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。