Springboot中获取配置文件变量的工具类。

YML配置文件

代码如下

import org.springframework.beans.factory.config.YamlPropertiesFactoryBean;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;

import java.util.Properties;

/**
 * @ClassName PropertiesUtil
 * @Author by septzhang
 * @Date 2022/7/25 18:20
 * @Description 获取配置文件变量工具
 * @Version 1.0
 **/
public class PropertiesUtil {
    //配置常量配置文件，默认路径是 /resources/*。yml
    private static String PROPERTY_NAME = "application-common.yml";

    /**
     *  获取参数
     * @param key String参数key值
     * @return String Value
     */
    public static Object getCommonYml(Object key){
        Resource resource = new ClassPathResource(PROPERTY_NAME);
        Properties properties = null;
        try {
            YamlPropertiesFactoryBean yamlFactory = new YamlPropertiesFactoryBean();
            yamlFactory.setResources(resource);
            properties =  yamlFactory.getObject();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
        return properties.get(key);
    }
}

Junit测试

/**
 * @ClassName PropertiesUtilTest
 * @Author by septzhang
 * @Date 2022/7/25 18:33
 * @Description 测试工具类 PropertiesUtil
 * @Version 1.0
 **/
@SpringBootTest
public class PropertiesUtilTest {
    @Test
    public void testgetProperties(){
        assertEquals(PropertiesUtil.getCommonYml("tg.token"),"********************");
        assertEquals(PropertiesUtil.getCommonYml("tg.username"),"ajie1024_bot");