springboot项目自定义读取多环境yml配置

有一些支付相关的配置文件是在不想写到application里的。

不说原理了，网上一大堆。

 

1：直接上代码：

import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.env.EnvironmentPostProcessor;
import org.springframework.boot.env.PropertySourceLoader;
import org.springframework.boot.env.YamlPropertySourceLoader;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ResourceUtils;

import java.io.IOException;
import java.util.List;

/**
 * @version V1.0
 * @description: 读取自定义的payment-*.yml的环境处理器（只读yml文件，其他文件忽略）
 * @author: SunF
 * @create: 2019-08-21 09:59
 **/
@Slf4j
public class PaymentEnvironmentPostProcessor implements EnvironmentPostProcessor {
    private final ResourcePatternResolver patternResolver = new PathMatchingResourcePatternResolver();

    private final PropertySourceLoader propertySourceLoader;

    public PaymentEnvironmentPostProcessor() {
        super();
        propertySourceLoader =  new YamlPropertySourceLoader();
    }

    @Override
    public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
        String[] activeProfiles = environment.getActiveProfiles();
        for (String activeProfile: activeProfiles){
            String location = ResourceUtils.CLASSPATH_URL_PREFIX  + "paymentConfig/payment-"+activeProfile+ ".yml";
            try{
                Resource[] resourceArray = this.patternResolver.getResources(location);
                for(Resource resource : resourceArray){
                    List<PropertySource<?>> propertySourceList = propertySourceLoader.load(resource.getFilename(), resource);
                    if(! CollectionUtils.isEmpty(propertySourceList)){
                        propertySourceList.stream().forEach(environment.getPropertySources() :: addLast);
                    }
                }
            }catch(IOException e){
                log.error("加载支付配置文件失败，因为{}", e.getMessage(), e);
            }
        }
    }

}

 

这里只读了yml文件，我们不需要其他文件，如果采用多种PropertySourceLoader进行扫描加载无疑增加了不必要的启动时间。

如果有需要可以将propertySourceLoader改为List,后续的环节在for循环中处理：

    private final List<PropertySourceLoader> propertySourceLoaderList;

    public PaymentEnvironmentPostProcessor() {
        super();
        this.propertySourceLoaderList = SpringFactoriesLoader.loadFactories(PropertySourceLoader.class,
                getClass().getClassLoader());
    }

 

2：添加spring.factories
在resources下添加META-INF文件夹，在旗下创建spring.factories文件，内容为：

org.springframework.boot.env.EnvironmentPostProcessor=com.xxx.xxx.config.PaymentEnvironmentPostProcessor

 

转载于:https://my.oschina.net/dlam/blog/3095443