com.graphaware.module.es.search.Searcher 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
Show all versions of neo4j-to-elasticsearch Show documentation
GraphAware Framework Module for Integrating Neo4j with Elasticsearch
/*
* 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
* getRelationshipResolver() {
return match -> {
Relationship rel;
try {
rel = database.getRelationshipById(uuidReader.getRelationshipIdByUuid(match.uuid));
} catch(NotFoundException e) {
rel = null;
}
if (rel == null) {
LOG.warn("Could not find relationship with uuid (" + keyProperty + "): " + match.uuid);
}
return rel;
};
}
private Function getNodeResolver() {
return match -> {
Node node = null;
try {
node = database.getNodeById(uuidReader.getNodeIdByUuid(match.uuid));
} catch (NotFoundException e){
node = null;
}
if (node == null) {
LOG.warn("Could not find node with uuid (" + keyProperty + "): " + match.uuid);
}
return node;
};
}
private List> resolveMatchItems(List> searchMatches, Function resolver) {
List> resolvedResults = new ArrayList<>();
try (Transaction tx = database.beginTx()) {
searchMatches.stream().forEach(match -> {
T item = resolver.apply(match);
if (item != null) {
match.setItem(item);
resolvedResults.add(match);
}
});
tx.success();
}
return resolvedResults;
}
private List> buildSearchMatches(SearchResult searchResult) {
List> matches = new ArrayList<>();
Set> entrySet = searchResult.getJsonObject().entrySet();
entrySet.stream()
.filter((item) -> (item.getKey().equalsIgnoreCase("hits")))
.map((item) -> (JsonObject) item.getValue())
.filter((hits) -> (hits != null))
.map((hits) -> hits.getAsJsonArray("hits"))
.filter((hitsArray) -> (hitsArray != null))
.forEach((hitsArray) -> {
for (JsonElement element : hitsArray) {
JsonObject obj = (JsonObject) element;
Double score = obj.get("_score") != null && !obj.get("_score").toString().equals("null") ? Double.valueOf(obj.get("_score").toString()) : null;
String keyValue = obj.get("_id") != null ? obj.get("_id").getAsString() : null;
if (keyValue == null) {
LOG.warn("No key found in search result: " + obj.getAsString());
} else {
matches.add(new SearchMatch<>(keyValue, score));
}
}
});
return matches;
}
public static JestClient createClient(String uri, String port, String authUser, String authPassword) {
notNull(uri);
notNull(port);
LOG.info("Creating Jest Client...");
JestClientFactory factory = new JestClientFactory();
String esHost = String.format("http://%s:%s", uri, port);
HttpClientConfig.Builder clientConfigBuilder = new HttpClientConfig.Builder(esHost).multiThreaded(true);
if (authUser != null && authPassword != null) {
BasicCredentialsProvider customCredentialsProvider = new BasicCredentialsProvider();
customCredentialsProvider.setCredentials(
new AuthScope(uri, Integer.parseInt(port)),
new UsernamePasswordCredentials(authUser, authPassword)
);
LOG.info("Enabling Auth for ElasticSearch: " + authUser);
clientConfigBuilder.credentialsProvider(customCredentialsProvider);
}
factory.setHttpClientConfig(clientConfigBuilder.build());
LOG.info("Created Jest Client.");
return factory.getObject();
}
/**
*
* @param query The query to send to the index
* @param clazz {@link Node} or {@link Relationship}, to decide which index to send the query to.
* @param {@link Node} or {@link Relationship}
* @return the query response
*/
private SearchResult doQuery(String query, Class clazz) {
Search search = new Search.Builder(query).addIndex(mapping.getIndexFor(clazz)).build();
SearchResult result;
try {
result = client.execute(search);
} catch (IOException ex) {
throw new RuntimeException("Error while performing query on ElasticSearch", ex);
}
return result;
}
/**
* Search for nodes or relationships
*
* @param query An ElasticSearch query in JSON format (serialized as a string)
* @param clazz {@link Node} or {@link Relationship}
* @param {@link Node} or {@link Relationship}
* @return a list of matches (with node or a relationship)
*/
public List> search(String query, Class clazz) {
SearchResult result = doQuery(query, clazz);
List> matches = buildSearchMatches(result);
@SuppressWarnings("unchecked")
Function resolver = (Function) (
clazz.equals(Node.class) ? getNodeResolver() : getRelationshipResolver()
);
return resolveMatchItems(matches, resolver);
}
/**
*
* @param query The search query
* @param clazz The index key ({@link Node} or {@link Relationship})
* @param {@link Node} or {@link Relationship}
* @return a JSON string
*/
public String rawSearch(String query, Class clazz) {
SearchResult r = doQuery(query, clazz);
return r.getJsonString();
}
@Override
protected void finalize() throws Throwable {
super.finalize();
if (client == null) { return; }
client.shutdownClient();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy