【Spring注解驱动开发】自定义TypeFilter指定@ComponentScan注解的过滤规则

写在前面

Spring的强大之处不单单是提供了IOC容器，可以经过过滤规则指定排除和只包含哪些组件，它还可以经过自定义TypeFilter来指定过滤规则。若是Spring内置的过滤规则不可以知足咱们的需求，那么咱们就能够经过自定义TypeFilter来实现咱们本身的过滤规则。

项目工程源码已经提交到GitHub：https://github.com/sunshinelyz/spring-annotation

FilterType中经常使用的规则

在使用@ComponentScan注解实现包扫描时，咱们可使用@Filter指定过滤规则，在@Filter中，经过type指定过滤的类型。而@Filter注解的type属性是一个FilterType枚举，以下所示。

package org.springframework.context.annotation;

public enum FilterType {
	ANNOTATION,
	ASSIGNABLE_TYPE,
	ASPECTJ,
	REGEX,
	CUSTOM
}

每一个枚举值的含义以下所示。

（1）ANNOTATION：按照注解进行过滤。

例如，使用@ComponentScan注解进行包扫描时，按照注解只包含标注了@Controller注解的组件，以下所示。

@ComponentScan(value = "io.mykit.spring", includeFilters = {
    @Filter(type = FilterType.ANNOTATION, classes = {Controller.class})
}, useDefaultFilters = false)

（2）ASSIGNABLE_TYPE：按照给定的类型进行过滤。

例如，使用@ComponentScan注解进行包扫描时，按照给定的类型只包含PersonService类（接口）或其子类（实现类或子接口）的组件，以下所示。

@ComponentScan(value = "io.mykit.spring", includeFilters = {
    @Filter(type = FilterType.ASSIGNABLE_TYPE, classes = {PersonService.class})
}, useDefaultFilters = false)

此时，只要是PersonService类型的组件，都会被加载到容器中。也就是说：当PersonService是一个Java类时，Person类及其子类都会被加载到Spring容器中；当PersonService是一个接口时，其子接口或实现类都会被加载到Spring容器中。

（3）ASPECTJ：按照ASPECTJ表达式进行过滤

例如，使用@ComponentScan注解进行包扫描时，按照ASPECTJ表达式进行过滤，以下所示。

@ComponentScan(value = "io.mykit.spring", includeFilters = {
    @Filter(type = FilterType.ASPECTJ, classes = {AspectJTypeFilter.class})
}, useDefaultFilters = false)

（4）REGEX：按照正则表达式进行过滤

例如，使用@ComponentScan注解进行包扫描时，按照正则表达式进行过滤，以下所示。

@ComponentScan(value = "io.mykit.spring", includeFilters = {
    @Filter(type = FilterType.REGEX, classes = {RegexPatternTypeFilter.class})
}, useDefaultFilters = false)

（5）CUSTOM：按照自定义规则进行过滤。

若是实现自定义规则进行过滤时，自定义规则的类必须是org.springframework.core.type.filter.TypeFilter接口的实现类。

例如，按照自定义规则进行过滤，首先，咱们须要建立一个org.springframework.core.type.filter.TypeFilter接口的实现类MyTypeFilter，以下所示。

public class MyTypeFilter implements TypeFilter {
    @Override
    public boolean match(MetadataReader metadataReader, MetadataReaderFactory metadataReaderFactory) throws IOException {
        return false;
    }
}

当咱们实现TypeFilter接口时，须要实现TypeFilter接口中的match()方法，match()方法的返回值为boolean类型。当返回true时，表示符合规则，会包含在Spring容器中；当返回false时，表示不符合规则，不会包含在Spring容器中。另外，在match()方法中存在两个参数，分别为MetadataReader类型的参数和MetadataReaderFactory类型的参数，含义分别以下所示。

• metadataReader：读取到的当前正在扫描的类的信息。
• metadataReaderFactory：能够获取到其余任务类的信息。

接下来，使用@ComponentScan注解进行以下配置。

@ComponentScan(value = "io.mykit.spring", includeFilters = {
    @Filter(type = FilterType.CUSTOM, classes = {MyTypeFilter.class})
}, useDefaultFilters = false)

在FilterType枚举中，ANNOTATION和ASSIGNABLE_TYPE是比较经常使用的，ASPECTJ和REGEX不太经常使用，若是FilterType枚举中的类型没法知足咱们的需求时，咱们也能够经过实现org.springframework.core.type.filter.TypeFilter接口来自定义过滤规则，此时，将@Filter中的type属性设置为FilterType.CUSTOM，classes属性设置为自定义规则的类对应的Class对象。

实现自定义过滤规则

在项目的io.mykit.spring.plugins.register.filter包下新建MyTypeFilter，并实现org.springframework.core.type.filter.TypeFilter接口。此时，咱们先在MyTypeFilter类中打印出当前正在扫描的类名，以下所示。

package io.mykit.spring.plugins.register.filter;

import org.springframework.core.io.Resource;
import org.springframework.core.type.AnnotationMetadata;
import org.springframework.core.type.ClassMetadata;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.core.type.filter.TypeFilter;

import java.io.IOException;

/**
 * @author binghe
 * @version 1.0.0
 * @description 自定义过滤规则
 */
public class MyTypeFilter implements TypeFilter {
    /**
     * metadataReader：读取到的当前正在扫描的类的信息
     * metadataReaderFactory：能够获取到其余任务类的信息
     */
    @Override
    public boolean match(MetadataReader metadataReader, MetadataReaderFactory metadataReaderFactory) throws IOException {
        //获取当前类注解的信息
        AnnotationMetadata annotationMetadata = metadataReader.getAnnotationMetadata();
        //获取当前正在扫描的类的信息
        ClassMetadata classMetadata = metadataReader.getClassMetadata();
        //获取当前类的资源信息，例如：类的路径等信息
        Resource resource = metadataReader.getResource();
        //获取当前正在扫描的类名
        String className = classMetadata.getClassName();
        //打印当前正在扫描的类名
        System.out.println("-----> " + className);
        return false;
    }
}

接下来，咱们在PersonConfig类中配置自定义过滤规则，以下所示。

@Configuration
@ComponentScans(value = {
        @ComponentScan(value = "io.mykit.spring", includeFilters = {
                @Filter(type = FilterType.CUSTOM, classes = {MyTypeFilter.class})
        }, useDefaultFilters = false)
})
public class PersonConfig {

    @Bean("person")
    public Person person01(){
        return new Person("binghe001", 18);
    }
}

接下来，咱们运行SpringBeanTest类中的testComponentScanByAnnotation()方法进行测试，输出的结果信息以下所示。

-----> io.mykit.spring.test.SpringBeanTest
-----> io.mykit.spring.bean.Person
-----> io.mykit.spring.plugins.register.controller.PersonController
-----> io.mykit.spring.plugins.register.dao.PersonDao
-----> io.mykit.spring.plugins.register.filter.MyTypeFilter
-----> io.mykit.spring.plugins.register.service.PersonService
org.springframework.context.annotation.internalConfigurationAnnotationProcessor
org.springframework.context.annotation.internalAutowiredAnnotationProcessor
org.springframework.context.annotation.internalCommonAnnotationProcessor
org.springframework.context.event.internalEventListenerProcessor
org.springframework.context.event.internalEventListenerFactory
personConfig
person

能够看到，已经输出了当前正在扫描的类的名称，同时，除了Spring内置的bean名称外，只输出了personConfig和person，没有输出使用@Repository、@Service、@Controller注解标注的组件名称。这是由于当前PersonConfig上标注的@ComponentScan注解是使用自定义的规则，而在MyTypeFilter自定义规则的实现类中，直接返回了false值，将全部的bean都排除了。

咱们能够在MyTypeFilter类中简单的实现一个规则，例如，当前扫描的类名称中包含有字符串Person，就返回true，不然返回false。此时，MyTypeFilter类中match()方法的实现以下所示。

@Override
    public boolean match(MetadataReader metadataReader, MetadataReaderFactory metadataReaderFactory) throws IOException {
        //获取当前类注解的信息
        AnnotationMetadata annotationMetadata = metadataReader.getAnnotationMetadata();
        //获取当前正在扫描的类的信息
        ClassMetadata classMetadata = metadataReader.getClassMetadata();
        //获取当前类的资源信息，例如：类的路径等信息
        Resource resource = metadataReader.getResource();
        //获取当前正在扫描的类名
        String className = classMetadata.getClassName();
        //打印当前正在扫描的类名
        System.out.println("-----> " + className);
        return className.contains("Person");
    }

此时，在io.mykit.spring包下的全部类都会经过MyTypeFilter类的match()方法，来验证类名是否包含Person，若是包含则返回true，不然返回false。

咱们再次运行SpringBeanTest类中的testComponentScanByAnnotation()方法进行测试，输出的结果信息以下所示。

-----> io.mykit.spring.test.SpringBeanTest
-----> io.mykit.spring.bean.Person
-----> io.mykit.spring.plugins.register.controller.PersonController
-----> io.mykit.spring.plugins.register.dao.PersonDao
-----> io.mykit.spring.plugins.register.filter.MyTypeFilter
-----> io.mykit.spring.plugins.register.service.PersonService
org.springframework.context.annotation.internalConfigurationAnnotationProcessor
org.springframework.context.annotation.internalAutowiredAnnotationProcessor
org.springframework.context.annotation.internalCommonAnnotationProcessor
org.springframework.context.event.internalEventListenerProcessor
org.springframework.context.event.internalEventListenerFactory
personConfig
person
personController
personDao
personService

此时，结果信息中输出了使用@Repository、@Service、@Controller注解标注的组件名称，分别为：personDao、personService和personController。

好了，我们今天就聊到这儿吧！别忘了给个在看和转发，让更多的人看到，一块儿学习一块儿进步！！

项目工程源码已经提交到GitHub：https://github.com/sunshinelyz/spring-annotation

写在最后

若是以为文章对你有点帮助，请微信搜索并关注「 冰河技术 」微信公众号，跟冰河学习Spring注解驱动开发。公众号回复“spring注解”关键字，领取Spring注解驱动开发核心知识图，让Spring注解驱动开发再也不迷茫。