io.github.sskorol.data.YamlReader Maven / Gradle / Ivy

Go to download
package io.github.sskorol.data;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import one.util.streamex.StreamEx;

import static com.fasterxml.jackson.databind.DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES;
import static java.lang.String.format;

public class YamlReader implements DataReader {

    private final Class entityClass;
    private final String path;

    public YamlReader(final Class entityClass) {
        this(entityClass, "");
    }

    public YamlReader(final Class entityClass, final String path) {
        this.entityClass = entityClass;
        this.path = path;
    }

    @SuppressWarnings("unchecked")
    public StreamEx read() {
        try {
            var yamlFactory = new YAMLFactory();
            return StreamEx.of(new ObjectMapper(yamlFactory)
                .configure(FAIL_ON_UNKNOWN_PROPERTIES, false)
                .readValues(yamlFactory.createParser(getUrl()), entityClass)
                .readAll());
        } catch (Exception ex) {
            throw new IllegalArgumentException(format("Unable to read YAML data to %s.", entityClass), ex);
        }
    }

    @Override
    public Class getEntityClass() {
        return entityClass;
    }

    @Override
    public String getPath() {
        return path;
    }
}