com.graphaware.module.es.ElasticSearchConfiguration 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 {
private static final String DEFAULT_KEY_PROPERTY = "uuid";
private static final boolean DEFAULT_RETRY_ON_ERROR = false;
private static final boolean DEFAULT_EXECUTE_BULK = true;
private static final InclusionPolicies DEFAULT_INCLUSION_POLICIES = InclusionPoliciesFactory.allBusiness().with(IncludeNoRelationships.getInstance());
private static final Mapping DEFAULT_MAPPING = DefaultMapping.newInstance();
static {
DEFAULT_MAPPING.configure(new HashMap<>());
}
private static final int DEFAULT_QUEUE_CAPACITY = 10000;
private static final int DEFAULT_REINDEX_BATCH_SIZE = 1000;
private static final String DEFAULT_AUTH_USER = null;
private static final String DEFAULT_AUTH_PASSWORD = null;
private final String uri;
private final String port;
private final String keyProperty;
private final boolean retryOnError;
private final int queueCapacity;
private final boolean executeBulk;
private final String authUser;
private final String authPassword;
private final Mapping mapping;
private final int reindexBatchSize;
/**
* Construct a new configuration.
*
* @param inclusionPolicies specifying which nodes and node properties to index in Elasticsearch. Must not be null.
* @param initializeUntil until what time in ms since epoch it is ok to re-index the entire database in case the configuration has changed since
* the last time the module was started, or if it is the first time the module was registered.
* 0 for never. The purpose of this is not to re-index all the time if the user is unaware.
* @param uri Elasticsearch URI. Must not be null.
* @param port Elasticsearch port. Must not be null.
* @param keyProperty name of the node property that serves as the key, under which the node will be indexed in Elasticsearch. Must not be null or empty.
* @param retryOnError whether to retry an index update after a failure (true) or throw the update away (false).
* @param queueCapacity capacity of the queue holding operations to be written to Elasticsearch. Must be positive.
* @param executeBulk whether or not to execute updates against Elasticsearch in bulk. It is recommended to set this to true.*
* @param mapping name of the mapping class to use to convert Neo4j node/relationships to ElasticSearch documents.
*/
private ElasticSearchConfiguration(InclusionPolicies inclusionPolicies, long initializeUntil, String uri, String port, String keyProperty, boolean retryOnError, int queueCapacity, int reindexBatchSize, boolean executeBulk, String authUser, String authPassword, Mapping mapping) {
super(inclusionPolicies, initializeUntil);
this.uri = uri;
this.port = port;
this.keyProperty = keyProperty;
this.retryOnError = retryOnError;
this.queueCapacity = queueCapacity;
this.reindexBatchSize = reindexBatchSize;
this.executeBulk = executeBulk;
this.authUser = authUser;
this.authPassword = authPassword;
this.mapping = mapping;
}
/**
* {@inheritDoc}
*/
@Override
public ElasticSearchConfiguration newInstance(InclusionPolicies inclusionPolicies, long initializeUntil) {
return new ElasticSearchConfiguration(inclusionPolicies, initializeUntil, getUri(), getPort(), getKeyProperty(), isRetryOnError(), getQueueCapacity(), getReindexBatchSize(), isExecuteBulk(), DEFAULT_AUTH_USER, DEFAULT_AUTH_PASSWORD, getMapping());
}
public static ElasticSearchConfiguration defaultConfiguration() {
return new ElasticSearchConfiguration(DEFAULT_INCLUSION_POLICIES, NEVER, null, null, DEFAULT_KEY_PROPERTY, DEFAULT_RETRY_ON_ERROR, DEFAULT_QUEUE_CAPACITY, DEFAULT_REINDEX_BATCH_SIZE, DEFAULT_EXECUTE_BULK, DEFAULT_AUTH_USER, DEFAULT_AUTH_PASSWORD, DEFAULT_MAPPING);
}
public ElasticSearchConfiguration withUri(String uri) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), uri, getPort(), getKeyProperty(), isRetryOnError(), getQueueCapacity(), getReindexBatchSize(), isExecuteBulk(), getAuthUser(), getAuthPassword(), getMapping());
}
public ElasticSearchConfiguration withPort(String port) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), port, getKeyProperty(), isRetryOnError(), getQueueCapacity(), getReindexBatchSize(), isExecuteBulk(), getAuthUser(), getAuthPassword(), getMapping());
}
public ElasticSearchConfiguration withKeyProperty(String keyProperty) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), getPort(), keyProperty, isRetryOnError(), getQueueCapacity(), getReindexBatchSize(), isExecuteBulk(), getAuthUser(), getAuthPassword(), getMapping());
}
public ElasticSearchConfiguration withRetryOnError(boolean retryOnError) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), getPort(), getKeyProperty(), retryOnError, getQueueCapacity(), getReindexBatchSize(), isExecuteBulk(), getAuthUser(), getAuthPassword(), getMapping());
}
public ElasticSearchConfiguration withQueueCapacity(int queueCapacity) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), getPort(), getKeyProperty(), isRetryOnError(), queueCapacity, getReindexBatchSize(), isExecuteBulk(), getAuthUser(), getAuthPassword(), getMapping());
}
public ElasticSearchConfiguration withReindexBatchSize(int reindexBatchSize) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), getPort(), getKeyProperty(), isRetryOnError(), getQueueCapacity(), reindexBatchSize, isExecuteBulk(), getAuthUser(), getAuthPassword(), getMapping());
}
public ElasticSearchConfiguration withExecuteBulk(boolean executeBulk) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), getPort(), getKeyProperty(), isRetryOnError(), getQueueCapacity(), getReindexBatchSize(), executeBulk, getAuthUser(), getAuthPassword(), getMapping());
}
public ElasticSearchConfiguration withAuthCredentials(String authUser, String authPassword) {
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), getPort(), getKeyProperty(), isRetryOnError(), getQueueCapacity(), getReindexBatchSize(), isExecuteBulk(), authUser, authPassword, getMapping());
}
public ElasticSearchConfiguration withMapping(Mapping mapping, Map mappingConfig) {
// prevents mappings from being started without configure() from being called
mapping.configure(mappingConfig);
return new ElasticSearchConfiguration(getInclusionPolicies(), initializeUntil(), getUri(), getPort(), getKeyProperty(), isRetryOnError(), getQueueCapacity(), getReindexBatchSize(), isExecuteBulk(), getAuthUser(), getAuthPassword(), mapping);
}
public String getUri() {
return uri;
}
public String getPort() {
return port;
}
public String getKeyProperty() {
return keyProperty;
}
public boolean isRetryOnError() {
return retryOnError;
}
public int getQueueCapacity() {
return queueCapacity;
}
public int getReindexBatchSize() {
return reindexBatchSize;
}
public boolean isExecuteBulk() {
return executeBulk;
}
public String getAuthUser() {
return authUser;
}
public String getAuthPassword() { return authPassword; }
public Mapping getMapping() { return mapping; }
/**
* {@inheritDoc}
*/
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
if (!super.equals(o)) {
return false;
}
ElasticSearchConfiguration that = (ElasticSearchConfiguration) o;
if (initializeUntil() != that.initializeUntil()) { //a bit of a hack to force initialization if this changes
return false;
}
if (retryOnError != that.retryOnError) {
return false;
}
if (queueCapacity != that.queueCapacity) {
return false;
}
if (reindexBatchSize != that.reindexBatchSize) {
return false;
}
if (executeBulk != that.executeBulk) {
return false;
}
if (!uri.equals(that.uri)) {
return false;
}
if (!port.equals(that.port)) {
return false;
}
return keyProperty.equals(that.keyProperty);
}
/**
* {@inheritDoc}
*/
@Override
public int hashCode() {
int result = super.hashCode();
result = 31 * result + (int) (initializeUntil() ^ (initializeUntil() >>> 32));
result = 31 * result + uri.hashCode();
result = 31 * result + port.hashCode();
result = 31 * result + keyProperty.hashCode();
result = 31 * result + (retryOnError ? 1 : 0);
result = 31 * result + queueCapacity;
result = 31 * result + reindexBatchSize;
result = 31 * result + (executeBulk ? 1 : 0);
return result;
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy