讀取自定義yml文件與屬性

在網上找了很多種讀取自定義yml文件與屬性的東西,大部分都要定義實體類字段與配置文件中屬性對應,新加屬性或編輯自定義屬性名會很麻煩,下面的方法只需要使用注解@value即可

1.首先寫配置文件,用來讀取yml文件 2.

package com.gd.ctripbusiness.config;

import org.springframework.beans.factory.config.YamlPropertiesFactoryBean;
import org.springframework.core.env.PropertiesPropertySource;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.support.EncodedResource;
import org.springframework.core.io.support.PropertySourceFactory;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.Properties;


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) {
            Throwable cause = e.getCause();
            if (cause instanceof FileNotFoundException) {
                throw (FileNotFoundException) e.getCause();
            }
            throw e;
        }
    }


}

2.啟動類加上自定義的的yml文件

 

 

@PropertySource(factory = YamlPropertySourceFactory.class, value = "classpath:xxx.yml")

 

3.在使用的時候用@value注解即可

 

 

 

 @Value("${ccbb.des.a}")
    public String as1;

 

參考文章: https://blog.csdn.net/zxl8899/article/details/106382719