com.networknt.schema.resource.MapSchemaLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of json-schema-validator Show documentation

A json schema validator that supports draft v4, v6, v7, v2019-09 and v2020-12

There is a newer version: 1.5.3

Show newest version

package com.networknt.schema.resource;

import java.io.ByteArrayInputStream;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.function.Function;

import com.networknt.schema.AbsoluteIri;

/**
 * Map implementation of {@link SchemaLoader}.
 */
public class MapSchemaLoader implements SchemaLoader {
    private final Function mappings;

    /**
     * Sets the schema data by absolute IRI.
     * 
     * @param mappings the mappings
     */
    public MapSchemaLoader(Map mappings) {
        this(mappings::get);
    }

    /**
     * Sets the schema data by absolute IRI function.
     * 
     * @param mappings the mappings
     */
    public MapSchemaLoader(Function mappings) {
        this.mappings = mappings;
    }

    /**
     * Sets the schema data by using two mapping functions.
     * 
     * Firstly to map the IRI to an object. If the object is null no mapping is
     * performed.
     * 
     * Next to map the object to the schema data.
     * 
     * @param              the type of the object
     * @param mapIriToObject  the mapping of IRI to object
     * @param mapObjectToData the mappingof object to schema data
     */
    public  MapSchemaLoader(Function mapIriToObject, Function mapObjectToData) {
        this.mappings = iri -> {
            T result = mapIriToObject.apply(iri);
            if (result != null) {
                return mapObjectToData.apply(result);
            }
            return null;
        };
    }

    @Override
    public InputStreamSource getSchema(AbsoluteIri absoluteIri) {
        try {
            String result = mappings.apply(absoluteIri.toString());
            if (result != null) {
                return () -> new ByteArrayInputStream(result.getBytes(StandardCharsets.UTF_8));
            }
        } catch (Exception e) {
            // Do nothing
        }
        return null;
    }
}