[轉] Spring Boot 自動配置之@Enable* 與@Import注解

SpringBoot 的自動配置如此強大，比如我們經常使用的@Enable* 注解來開啟對某方面的支持。那么@Enable* 注解的原理是什么呢？

一、@Enable* 注解與 @Import 注解之間的關系

@Enable* 舉例：

@EnableScheduling 開啟計劃任務的支持
@EnableAsync 開啟異步方法的支持
@EnableAspectJAutoProxy 開啟對 AspectJ 代理的支持
@EnableTransactionManagement 開啟對事務的支持
@EnableCaching 開啟對注解式緩存的支持

等等

我們觀察這些@Enable* 的源碼可以看出，所有@Enable* 注解都是有@Import的組合注解，@Enable* 自動開啟的實現其實就是導入了一些自動配置的Bean

看下 Spring Boot Reference Guide原文

You need not put all your @Configuration into a single class. The @Import annotation
can be used to import additional configuration classes.

您不需要把所有的 @Configuration 放到一個類中。@Import 注解可以導入額外的配置類。

@Import 注解的最主要功能就是導入額外的配置信息

二、 @Import 注解的用法

官方介紹：

* <p>Provides functionality equivalent to the {@code <import/>} element in Spring XML.
* Allows for importing {@code @Configuration} classes, {@link ImportSelector} and
* {@link ImportBeanDefinitionRegistrar} implementations, as well as regular component
* classes (as of 4.2; analogous to {@link AnnotationConfigApplicationContext#register}).

有以下三種使用方式

1、直接導入配置類（@Configuration 類）

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Import(SchedulingConfiguration.class)
@Documented
public @interface EnableScheduling {

}

可以看到EnableScheduling注解直接導入配置類SchedulingConfiguration，這個類注解了@Configuration，且注冊了一個scheduledAnnotationProcessor的Bean，SchedulingConfiguration的源碼如下：

@Configuration
@Role(BeanDefinition.ROLE_INFRASTRUCTURE)
public class SchedulingConfiguration {
    @Bean(name = TaskManagementConfigUtils.SCHEDULED_ANNOTATION_PROCESSOR_BEAN_NAME)
    @Role(BeanDefinition.ROLE_INFRASTRUCTURE)
    public ScheduledAnnotationBeanPostProcessor scheduledAnnotationProcessor() {
        return new ScheduledAnnotationBeanPostProcessor();
    }
}

2、依據條件選擇配置類（實現 ImportSelector 接口）

如果并不確定引入哪個配置類，需要根據@Import注解所標識的類或者另一個注解(通常是注解)里的定義信息選擇配置類的話，用這種方式。

ImportSelector接口只有一個方法

String[] selectImports(AnnotationMetadata importingClassMetadata);

AnnotationMetadata：用來獲得當前配置類上的注解

例：

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Import(AsyncConfigurationSelector.class)
public @interface EnableAsync {

    Class<? extends Annotation> annotation() default Annotation.class;
    
    boolean proxyTargetClass() default false;

    AdviceMode mode() default AdviceMode.PROXY;

    int order() default Ordered.LOWEST_PRECEDENCE;

}

AsyncConfigurationSelector繼承AdviceModeImportSelector，AdviceModeImportSelector類實現ImportSelector接口 根據AdviceMode的不同來選擇生明不同的Bean

public class AsyncConfigurationSelector extends AdviceModeImportSelector<EnableAsync> {

    private static final String ASYNC_EXECUTION_ASPECT_CONFIGURATION_CLASS_NAME = "org.springframework.scheduling.aspectj.AspectJAsyncConfiguration";

    @Override
    @Nullable
    public String[] selectImports(AdviceMode adviceMode) {
        switch (adviceMode) {
            case PROXY:
                return new String[] {ProxyAsyncConfiguration.class.getName()};
            case ASPECTJ:
                return new String[] {ASYNC_EXECUTION_ASPECT_CONFIGURATION_CLASS_NAME};
            default:
                return null;
        }
    }
}

3、動態注冊Bean（實現 ImportBeanDefinitionRegistrar 接口）

一般只要用戶確切知道哪些Bean需要放入容器的話，自己可以通過spring 提供的注解來標識就可以了，比如@Component,@Service,@Repository,@Bean等。如果是不確定的類，或者不是spring專用的，所以并不想用spring的注解進行侵入式標識，那么就可以通過@Import注解，實現ImportBeanDefinitionRegistrar接口來動態注冊Bean。比如：

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Import(AspectJAutoProxyRegistrar.class)
public @interface EnableAspectJAutoProxy {

    boolean proxyTargetClass() default false;
    
    boolean exposeProxy() default false;

}

AspectJAutoProxyRegistrar實現了ImportBeanDefinitionRegistrar接口，ImportBeanDefinitionRegistrar的作用是在運行時自動添加Bean到已有的配置類，通過重寫方法：

public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry);

AnnotationMetadata 參數用來獲得當前配置類上的注解
BeanDefinitionRegistry 參數用來注冊Bean

源碼：

@Override
public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
    AopConfigUtils.registerAspectJAnnotationAutoProxyCreatorIfNecessary(registry);

    AnnotationAttributes enableAspectJAutoProxy = AnnotationConfigUtils.attributesFor(importingClassMetadata, EnableAspectJAutoProxy.class);
    if (enableAspectJAutoProxy != null) {
        if (enableAspectJAutoProxy.getBoolean("proxyTargetClass")) {
            AopConfigUtils.forceAutoProxyCreatorToUseClassProxying(registry);
        }
        if (enableAspectJAutoProxy.getBoolean("exposeProxy")) {
            AopConfigUtils.forceAutoProxyCreatorToExposeProxy(registry);
        }
    }
}

Mybatis 中大名鼎鼎的@MapperScan 也是如此

三、官方文檔

官方文檔

轉自：https://juejin.im/post/5c761c096fb9a049b41d2299

posted @ 2020-05-24 20:19 dai.sp 閱讀(506) 評論(0) 收藏舉報

刷新頁面返回頂部

Loading

.x->y=z