快捷導(dǎo)航

SpringBoot如何啟動自動加載自定義模塊yml文件(PropertySourceFactory)

更新時間：2024年07月10日 09:00:27 作者：君哥聊編程

這篇文章主要介紹了SpringBoot如何啟動自動加載自定義模塊yml文件(PropertySourceFactory),具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

背景

自定義模塊時經(jīng)常會編寫當(dāng)前模塊的一些核心yml配置，如果要被Spring容器加載到通常的做法是將自定義模塊的yml命名為application-模塊名.yml，比如db模塊命名為application-db.yml，然后再在spring.profiles.include中引入該db，即可加載到子模塊配置（前提是maven構(gòu)建時能夠?qū)esources資源目錄編譯進去）。

上面說的這種限制有很多，子模塊對于父模塊是黑箱操作，我們無法得知有多少個子模塊要被include進來，而且前綴必須以application開頭。

今天我們來看另外一種更加靈活的方式，自定義名稱的yml直接在子模塊中加載，各位朋友繼續(xù)往下看PropertySourceFactory如何實現(xiàn)。

總結(jié)3步走

定義一個YamlPropertySourceFactory，實現(xiàn)PropertySourceFactory，重寫createPropertySource方法

將接口給過來的資源流文件使用加載到spring提供的YmlPropertiesFactorBean中，由該工廠Bean產(chǎn)出一個Properties對象

將生成的Properties對象傳入PropertiesPropertySource中產(chǎn)出一個PropertySource對象

import lombok.AllArgsConstructor;
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;

/**
 * @description:
 * @author: jianjun.ren
 * @date: Created in 2020/9/24 12:05
 */
@AllArgsConstructor
public class YamlPropertySourceFactory implements PropertySourceFactory {
	//這個方法有2個入?yún)?，分別為資源名稱和資源對象，這是第一步
    public PropertySource< ? > createPropertySource(String name, EncodedResource resource) throws IOException {
    
    	//這是第二步，根據(jù)流對象產(chǎn)出Properties對象
        Properties propertiesFromYaml = loadYamlIntoProperties(resource);
        String sourceName = name != null ? name : resource.getResource().getFilename();
        assert sourceName != null;
        
        //這是第三部，根據(jù)Properties對象，產(chǎn)出PropertySource對象，放入到Spring中
        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;
        }
    }
}

創(chuàng)建咱們是說完了，小伙伴有很多問號，怎么去使用呢？

接下來我們再看康康：

在任意一個@Configuration類中，你自己寫一個也可以，反正能夠被Spring所掃描的

加入下面這段代碼，其實就一行：

@Configuration
//這個地方的YamlPropertySourceFactory就是上面編寫的代碼類
@PropertySource(factory = YamlPropertySourceFactory.class, value = "classpath:xxx-xxx.yml")
public class XXXXConfiguration {
		
}