springboot中通过@PropertySource加载自定义yml文件

最新推荐文章于 2024-06-25 08:33:28 发布

mobile18600007978

最新推荐文章于 2024-06-25 08:33:28 发布

阅读量4.5k

点赞数 2

分类专栏： spring,springboot

本文链接：https://blog.csdn.net/mobile18611667978/article/details/104439706

版权

spring,springboot 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

使用@PropertySource默认加载的是.xml或者 .properties文件，因为在注解源码默认使用的是DefaultPropertySourceFactory实现处理文件内容，spring
使用ResourcePropertySource从Resource构建Properties传给Spring。
在这里插入图片描述

系统的应用，比如加载自定义的文件，将配置文件内容存储在内存，如下：
在这里插入图片描述

那么加载一个自定义的.yml文件，就需要自定义实现ResourcePropertySource来处理yml文件的类，
在这里插入图片描述

public class YamlPropertySourceFactory implements PropertySourceFactory {
    @Override
    public PropertySource<?> createPropertySource(String name, EncodedResource resource) throws IOException {
        Properties propertiesFromYaml = loadYamlIntoProperties(resource);
        String sourceName = name != null ? name : resource.getResource().getFilename();
        return new PropertiesPropertySource(sourceName, propertiesFromYaml);
    }

    private Properties loadYamlIntoProperties(EncodedResource resource) throws FileNotFoundException {
        try {
            YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
            factory.setResources(resource.getResource());
            factory.afterPropertiesSet();
            return factory.getObject();
        } catch (IllegalStateException e) {
            // for ignoreResourceNotFound
            Throwable cause = e.getCause();
            if (cause instanceof FileNotFoundException)
                throw (FileNotFoundException) e.getCause();
            throw e;
        }
    }
}