SpringBoot 动态过滤自动配置类

太过爱你忘了你带给我的痛 2023-10-10 13:46 61阅读 0赞

问题描述：有时候我们引入SpringBoot的Start组件的时候，默认就根据SPI机制加载了相应的自动配置的类;比如通过META-INF下的spring.factories，来读取你需要默认引入的自动配置的类；但是有时候你不想要这些功能，简单一点来说就是在启动类中过滤掉该类

**实现方式一：**

@SpringBootApplication(exclude = {KafkaConsumerConfig.class})

功能说明:启动容器时排除指定配置类。

**实现方式二：**

功能要求：通过一个**开关配置或者自定义规则来决定是否加载该类**。

如何实现：SpringBoot中有一个针对SPI加载的类过滤接口名字`AutoConfigurationImportFilter`  
这个类会在启动的时候通过SPI的机制回调，然后执行它的match方法来绝对目前SPI加载的类是否会被过滤掉；

功能实战：

import org.springframework.boot.autoconfigure.AutoConfigurationImportFilter;
    import org.springframework.boot.autoconfigure.AutoConfigurationMetadata;
    import org.springframework.context.EnvironmentAware;
    import org.springframework.core.env.Environment;
    import org.springframework.stereotype.Component;
    
    import java.util.*;
    
    @Component
    public class KafkaAutoConfigurationImportFilter implements AutoConfigurationImportFilter, EnvironmentAware {
    
        private Environment environment;
        /**
         * 过滤的map
         * K : 代表配置文件中的属性是否为true，如果为true的话就会过滤掉value中的值
         * V ： 过滤掉configuration的全路径
         */
        private Map<String, String> filterAutoConfigurationMap = new LinkedHashMap<>();
    
        public KafkaAutoConfigurationImportFilter() {
            filterAutoConfigurationMap
                .put("spring.kafka.consumer.multiple-group", "com.elab.core.kafka.config.KafkaConsumerConfig");
        }
    
        @Override
        public void setEnvironment(Environment environment) {
            this.environment = environment;
        }
    
        @Override
        public boolean[] match(String[] autoConfigurationClasses, AutoConfigurationMetadata autoConfigurationMetadata) {
    
            Set<String> filterClass = new HashSet<>();
            
            // 获取配置文件中的属性名称的值，并决定是否过滤
            filterAutoConfigurationMap.forEach((k, v) -> {
                Boolean enableProperty = this.environment.getProperty(k, Boolean.class);
                if (enableProperty) {
                    filterClass.add(v);
                }
            });
    
            if (filterClass.size() > 0) {
                boolean[] skip = new boolean[autoConfigurationClasses.length];
                for (int i = 0; i < autoConfigurationClasses.length; i++) {
                    String autoConfigurationClass = autoConfigurationClasses[i];
                    if (!filterClass.contains(autoConfigurationClass)) {
                        skip[i] = true;
                    }
                }
                return skip;
            }
    
            return new boolean[0];
        }
    }

功能代码说明:根据配置文件中的变量:spring.kafka.consumer.multiple-group得到的值是**true**还是**false**决定是否过滤掉com.elab.core.kafka.config.KafkaConsumerConfig类;

注意：在启动的时候引入它，在项目resouce文件新建META-INF文件夹在其中新建一个spring.factories类然后加入内容:

org.springframework.boot.autoconfigure.AutoConfigurationImportFilter=com.elab.data.route.config.KafkaAutoConfigurationImportFilter

启动的时候就会回调该类，基于类中定义的规则来决定是否加载SPI中默认引用的类了；

**想要了解运行流程可以查看**`org.springframework.boot.autoconfigure.AutoConfigurationImportSelector#filter`