/** * Copyright 2014 Flipkart Internet Pvt. Ltd. * <p> * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * <p> * http://www.apache.org/licenses/LICENSE-2.0 * <p> * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.flipkart.foxtrot.core; import com.google.common.base.Stopwatch; import org.apache.commons.io.FileUtils; import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest; import org.elasticsearch.action.admin.indices.refresh.RefreshRequest; import org.elasticsearch.client.Client; import org.elasticsearch.common.settings.Settings; import org.elasticsearch.node.Node; import org.elasticsearch.node.NodeValidationException; import java.io.File; import java.io.IOException; import java.util.UUID; import java.util.concurrent.TimeUnit; /** * Created by rishabh.goyal on 16/04/14. */ public class MockElasticsearchServer { private final Node node; private String DATA_DIRECTORY = UUID.randomUUID() .toString() + "/elasticsearch-data"; public MockElasticsearchServer(String directory) throws NodeValidationException { this.DATA_DIRECTORY = UUID.randomUUID() .toString() + "/" + directory; Settings settings = Settings.builder() .put("http.enabled", "false") .put("path.home", "target/" + DATA_DIRECTORY) .put("transport.type", "local") .build(); Stopwatch stopwatch = Stopwatch.createStarted(); node = new Node(settings).start(); System.out.printf("TimeTakenForEstart: %d%n", stopwatch.elapsed(TimeUnit.MILLISECONDS)); } public void refresh(final String index) { node.client() .admin() .indices() .refresh(new RefreshRequest().indices(index)) .actionGet(); } public Client getClient() { return node.client(); } public void shutdown() throws IOException { node.client() .admin() .indices() .delete(new DeleteIndexRequest("table-meta")); node.close(); deleteDataDirectory(); } private void deleteDataDirectory() throws IOException { System.out.println("Deleting DATA DIRECTORY"); FileUtils.deleteDirectory(new File(DATA_DIRECTORY)); } }