org.molgenis.data.elasticsearch.util.ElasticsearchUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of molgenis-data-elasticsearch Show documentation

The newest version!

package org.molgenis.data.elasticsearch.util;

import static org.elasticsearch.client.Requests.refreshRequest;

import java.util.concurrent.TimeUnit;

import org.elasticsearch.action.bulk.BulkProcessor;
import org.elasticsearch.client.Client;
import org.molgenis.data.MolgenisDataException;

public class ElasticsearchUtils
{
	private final Client client;

	public ElasticsearchUtils(Client client)
	{
		this.client = client;
	}

	public void deleteIndex(String index)
	{
		client.admin().indices().prepareDelete(index).execute().actionGet();
	}

	public boolean indexExists(String index)
	{
		return client.admin().indices().prepareExists(index).execute().actionGet().isExists();
	}

	// Wait until elasticsearch is ready
	public void waitForYellowStatus()
	{
		client.admin().cluster().prepareHealth().setWaitForYellowStatus().execute().actionGet();
	}

	public void refreshIndex(String index)
	{
		client.admin().indices().refresh(refreshRequest(index)).actionGet();
	}

	public void waitForCompletion(BulkProcessor bulkProcessor)
	{
		try
		{
			boolean isCompleted = bulkProcessor.awaitClose(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
			if (!isCompleted)
			{
				throw new MolgenisDataException("Failed to complete bulk request within the given time");
			}
		}
		catch (InterruptedException e)
		{
			throw new RuntimeException(e);
		}
	}

}