Yaml的配置文件读取

---

引入依赖

<dependency>
    <groupId>org.yaml</groupId>
    <artifactId>snakeyaml</artifactId>
    <version>1.29</version>
</dependency>
<dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
    <version>1.18.20</version>
    <scope>provided</scope>
</dependency>

配置文件

在项目resources下存在文件：config.yaml
内容如下：

code: 000
message: "成功"
stringList:
  - "hello"
  - "world"

只要定义同等名称及类型即可赋值获取

类YamlUtils

采用 @Data 注解 省略set/get 方法

import lombok.Data;
import org.yaml.snakeyaml.Yaml;

import java.io.InputStream;
import java.lang.reflect.Field;
import java.util.List;
import java.util.Map;

/**
 * author ：demon
 * date ：2021 2021/8/15 14:51
 * desc ：读取yaml文件，并获取值
 */
@Data
public class YamlUtils {

    private static YamlUtils yamlUtils;

    public static YamlUtils getInstance() {
        if (yamlUtils == null) {
            yamlUtils = new YamlUtils();
        }
        return yamlUtils;
    }

    static {
        try {
            Yaml yaml = new Yaml();
            InputStream inputStream = YamlUtils.class.getResourceAsStream("/config.yaml");
            // 可以将值转换为Map，通过map我们取值就可以了.
            Map<String, Object> ymlMap = yaml.load(inputStream);
            // 动态赋值
            yamlUtils = YamlUtils.class.newInstance();
            for (Field field : yamlUtils.getClass().getDeclaredFields()) {
                if (ymlMap.containsKey(field.getName())) {
                    boolean flag = field.isAccessible();
                    field.setAccessible(true);
                    Object object = ymlMap.get(field.getName());
                    if (object != null && field.getType().isAssignableFrom(object.getClass())) {
                        field.set(yamlUtils, object);
                    }
                    field.setAccessible(flag);
                }
            }

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private Integer code;
    private String message;
    private List<String> stringList;
}

测试

@Test
public void Test1(){
  System.out.println(YamlUtils.getInstance().getCode());
  System.out.println(YamlUtils.getInstance().getMessage());
  System.out.println(YamlUtils.getInstance().getStringList());
}