apoc.mongodb.MongoDBUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of apoc-extended Show documentation

Extended package for Neo4j Procedures

There is a newer version: 5.26.0

package apoc.mongodb;

import apoc.util.JsonUtil;
import apoc.util.MissingDependencyException;
import org.bson.Document;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class MongoDBUtils {
    public static final String MONGO_MISSING_DEPS_ERROR = """
            Cannot find the jar into the plugins folder.\s
            Please put the apoc-mongodb-dependencies-5.x.x-all.jar into plugin folder.
            See the documentation: See the documentation: https://neo4j.com/labs/apoc/5/database-integration/mongo/#mongodb-dependencies""";


    protected static Document getDocument(Object query) {
        if (query == null) {
            return new Document();
        }
        String json = query instanceof String
                ? (String) query
                : JsonUtil.writeValueAsString(query);
        
        json = adaptLegacyDocuments(json);
        
        Document document = Document.parse(json);
        
        return document;
    }

    /**
     * In case someone use an old notation, e.g. {`$binary`: $bytes, `$subType`: '00'}
     */
    private static String adaptLegacyDocuments(String json) {
        return json.replace("'$subType'", "'$type'")
                .replace("\"$subType\"", "\"$type\"");
    }

    protected static List getDocuments(List> pipeline) {
        return pipeline.stream()
                .map(MongoDBUtils::getDocument)
                .collect(Collectors.toList());
    }

    protected static MongoDbConfig getMongoConfig(Map config) {
        try {
            return new MongoDbConfig(config);
        } catch (NoClassDefFoundError e) {
            throw new MissingDependencyException(MONGO_MISSING_DEPS_ERROR);
        }
    }
}