com.lordofthejars.nosqlunit.elasticsearch2.EmbeddedElasticsearch Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nosqlunit-elasticsearch2 Show documentation
The newest version!
package com.lordofthejars.nosqlunit.elasticsearch2;

import org.elasticsearch.common.settings.Settings;
import org.junit.rules.ExternalResource;

import java.io.File;

public class EmbeddedElasticsearch extends ExternalResource {

	private EmbeddedElasticsearch() {
		super();
	}

	protected EmbeddedElasticsearchLifecycleManager elasticsearchLifecycleManager;

	public static class EmbeddedElasticsearchRuleBuilder {

		private EmbeddedElasticsearchLifecycleManager elasticsearchLifecycleManager;

		private EmbeddedElasticsearchRuleBuilder() {
			this.elasticsearchLifecycleManager = new EmbeddedElasticsearchLifecycleManager();
		}

		public static EmbeddedElasticsearchRuleBuilder newEmbeddedElasticsearchRule() {
			return new EmbeddedElasticsearchRuleBuilder();
		}

		public EmbeddedElasticsearchRuleBuilder homePath(File homePath) {
			this.elasticsearchLifecycleManager.setHomePath(homePath);
			return this;
		}

		public EmbeddedElasticsearchRuleBuilder targetPath(File targetPath) {
			this.elasticsearchLifecycleManager.setDataPath(targetPath);
			return this;
		}

		public EmbeddedElasticsearchRuleBuilder clusterName(String clusterName) {
			this.elasticsearchLifecycleManager.setClusterName(clusterName);
			return this;
		}

		public EmbeddedElasticsearchRuleBuilder client(boolean client) {
			this.elasticsearchLifecycleManager.setClient(client);
			return this;
		}

		public EmbeddedElasticsearchRuleBuilder settings(Settings settings) {
			this.elasticsearchLifecycleManager.setSettings(settings);
			return this;
		}

		public EmbeddedElasticsearchRuleBuilder local(boolean local) {
			this.elasticsearchLifecycleManager.setLocal(local);
			return this;
		}

		public EmbeddedElasticsearchRuleBuilder data(boolean data) {
			this.elasticsearchLifecycleManager.setData(data);
			return this;
		}

		public EmbeddedElasticsearch build() {

			if (this.elasticsearchLifecycleManager.getDataPath() == null) {
				throw new IllegalArgumentException("No Path to Embedded Elasticsearch is provided.");
			}

			EmbeddedElasticsearch embeddedElasticsearch = new EmbeddedElasticsearch();
			embeddedElasticsearch.elasticsearchLifecycleManager = this.elasticsearchLifecycleManager;

			return embeddedElasticsearch;

		}

	}

	@Override
	protected void before() throws Throwable {
		this.elasticsearchLifecycleManager.startEngine();
	}

	@Override
	protected void after() {
		this.elasticsearchLifecycleManager.stopEngine();
	}
}