SpringBoot之@Enable*注解的工作原理

Z时代
2024-01-10
分类：综合

一、观察任一@Enable*注解的源码，以`@EnableAsync`为例

@EnableAsync源码

@EnableAsync的作用是启用异步执行，使标注@Async注解的方法能够和其他方法异步执行。读者可以Google一下@EnableAsync这个注解的使用场景，本文不再赘述

我们发现，这个注解的重点在我标红的@Import({AsyncConfigurationSelector.class})这段代码。解释一下@Import和XxxSelector.class的作用。

1）`@Import`

用来导入一个或多个class，这些类会注入到spring容器中，或者配置类，配置类里面定义的bean都会被spring容器托管。在这里我们加入的AsyncConfigurationSelector.class放入Spring容器中管理。

2）`AsyncConfigurationSelector.class`

我们从源码一直追溯这个类的父类，最终找到顶端的父类ImportSelector.class

追溯父类ImportSelector.class

打开

ImportSelector.class阅读源码：

ImportSelector.class

Spring会把实现ImportSelector接口的类中的SelectImport方法返回的值注入到Spring容器中。这个方法的返回值必须是一个class的全类名的String[]。举个例子：

publicclassMyImportSelectorimplementsImportSelector{
@Override
publicString[]selectImports(AnnotationMetadata annotationMetadata){
returnnewString[]{"com.springboot.enable.User","com.springboot.enable.Car"};
}
}

spring容器会把com.springboot.enable包下的User和Car这两个类放入容器中。
回归正题，我们不是需要通过@Enable*注解开启一些功能嘛？答案就我自定义的MyImportSelector中。简单来说就是@Enable*会将XxxImportSelector放入容器中，当Spring启动，会执行selectImports(AnnotationMetadata annotationMetadata)方法，在这个方法中我们做了某些处理，使得和@Enable*搭配使用的注解生效。哈哈，是不是很绕，多阅读两遍，你就理解了。

还有一个和ImportSelector功能差不多的类，ImportBeanDefinitionRegistrar使用beanDefinitionRegistry对象将bean加入Spring容器中，源码如下：

ImportBeanDefinitionRegistrar

二、小实验：

下面我们做一个小实验印证一下，下图有三个包，每个包下分别有三个bean，他们都加了@Component注解，会被spring加入到容器中。

beans

User

Bird

Car

需求是，当注入dto和vo两个包下的bean时，输出一段话：echo bean ：+ bean的全类名，注入entity包下的bean时，不输出。

1）

创建EchoBeanPostProcessor.class，实现BeanPostProcessor接口，作用是实现上文的业务逻辑。我们同样可以创建一个@EchoBean，然后通过AOP的方式实现。

//实现BeanPostProcessor接口的类，放入spring容器中后，容器启动和关闭时会执行以下两个重写的方法
publicclassEchoBeanPostProcessorimplementsBeanPostProcessor{
//getter、setter省略，读者在试验的时候要加上
privateList<String> packages;
//该方法在spring容器初始化前执行
@Override
publicObjectpostProcessBeforeInitialization(Object bean,String s)throwsBeansException{
for(String pack : packages){
if(bean.getClass().getName().startsWith(pack)){
System.out.println("echo bean: "+ bean.getClass().getName());
}
}
return bean;
}
@Override
publicObjectpostProcessAfterInitialization(Object bean,String s)throwsBeansException{
return bean;
}
}

2）

创建BamuImportBeanDefinitionRegistrar .class，实现ImportBeanDefinitionRegistrar

publicclassBamuImportBeanDefinitionRegistrarimplementsImportBeanDefinitionRegistrar{
@Override
publicvoidregisterBeanDefinitions(AnnotationMetadata annotationMetadata,BeanDefinitionRegistry beanDefinitionRegistry){
//获取EnableEcho注解的所有属性的value
Map<String,Object> attributes = annotationMetadata.getAnnotationAttributes(EnableEcho.class.getName());
//获取package属性的value
List<String> packages =Arrays.asList((String[]) attributes.get("packages"));
//使用beanDefinitionRegistry对象将EchoBeanPostProcessor注入至Spring容器中
BeanDefinitionBuilder beanDefinitionBuilder =BeanDefinitionBuilder.rootBeanDefinition(EchoBeanPostProcessor.class);
//给EchoBeanPostProcessor.class中注入packages
        beanDefinitionBuilder.addPropertyValue("packages", packages);
        beanDefinitionRegistry.registerBeanDefinition(EchoBeanPostProcessor.class.getName(), beanDefinitionBuilder.getBeanDefinition());
}
}

3）

创建注解@EnableEcho，ImportBamuImportBeanDefinitionRegistrar.class

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Import({BamuImportBeanDefinitionRegistrar.class})
public@interfaceEnableEcho{
//传入包名
String[]packages()default"";
}

4）

在springboot启动类中加入我们创建的注解，并传入指定的包名，执行main方法：

@SpringBootApplication
@EnableEcho(packages ={"com.springboot.vo","com.springboot.dto"})
publicclassBlogApplication{
publicstaticvoidmain(String[] args){
ConfigurableApplicationContext context =SpringApplication.run(BlogApplication.class, args);
        context.close();
}
}