com.graphaware.module.es.proc.ElasticSearchProcedures Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-to-elasticsearch Show documentation
GraphAware Framework Module for Integrating Neo4j with Elasticsearch
There is a newer version: 3.5.14.58.10
Show newest version
/*
 * Copyright (c) 2013-2016 GraphAware
 *
 * This file is part of the GraphAware Framework.
 *
 * GraphAware Framework is free software: you can redistribute it and/or modify it under the terms of
 * the GNU General Public License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of
 * the GNU General Public License along with this program.  If not, see
 * .
 */

package com.graphaware.module.es.proc;

import com.graphaware.common.log.LoggerFactory;
import com.graphaware.module.es.ElasticSearchModule;
import com.graphaware.module.es.proc.result.JsonSearchResult;
import com.graphaware.module.es.proc.result.NodeSearchResult;
import com.graphaware.module.es.proc.result.RelationshipSearchResult;
import com.graphaware.module.es.proc.result.StatusResult;
import com.graphaware.module.es.search.Searcher;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.Relationship;
import org.neo4j.logging.Log;
import org.neo4j.logging.Logger;
import org.neo4j.procedure.Context;
import org.neo4j.procedure.Name;
import org.neo4j.procedure.PerformsWrites;
import org.neo4j.procedure.Procedure;

import java.util.stream.Stream;

import static com.graphaware.runtime.RuntimeRegistry.getStartedRuntime;

public class ElasticSearchProcedures {

    private static final Log LOG = LoggerFactory.getLogger(ElasticSearchProcedures.class);

    @Context
    public GraphDatabaseService database;

    private ElasticSearchModule getModule(GraphDatabaseService database) {
        return getStartedRuntime(database).getModule(ElasticSearchModule.class);
    }

    private static Searcher getSearcher(GraphDatabaseService database) {
        return new Searcher(database);
    }

    @Procedure("ga.es.queryNode")
    @PerformsWrites
    public Stream queryNode(@Name("query") String query) {
        try {
            return getSearcher(database).search(query, Node.class).stream().map(match -> {
                return new NodeSearchResult(match.getItem(), match.score);
            });
        } catch (Exception e) {
            LOG.error("", e);

            return Stream.empty();
        }
    }

    @Procedure("ga.es.queryRelationship")
    @PerformsWrites
    public Stream queryRelationship(@Name("query") String query) {
        return getSearcher(database).search(query, Relationship.class).stream().map(match -> {
            return new RelationshipSearchResult(match.getItem(), match.score);
        });
    }

    @Procedure("ga.es.queryNodeRaw")
    @PerformsWrites
    public Stream queryNodeRaw(@Name("query") String query) {
        return Stream.of(new JsonSearchResult(getSearcher(database).rawSearch(query, Node.class)));
    }

    @Procedure("ga.es.queryRelationshipRaw")
    @PerformsWrites
    public Stream queryRelationshipRaw(@Name("query") String query) {
        return Stream.of(new JsonSearchResult(getSearcher(database).rawSearch(query, Relationship.class)));
    }

    @Procedure("ga.es.initialized")
    public Stream initialized() {
        return Stream.of(new StatusResult(getModule(database).isReindexCompleted()));
    }
}