spring自定义加载属性文件(yml、yaml、properties)

直接上代码

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.support.PropertySourcesPlaceholderConfigurer;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.FileSystemResource;
import org.springframework.util.CollectionUtils;

import java.io.File;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * 简述： 属性文件配置
 *
 * @author wenlong
 */
@Slf4j
@Configuration
public class PropertiesConfig {

    @Bean
    public PropertySourcesPlaceholderConfigurer properties(){
        PropertySourcesPlaceholderConfigurer configurer = new PropertySourcesPlaceholderConfigurer();
        configurer.setIgnoreUnresolvablePlaceholders(true);

        //处理classPath中的属性文件
        dealClassPathProperties(configurer);

        //处理自定义属性文件
        dealCustomProperties(configurer);
        return configurer;
    }

    /**
     * 处理自定义属性文件
     * @param configurer
     */
    private void dealCustomProperties(PropertySourcesPlaceholderConfigurer configurer) {
        String path = getPath();
        if(StringUtils.isBlank(path)){
            return;
        }

        File file = new File(path);
        if(file.exists() && file.isDirectory()){
            List<String> files = Arrays.asList(file.list());
            Set<String> ymlSet = new HashSet<>();
            Set<String> yamlSet = new HashSet<>();
            Set<String> propertiesSet = new HashSet<>();
            if(!CollectionUtils.isEmpty(files)){
                for(String s : files) {
                   if(s.endsWith(".yml")){
                       ymlSet.add(s);
                   }else if(s.endsWith(".yaml")){
                       yamlSet.add(s);
                   }else if(s.endsWith(".properties")){
                       propertiesSet.add(s);
                   }
                }

                //处理自定义属性文件
                dealCustomProperties(configurer, path, ymlSet);
                dealCustomProperties(configurer, path, yamlSet);
                dealCustomProperties(configurer, path, propertiesSet);
            }
        }else{
            log.warn("自定义属性配置文件目录{}不存在！", path);
        }
    }

    /**
     * 处理classPath中的属性文件
     * @param configurer
     */
    private void dealClassPathProperties(PropertySourcesPlaceholderConfigurer configurer) {
        ClassPathResource yml = new ClassPathResource("application.yml");
        if(yml.exists()){
            configurer.setLocations(yml);
        }

        ClassPathResource yaml = new ClassPathResource("application.yaml");
        if(yaml.exists()){
            configurer.setLocations(yaml);
        }

        ClassPathResource properties = new ClassPathResource("application.properties");
        if(properties.exists()){
            configurer.setLocations(properties);
        }
    }

    /**
     * 处理yml及yaml类型的属性文件
     * @param configurer
     * @param path
     * @param ymlSet
     */
    private void dealCustomProperties(PropertySourcesPlaceholderConfigurer configurer, String path, Set<String> ymlSet) {
        if (!CollectionUtils.isEmpty(ymlSet)) {
            ymlSet.forEach(s -> {
                FileSystemResource propertiesResource = new FileSystemResource(path.concat(File.separator).concat(s));
                configurer.setLocations(propertiesResource);
            });
        }
    }

    /**
     * 获取自定义属性配置目录
     * @return
     */
    private String getPath() {
        String path = this.getClass().getProtectionDomain().getCodeSource().getLocation().getPath();
        if (System.getProperty("os.name").contains("dows")) {
            path = path.substring(1);
        }
        if (path.contains("jar")) {
            path = path.substring(0, path.lastIndexOf("/"));
            path = path.substring(0, path.lastIndexOf("/"));
            path = path.substring(0, path.lastIndexOf("/"));
        }else{
            return "";
        }
        return path.concat(File.separator).concat("config");
    }
}