Yaml 工具类

最新推荐文章于 2024-03-22 22:29:58 发布

小厂程序员

最新推荐文章于 2024-03-22 22:29:58 发布

阅读量289

点赞数

分类专栏：解决方案 Java 文章标签： java servlet spring

本文链接：https://blog.csdn.net/fofcn/article/details/129052466

版权

Java 同时被 2 个专栏收录

12 篇文章 0 订阅

订阅专栏

解决方案

5 篇文章 0 订阅

订阅专栏

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import lombok.extern.slf4j.Slf4j;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;

/**
 * YAML解析
 *
 * @author errorfatal89@gmail.com
 * @date 2022/03/11
 */
@Slf4j
public class YamlUtil {

    /**
     * 将yaml配置转换为对象
     * @param clazz 类定义
     * @param path 文件路径
     * @param <T>
     * @return java对象
     */
    public static <T> T readObject(Class<T> clazz, String path) {
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        mapper.findAndRegisterModules();
        mapper.enable(DeserializationFeature.ACCEPT_EMPTY_STRING_AS_NULL_OBJECT);
        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        try {
            return mapper.readValue(new File(path), clazz);
        } catch (JsonProcessingException e) {
            log.error("json process error, value: {}", path, e);
        } catch (IOException e) {
            log.error("json process error, value: {}", path, e);
        }

        return null;
    }

    /**
     * 将yaml配置转换为对象
     * @param clazz 类定义
     * @param in 文件流
     * @param <T>
     * @return java对象
     */
    public static <T> T readObject(Class<T> clazz, InputStream in) {
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        mapper.findAndRegisterModules();
        mapper.enable(DeserializationFeature.ACCEPT_EMPTY_STRING_AS_NULL_OBJECT);
        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        try {
            return mapper.readValue(in, clazz);
        } catch (JsonProcessingException e) {
            log.error("json process error", e);
        } catch (IOException e) {
            log.error("json process error",  e);
        } finally {
            try {
                in.close();
            } catch (IOException e) {
                log.error("json process error",  e);
            }
        }

        return null;
    }

    /**
     * 将yaml配置转换为对象
     * @param type 类定义
     * @param in 文件流
     * @param <T>
     * @return java对象
     */
    public static <T> T readObject(TypeReference<T> type, InputStream in) {
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        mapper.findAndRegisterModules();
        mapper.enable(DeserializationFeature.ACCEPT_EMPTY_STRING_AS_NULL_OBJECT);
        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        try {
            return mapper.readValue(in, type);
        } catch (JsonProcessingException e) {
            log.error("json process error", e);
        } catch (IOException e) {
            log.error("json process error",  e);
        } finally {
            try {
                in.close();
            } catch (IOException e) {
                log.error("json process error",  e);
            }
        }

        return null;
    }


    /**
     * 将yaml配置转换为对象
     * @param clazz 类定义
     * @param text 内容
     * @param <T>
     * @return java对象
     */
    public static <T> T readObject(String text, Class<T> clazz) {
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        mapper.findAndRegisterModules();
        mapper.enable(DeserializationFeature.ACCEPT_EMPTY_STRING_AS_NULL_OBJECT);
        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        try {
            return mapper.readValue(new File(text), clazz);
        } catch (JsonProcessingException e) {
            log.error("json process error, value: {}", text, e);
        } catch (IOException e) {
            log.error("json process error, value: {}", text, e);
        }

        return null;
    }

    /**
     * 将yaml配置转换为对象
     * @param type 类定义
     * @param in 文件流
     * @param <T>
     * @return java对象
     */
    public static <T> T readObject(InputStream in, Class<T> clazz, String prefix) {
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        mapper.findAndRegisterModules();
        mapper.enable(DeserializationFeature.ACCEPT_EMPTY_STRING_AS_NULL_OBJECT);
        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        try {
            Map<String, Object> map = mapper.readValue(in, new TypeReference<Map<String, Object>>(){});
            return convertObject(clazz, prefix, map);
        } catch (JsonProcessingException e) {
            log.error("json process error", e);
        } catch (IOException e) {
            log.error("json process error",  e);
        } finally {
            try {
                in.close();
            } catch (IOException e) {
                log.error("json process error",  e);
            }
        }

        return null;
    }

    private static <T> T convertObject(Class<T> clazz, String prefix, Map<String, Object> map) {
        ObjectMapper mapper = new ObjectMapper();
        return mapper.convertValue(map.get(prefix), clazz);
    }
}