apoc.load.LoadJson Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of apoc-core Show documentation
Core package for Neo4j Procedures
The newest version!
/*
 * Copyright (c) "Neo4j"
 * Neo4j Sweden AB [http://neo4j.com]
 *
 * This file is part of Neo4j.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package apoc.load;

import static apoc.load.LoadJsonUtils.loadJsonStream;
import static apoc.util.CompressionConfig.COMPRESSION;

import apoc.result.MapResult;
import apoc.result.ObjectResult;
import apoc.util.CompressionAlgo;
import apoc.util.JsonUtil;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Stream;
import org.neo4j.graphdb.security.URLAccessChecker;
import org.neo4j.procedure.Context;
import org.neo4j.procedure.Description;
import org.neo4j.procedure.Name;
import org.neo4j.procedure.Procedure;
import org.neo4j.procedure.TerminationGuard;

public class LoadJson {

    @Context
    public TerminationGuard terminationGuard;

    @Context
    public URLAccessChecker urlAccessChecker;

    @SuppressWarnings("unchecked")
    @Procedure("apoc.load.jsonArray")
    @Description("Loads array from a JSON URL (e.g. web-API) to then import the given JSON file as a stream of values.")
    public Stream jsonArray(
            @Name("url") String url,
            @Name(value = "path", defaultValue = "") String path,
            @Name(value = "config", defaultValue = "{}") Map config) {
        return JsonUtil.loadJson(
                        url, null, null, path, true, (List) config.get("pathOptions"), urlAccessChecker)
                .flatMap((value) -> {
                    if (value instanceof List) {
                        List list = (List) value;
                        if (list.isEmpty()) return Stream.empty();
                        if (list.get(0) instanceof Map) return list.stream().map(ObjectResult::new);
                    }
                    return Stream.of(new ObjectResult(value));
                });
    }

    @Procedure("apoc.load.json")
    @Description("Imports JSON file as a stream of values if the given JSON file is a `LIST`.\n"
            + "If the given JSON file is a `MAP`, this procedure imports a single value instead.")
    public Stream json(
            @Name("urlOrKeyOrBinary") Object urlOrKeyOrBinary,
            @Name(value = "path", defaultValue = "") String path,
            @Name(value = "config", defaultValue = "{}") Map config) {
        return jsonParams(urlOrKeyOrBinary, null, null, path, config);
    }

    @SuppressWarnings("unchecked")
    @Procedure("apoc.load.jsonParams")
    @Description(
            "Loads parameters from a JSON URL (e.g. web-API) as a stream of values if the given JSON file is a `LIST`.\n"
                    + "If the given JSON file is a `MAP`, this procedure imports a single value instead.")
    public Stream jsonParams(
            @Name("urlOrKeyOrBinary") Object urlOrKeyOrBinary,
            @Name("headers") Map headers,
            @Name("payload") String payload,
            @Name(value = "path", defaultValue = "") String path,
            @Name(value = "config", defaultValue = "{}") Map config) {
        if (config == null) config = Collections.emptyMap();
        boolean failOnError = (boolean) config.getOrDefault("failOnError", true);
        String compressionAlgo = (String) config.getOrDefault(COMPRESSION, CompressionAlgo.NONE.name());
        List pathOptions = (List) config.get("pathOptions");
        return loadJsonStream(
                urlOrKeyOrBinary,
                headers,
                payload,
                path,
                failOnError,
                compressionAlgo,
                pathOptions,
                terminationGuard,
                urlAccessChecker);
    }
}