com.graphaware.module.es.mapping.BaseMapping 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 config) {
indexPrefix = config.getOrDefault("index", DEFAULT_INDEX).trim();
if (indexPrefix.isEmpty()) { indexPrefix = DEFAULT_INDEX; }
LOG.info("ElasticSearch index-prefix set to %s", indexPrefix);
keyProperty = config.getOrDefault("keyProperty", DEFAULT_KEY_PROPERTY).trim();
if (keyProperty.isEmpty()) { keyProperty = DEFAULT_KEY_PROPERTY; }
LOG.info("ElasticSearch key-property set to %s", keyProperty);
forceStrings = config.getOrDefault("forceStrings", DEFAULT_FORCE_STRINGS).trim().toLowerCase().equals("true");
LOG.info("ElasticSearch force-strings set to %s", forceStrings);
}
/**
* @return the name of the ElasticSearch index name prefix to use for indexing.
*/
protected String getIndexPrefix() {
return indexPrefix;
}
/**+
* @return the node/relationship property used as document ID in ElasticSearch (usually "uuid")
*/
@Override
public String getKeyProperty() {
return keyProperty;
}
/**
* Get the key under which the given {@link NodeExpressions} or {@link RelationshipExpressions} will be indexed in Elasticsearch.
*
* @param propertyContainer Node or relationship to be indexed.
* @return key of the node.
*/
protected final String getKey(PropertyContainerExpressions propertyContainer) {
return String.valueOf(propertyContainer.getProperties().get(getKeyProperty()));
}
/**
* Creates an ElasticSearch document from a node or relationship.
* Includes all properties, except the keyProperty (usually "uuid") which is already the ElasticSearch document ID.
*
* @param item The node or relationship to build the ElasticSearch representation for.
* @return a document to index in ElasticSearch
*/
private Map commonMap(PropertyContainerExpressions item) {
String keyProperty = getKeyProperty();
Map source = new HashMap<>();
Map properties = item.getProperties();
if (item.getProperties() != null) {
for (String key : properties.keySet()) {
if (keyProperty.equals(key)) {
// don't index the key property in ElasticSearch
continue;
}
Object value = properties.get(key);
if (value.getClass().isArray()) {
int length = Array.getLength(value);
Object[] newValues = new Object[length];
for (int i = 0; i < length; i ++) {
newValues[i] = normalizeProperty(Array.get(value, i));
}
value = newValues;
} else {
value = normalizeProperty(value);
}
source.put(key, value);
}
}
return source;
}
/**
* Can be used to convert all properties to their String representation.
*
* @param propertyValue Property value to normalize
* @return normalized property value
*/
protected Object normalizeProperty(Object propertyValue) {
return forceStrings ? String.valueOf(propertyValue) : propertyValue;
}
/**
* Convert a Neo4j representation to a ElasticSearch representation of a node.
*
* @param node A Neo4j node
* @return a map of fields to store in ElasticSearch
*/
protected Map map(NodeExpressions node) {
Map source = commonMap(node);
addExtra(source, node);
return source;
}
/**
* Convert a Neo4j representation to a ElasticSearch representation of a relationship.
*
* @param relationship A Neo4j relationship
* @return a map of fields to store in ElasticSearch
*/
protected Map map(RelationshipExpressions relationship) {
Map source = commonMap(relationship);
addExtra(source, relationship);
return source;
}
protected void addExtra(Map data, NodeExpressions node) { }
protected void addExtra(Map data, RelationshipExpressions relationship) { }
/**
* Create the ElasticSearch index(es) and initialize the mapping
*
* @param client The ElasticSearch client to use.
*
* @throws Exception
*/
public void createIndexAndMapping(JestClient client) throws Exception {
List indexes = Arrays.asList(
getIndexFor(Node.class),
getIndexFor(Relationship.class)
);
indexes.stream().distinct().forEach(index -> {
try {
createIndexAndMapping(client, index);
} catch (Exception e) {
throw new RuntimeException(e);
}
});
}
private void createIndexAndMapping(JestClient client, String index) throws Exception {
if (client.execute(new IndicesExists.Builder(index).build()).isSucceeded()) {
LOG.info("Index " + index + " already exists in ElasticSearch.");
return;
}
LOG.info("Index " + index + " does not exist in ElasticSearch, creating...");
final JestResult execute = client.execute(new CreateIndex.Builder(index).build());
if (execute.isSucceeded()) {
LOG.info("Created ElasticSearch index.");
} else {
LOG.error("Failed to create ElasticSearch index. Details: " + execute.getErrorMessage());
}
}
public abstract String getIndexFor(Class searchedType);
protected List> emptyActions() {
return new ArrayList<>();
}
@Override
public boolean bypassInclusionPolicies() {
return false;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy