一个自动配置的实现

最新推荐文章于 2024-04-06 21:11:11 发布

弱冠季首

最新推荐文章于 2024-04-06 21:11:11 发布

阅读量522

点赞数

分类专栏： JAVA

本文链接：https://blog.csdn.net/ruoguan_jishou/article/details/100560076

版权

本文介绍了Spring中@Configuration注解的用法，用于定义配置类替代XML配置。配置类需遵循特定规则，如不能为final或匿名类。同时，通过实现ApplicationContextAware接口和SmartInitializingSingleton接口，可以实现在Spring容器初始化完成后自动获取ApplicationContext或执行特定操作。文章还简述了实现过程，包括afterSingletonsInstantiated方法的回调和bean的注册。

摘要由CSDN通过智能技术生成

原理

@Configuration
从Spring3.0，@Configuration用于定义配置类，可替换xml配置文件，被注解的类内部包含有一个或多个被@Bean注解的方法，这些方法将会被AnnotationConfigApplicationContext或AnnotationConfigWebApplicationContext类进行扫描，并用于构建bean定义，初始化Spring容器。

注意：
@Configuration注解的配置类有如下要求：
@Configuration不可以是final类型；
@Configuration不可以是匿名类；
嵌套的configuration必须是静态类。

ApplicationContextAware
在我们需要使用ApplicationContext的服务中实现ApplicationContextAware接口,系统启动时就可以自动给我们的服务注入applicationContext对象,我们就可以获取到ApplicationContext里的所有信息了。

SmartInitializingSingleton
当所有单例 bean都初始化完成以后,容器会回调SmartInitializingSingleton接口的 afterSingletonsInstantiated方法。以此实现在所有单例 bean 创建完成之后，可以在该回调中做一些事情。

实现

1.实现afterSingletonsInstantiated

@Override
public void afterSingletonsInstantiated() {
    // 1.获取所有使用RocketMQMessageListener注解的bean
    Map<String, Object> beans = this.applicationContext.getBeansWithAnnotation(ZMQMessageListener.class);

    if (Objects.nonNull(beans)) {
        // 2.注册bean(RocketMqListener容器)
        beans.forEach(this::registerContainer);
    }
}

2.注册容器

/**
 * Register the listener to container
 *
 * @param beanName
 * @param bean
 */
private void registerContainer(String beanName, Object bean) {
    Class<?> clazz = AopProxyUtils.ultimateTargetClass(bean);

    // 检查是否实现了RocketMQListener的方法
    if (!ZMQListener.class.isAssignableFrom(bean.getClass())) {
        throw new IllegalStateException(clazz + " is not instance of " + ZMQListener.class.getName());
    }

    ZMQMessageListener annotation = clazz.getAnnotation(ZMQMessageListener.class);

    String containerBeanName = String.format("%s_%s", DefaultRocketMQListenerContainer.class.getName(),
            counter.incrementAndGet());
    /**
     * Generic ApplicationContext 持有一个DefaultListableBeanFactory实例，并且没有假设一个特定的bean definition 的format。
     * 实现了BeanDefinitionRegistry接口以允许配置任何bean definition reader（也可以不是XmlBeanDefinitionReader）。
     */
    GenericApplicationContext genericApplicationContext = (GenericApplicationContext) applicationContext;
    // 注册bean 
    // 低版本的spring版本智能注册BeanDefinition，高版本的可以用registerBean方法直接注册bean
    genericApplicationContext.registerBeanDefinition(containerBeanName,
            createRocketMQListenerContainer(containerBeanName, bean, annotation).getBeanDefinition());

    // 注册完成