SpringBoot中的@Import注解怎么使用

這篇文章主要介紹了SpringBoot中的@Import注解怎么使用的相關知識，內容詳細易懂，操作簡單快捷，具有一定借鑒價值，相信大家閱讀完這篇SpringBoot中的@Import注解怎么使用文章都會有所收獲，下面我們一起來看看吧。

一、 @Import引入普通類

@Import引入普通的類可以幫助我們把普通的類定義為Bean。@Import可以添加在@SpringBootApplication(啟動類)、@Configuration(配置類)、@Component(組件類)對應的類上。

注意：@RestController、@Service、@Repository都屬于@Component

@SpringBootApplication
@Import(ImportBean.class) // 通過@Import注解把ImportBean添加到IOC容器里面去
public class MyBatisApplication {
    public static void main(String[] args) {
        SpringApplication.run(MyBatisApplication.class, args);
    }
}

二、 @Import引入配置類(@Configuration修飾的類)

@Import除了可以把普通的類定義為Bean，@Import還可以引入一個@Configuration修飾的類(引入配置類)，從而把讓配置類生效(配置類下的所有Bean添加到IOC容器里面去)。在自定義starter的時候經常會用到。

如果配置類在標準的SpringBoot包結構下(SpringBootApplication啟動類包的根目錄下)。是不需要@Import導入配置類的，SpringBoot自動幫做了。上面的情況一般用于@Configuration配置類不在標準的SpringBoot包結構下面。所以一般在自定義starter的時候用到。

@Configuration(proxyBeanMethods = false)
@Import({  // import了兩個哈
        XXXDataConfiguration.XXXPartOneConfiguration.class,
        XXXDataConfiguration.XXXPartTwoConfiguration.class
})
public class XXXDataAutoConfiguration {
}
public class XXXDataConfiguration {
    @Configuration(proxyBeanMethods = false)
    static class XXXPartOneConfiguration {
        @Bean
        @ConditionalOnMissingBean
        public BeanForIoc beanForIoc() {
            return new BeanForIoc();
        }
    }
    @Configuration(proxyBeanMethods = false)
    static class XXXPartTwoConfiguration {

        /**
         * 省略了@Bean的使用
         */
    }
}

三 、@Import引入ImportSelector的實現類

@Import還可以引入ImportSelector的實現類，把ImportSelector接口selectImports()方法返回的Class名稱都定義為bean。注意selectImports()方法的參數AnnotationMetadata，通過這個參數我們可以獲取到@Import標注的Class的各種信息。這一點特別有用，用于做一些參數的傳遞。在SpringBoot的自動化配置和@EnableXXX(功能性注解)都有它的存在。

public interface ImportSelector {
    /**
     * 用于指定需要注冊為bean的Class名稱
     * 當在@Configuration標注的Class上使用@Import引入了一個ImportSelector實現類后，會把實現類中返回的Class名稱都定義為bean。
     *
     * 通過其參數AnnotationMetadata importingClassMetadata可以獲取到@Import標注的Class的各種信息，
     * 包括其Class名稱，實現的接口名稱、父類名稱、添加的其它注解等信息，通過這些額外的信息可以輔助我們選擇需要定義為Spring bean的Class名稱
     */
    String[] selectImports(AnnotationMetadata importingClassMetadata);
}

關于@Import引入ImportSelector的實現類的使用，我們舉幾個簡單的使用場景(實際開發肯定比這復雜的多的)。

3.1 靜態import場景(注入已知的類)

靜態場景(注入已知的類)，很簡單通過實現ImportSelector類把我們需要定義為bean的類直接返回就好， 比如下面的例子。我們來一個EnableXXX的注解，通過XXXConfigurationSelector注入一個已知的類XXX。

/**
 * XXXConfigurationSelector一定要配合@Import使用
 */
public class XXXConfigurationSelector implements ImportSelector {
    @Override
    @NonNull
    public String[] selectImports(@NonNull AnnotationMetadata importingClassMetadata) {
        // 把XXX對應的類，定義為Bean
        return new String[]{XXX.class.getName()};
    }
}
/**
 * 注意 @Import(XXXConfigurationSelector.class)
 */
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Import(XXXConfigurationSelector.class)
public @interface EnableXXX {
}
@SpringBootApplication
@EnableXXX // 使之生效
public class MyBatisApplication {

    public static void main(String[] args) {
        SpringApplication.run(MyBatisApplication.class, args);
    }

}

3.2 動態import場景(注入指定條件的類)

我們做一個這樣的功能，我們需要把指定包路徑下所有實現了HelloService接口的類做為bean添加到IOC容器里面去。@ComponentScan注解用來幫我們指定路徑。具體實現如下：

public interface HelloService {
    void function();

}
public class DynamicSelectImport implements ImportSelector {
    /**
     * DynamicSelectImport需要配合@Import()注解使用
     * <p>
     * 通過其參數AnnotationMetadata importingClassMetadata可以獲取到@Import標注的Class的各種信息，
     * 包括其Class名稱，實現的接口名稱、父類名稱、添加的其它注解等信息，通過這些額外的信息可以輔助我們選擇需要定義為Spring bean的Class名稱
     */
    @Override
    public String[] selectImports(AnnotationMetadata importingClassMetadata) {
        // 第一步：獲取到通過ComponentScan指定的包路徑
        String[] basePackages = null;
        // @Import注解對應的類上的ComponentScan注解
        if (importingClassMetadata.hasAnnotation(ComponentScan.class.getName())) {
            Map<String, Object> annotationAttributes = importingClassMetadata.getAnnotationAttributes(ComponentScan.class.getName());
            basePackages = (String[]) annotationAttributes.get("basePackages");
        }
        if (basePackages == null || basePackages.length == 0) {
            //ComponentScan的basePackages默認為空數組
            String basePackage = null;
            try {
                // @Import注解對應的類的包名
                basePackage = Class.forName(importingClassMetadata.getClassName()).getPackage().getName();
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            }
            basePackages = new String[]{basePackage};
        }
        // 第er步，知道指定包路徑下所有實現了HelloService接口的類（ClassPathScanningCandidateComponentProvider的使用）
        ClassPathScanningCandidateComponentProvider scanner = new ClassPathScanningCandidateComponentProvider(false);
        TypeFilter helloServiceFilter = new AssignableTypeFilter(HelloService.class);
        scanner.addIncludeFilter(helloServiceFilter);
        Set<String> classes = new HashSet<>();
        for (String basePackage : basePackages) {
            scanner.findCandidateComponents(basePackage).forEach(beanDefinition -> classes.add(beanDefinition.getBeanClassName()));
        }
        // 第三步，返回添加到IOC容器里面去
        return classes.toArray(new String[0]);
    }
}
@Configuration
@ComponentScan("com.tuacy.collect.mybatis") // 指定路徑
@Import(DynamicSelectImport.class)
public class DynamicSelectConfig {
}

四、 @Import引入ImportBeanDefinitionRegistrar的實現類

@Import引入ImportBeanDefinitionRegistrar的實現類。一般用來動態注冊bean。最重要的一點是還可以對這些BeanDefinition進行額外的修改或增強。咱們經常使用的mybatis @MapperScan就是用這種方式實現的。

/**
 * ImportBeanDefinitionRegistrar，我們一般會實現ImportBeanDefinitionRegistrar類，然后配合一個自定義的注解一起使用。而且在注解類上@Import我們的這個實現類。
 * 通過自定義注解的配置，拿到注解的一些元數據。然后在ImportBeanDefinitionRegistrar的實現類里面做相應的邏輯處理，比如把自定義注解標記的類添加到Spring IOC容器里面去。
 */
public interface ImportBeanDefinitionRegistrar {

    /**
     * 根據注解的給定注釋元數據，根據需要注冊bean定義
     * @param importingClassMetadata 可以拿到@Import的這個class的Annotation Metadata
     * @param registry BeanDefinitionRegistry 就可以拿到目前所有注冊的BeanDefinition，然后可以對這些BeanDefinition進行額外的修改或增強。
     */
    void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry);

}

關于@Import引入ImportBeanDefinitionRegistrar的使用強烈建議大家去看看mybatis關于@MapperScan的處理源碼。特別有意思。我們也舉一個非常簡單的實例，來讓大家直觀的看到ImportBeanDefinitionRegistrar的使用，比如我們想把指定包路徑下所有添加了BeanIoc注解的類注冊為bean。

具體實現如下：

/**
 * 我們會把添加了該注解的類作為bean
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
@Documented
public @interface BeanIoc {

}

/**
 * 定義包路徑。(指定包下所有添加了BeanIoc注解的類作為bean)
 * 注意這里 @Import(BeanIocScannerRegister.class) 的使用
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
@Documented
@Import(BeanIocScannerRegister.class)
public @interface BeanIocScan {
    String[] basePackages() default "";
}


/**
 * 搜索指定包下所有添加了BeanIoc注解的類，并且把這些類添加到ioc容器里面去
 */
public class BeanIocScannerRegister implements ImportBeanDefinitionRegistrar, ResourceLoaderAware {

    private final static String PACKAGE_NAME_KEY = "basePackages";

    private ResourceLoader resourceLoader;

    @Override
    public void registerBeanDefinitions(AnnotationMetadata annotationMetadata, BeanDefinitionRegistry beanDefinitionRegistry) {
        //1. 從BeanIocScan注解獲取到我們要搜索的包路徑
        AnnotationAttributes annoAttrs = AnnotationAttributes.fromMap(annotationMetadata.getAnnotationAttributes(BeanIocScan.class.getName()));
        if (annoAttrs == null || annoAttrs.isEmpty()) {
            return;
        }
        String[] basePackages = (String[]) annoAttrs.get(PACKAGE_NAME_KEY);
        // 2. 找到指定包路徑下所有添加了BeanIoc注解的類，并且把這些類添加到IOC容器里面去
        ClassPathBeanDefinitionScanner scanner = new ClassPathBeanDefinitionScanner(beanDefinitionRegistry, false);
        scanner.setResourceLoader(resourceLoader);
        scanner.addIncludeFilter(new AnnotationTypeFilter(BeanIoc.class));
        scanner.scan(basePackages);
    }

    @Override
    public void setResourceLoader(ResourceLoader resourceLoader) {
        this.resourceLoader = resourceLoader;
    }
}


/**
 * 使用，使BeanIocScan生效
 */
@Configuration
@BeanIocScan(basePackages = "com.tuacy.collect.mybatis")
public class BeanIocScanConfig {
}

關于“SpringBoot中的@Import注解怎么使用”這篇文章的內容就介紹到這里，感謝各位的閱讀！相信大家對“SpringBoot中的@Import注解怎么使用”知識都有一定的了解，大家如果還想學習更多知識，歡迎關注億速云行業資訊頻道。