工作的时候,一般来说代码都是分环境的,比如dev,test,prd什么的,在用到@PropertySource 注解的时候,发现好像不能根据环境读取自定义的.properties文件,比如我有个systemProperties-dev.properties
文件,一开始只是systemProperties-${spring.profiles.active}.properties
这样的方式勉强能用,但是后来当我的环境变量变成多环境的时候,也就是spring.profiles.active = dev,test
这样的是,这个方法就不奏效了,(多傻啊,其实早就想到了,他会直接在“-”后面拼了一个“dev,test”)然后在网上看了看资料,参考了以下的一篇文章,然后参照了下源码,用了一个比较简单,但是很难看的方法实现了:P(感觉也是暂时解决问题。)。
参照文章:Springboot中PropertySource注解多环境支持以及原理
主要思想,重写PropertySourceFactory,在PropertySourceFactory中,重新取得resource,
SystemProperties.java
@Component
@PropertySource(name="systemConfig", value = {"classpath:/systemConfig-${spring.profiles.active}.properties"}, factory = SystemPropertySourceFactory.class)
public class SystemProperties {
// 自己的内容....
}
这里指定了 factory = SystemPropertySourceFactory.class
,接下来
SystemPropertySourceFactory.java
@Configuration
public class SystemPropertySourceFactory implements PropertySourceFactory {
@Override
public PropertySource<?> createPropertySource(String name, EncodedResource encodedResource) throws IOException {
FileSystemResourceLoader resourceLoader = new FileSystemResourceLoader();
//取得当前活动的环境名称(因为直接获取spring.profiles.active 失败,所以才把环境名称拼在文件名后面来拿)
//其实感觉应该有可以直接取的方法比如从环境里取
String[] actives = encodedResource.getResource().getFilename().split("\\.")[0].replace(name + "-", "").split(",");
//如果只有一个,就直接返回
if (actives.length <= 1) {
return (name != null ? new ResourcePropertySource(name, encodedResource) : new ResourcePropertySource(encodedResource));
}
//如果是多个
List<InputStream> inputStreamList = new ArrayList<>();
String suffix = fileproperty[1];
//遍历后把所有环境的url全部抓取到list中
Arrays.stream(actives).forEach(active -> {
InputStream in = this.getClass().getResourceAsStream("/" + name.concat("-" + active).concat(".").concat(suffix));
if (in != null) {
inputStreamList.add(in);
}
});
if (resourceUrls != null && resourceUrls.size() > 0) {
//串行流,将多个文件流合并车一个流
SequenceInputStream inputStream = new SequenceInputStream(Collections.enumeration(inputStreamList));
//转成resource
InputStreamResource resource = new InputStreamResource(inputStream);
return (name != null ? new ResourcePropertySource(name, new EncodedResource(resource)) : new ResourcePropertySource(new EncodedResource(resource)));
} else {
return (name != null ? new ResourcePropertySource(name, encodedResource) : new ResourcePropertySource(encodedResource));
}
}
}
这样实现后,就能将多个环境的Property文件加载进去了。
然后是关于spring.profiles.active 为什么要这么取,我试过@value,和用Environment 对象,都取不到,可能跟bean创建的先后顺序有关。没有继续调查,希望知道原因的朋友能帮忙解答~
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。