前景需求
通过加载yml文件的内容并设置为系统变量,在xml文件中引入该变量,而不是在xml文件直接赋值,便于在yml进行统一管理。如果需要保证敏感内容不容易曝光,可对重要内容进行加解密。
加载yml
Spring Framework 提供了两个方式获取yml内容:YamlMapFactoryBean和YamlPropertiesFactoryBean
两者的区别是返回的结果对象不同,两者各有优劣势,根据实际生产需要选择。
- YamlMapFactoryBean将yml加载为Map对象
- YamlPropertiesFactoryBean将yml加载为Properties对象
设置系统环境变量
通过上述两个方式读取到yml的内容,然后通过System.setProperty()方法来设置系统环境变量;
打印系统变量和环境变量
//打印系统变量和环境变量
Properties properties = System.getProperties();
for (Object o : properties.keySet()) {
System.out.println("系统变量: " + String.valueOf(o) + "-" + properties.get(o));
}
Map<String, String> envs = System.getenv();
for (String key : envs.keySet()) {
System.out.println("环境变量:" + key + "->" + envs.get(key));
}
示例代码
// 示例代码
private static void buildYmlSystemEnv(String fileName) {
YamlPropertiesFactoryBean yml = new YamlPropertiesFactoryBean();
yml.setResources(new ClassPathResource(fileName));
Objects.requireNonNull(yml.getObject()).keySet().forEach(e -> {
System.setProperty((String) e, (String) yml.getObject().get(e));
});
}