Java ClusterHealthStatus類代碼示例

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
@Override
public String checkStatus() throws Exception {

  final ClusterHealthResponse healthResponse = esClient
      .admin().cluster().prepareHealth().execute().actionGet();

  if (healthResponse.isTimedOut()) {
    return EsStatus.TIMEOUT.toString();

  } else if (ClusterHealthStatus.RED.equals(healthResponse.getStatus())) {
    logger.warn("Elastic search health status is reported RED");
    return EsStatus.ERROR.toString();

  } else if (ClusterHealthStatus.GREEN.equals(healthResponse.getStatus())) {
    return EsStatus.SUCCESS.toString();

  } else {
    logger.warn("Elastic search health status is unknown");
    return EsStatus.UNKNOWN.toString();
  }
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
/**
 * Ensures the cluster has a green state via the cluster health API. This method will also wait for relocations.
 * It is useful to ensure that all action on the cluster have finished and all shards that were currently relocating
 * are now allocated and started.
 *
 * @param timeout
 *            time out value to set on {@link org.elasticsearch.action.admin.cluster.health.ClusterHealthRequest}
 */
public ClusterHealthStatus ensureGreen(final TimeValue timeout, final String... indices) {
    final ClusterHealthResponse actionGet = client()
            .admin()
            .cluster()
            .health(Requests.clusterHealthRequest(indices).timeout(timeout).waitForGreenStatus().waitForEvents(Priority.LANGUID)
                    .waitForRelocatingShards(0)).actionGet();
    if (actionGet.isTimedOut()) {
        logger.info("ensureGreen timed out, cluster state:\n{}\n{}", client().admin().cluster().prepareState().get().getState()
                .prettyPrint(), client().admin().cluster().preparePendingClusterTasks().get().prettyPrint());
        assertThat("timed out waiting for green state", actionGet.isTimedOut(), equalTo(false));
    }
    assertThat(actionGet.getStatus(), equalTo(ClusterHealthStatus.GREEN));
    logger.debug("indices {} are green", indices.length == 0 ? "[_all]" : indices);
    return actionGet.getStatus();
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
/**
 * Waits for all relocating shards to become active and the cluster has reached the given health status
 * using the cluster health API.
 */
public ClusterHealthStatus waitForRelocation(final ClusterHealthStatus status) {
    final ClusterHealthRequest request = Requests.clusterHealthRequest().waitForRelocatingShards(0);
    if (status != null) {
        request.waitForStatus(status);
    }
    final ClusterHealthResponse actionGet = client().admin().cluster().health(request).actionGet();
    if (actionGet.isTimedOut()) {
        logger.info("waitForRelocation timed out (status={}), cluster state:\n{}\n{}", status, client().admin().cluster()
                .prepareState().get().getState().prettyPrint(), client().admin().cluster().preparePendingClusterTasks().get()
                .prettyPrint());
        assertThat("timed out waiting for relocation", actionGet.isTimedOut(), equalTo(false));
    }
    if (status != null) {
        assertThat(actionGet.getStatus(), equalTo(status));
    }
    return actionGet.getStatus();
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
@Test
public void testSimpleMoreLikeThis() throws Exception {
    logger.info("Creating index test");
    assertAcked(prepareCreate("test").addMapping("type1",
            jsonBuilder().startObject().startObject("type1").startObject("properties")
                    .startObject("text").field("type", "string").endObject()
                    .endObject().endObject().endObject()));

    logger.info("Running Cluster Health");
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));

    logger.info("Indexing...");
    client().index(indexRequest("test").type("type1").id("1").source(jsonBuilder().startObject().field("text", "lucene").endObject())).actionGet();
    client().index(indexRequest("test").type("type1").id("2").source(jsonBuilder().startObject().field("text", "lucene release").endObject())).actionGet();
    client().admin().indices().refresh(refreshRequest()).actionGet();

    logger.info("Running moreLikeThis");
    SearchResponse mltResponse = httpClient.moreLikeThis(moreLikeThisRequest("test").type("type1").id("1").minTermFreq(1).minDocFreq(1)).get();
    assertHitCount(mltResponse, 1l);
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
@Test
public void testSimpleMoreLikeOnLongField() throws Exception {
    logger.info("Creating index test");
    assertAcked(prepareCreate("test").addMapping("type1", "some_long", "type=long"));
    logger.info("Running Cluster Health");
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));

    logger.info("Indexing...");
    client().index(indexRequest("test").type("type1").id("1").source(jsonBuilder().startObject().field("some_long", 1367484649580l).endObject())).actionGet();
    client().index(indexRequest("test").type("type2").id("2").source(jsonBuilder().startObject().field("some_long", 0).endObject())).actionGet();
    client().index(indexRequest("test").type("type1").id("3").source(jsonBuilder().startObject().field("some_long", -666).endObject())).actionGet();


    client().admin().indices().refresh(refreshRequest()).actionGet();

    logger.info("Running moreLikeThis");
    SearchResponse mltResponse = httpClient.moreLikeThis(moreLikeThisRequest("test").type("type1").id("1").minTermFreq(1).minDocFreq(1)).get();
    assertHitCount(mltResponse, 0l);
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
@Test
public void testSimpleMoreLikeThisIds() throws Exception {
    logger.info("Creating index test");
    assertAcked(prepareCreate("test").addMapping("type1",
            jsonBuilder().startObject().startObject("type1").startObject("properties")
                    .startObject("text").field("type", "string").endObject()
                    .endObject().endObject().endObject()));

    logger.info("Running Cluster Health");
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));

    logger.info("Indexing...");
    List<IndexRequestBuilder> builders = new ArrayList<>();
    builders.add(client().prepareIndex("test", "type1").setSource("text", "lucene").setId("1"));
    builders.add(client().prepareIndex("test", "type1").setSource("text", "lucene release").setId("2"));
    builders.add(client().prepareIndex("test", "type1").setSource("text", "apache lucene").setId("3"));
    indexRandom(true, builders);

    logger.info("Running MoreLikeThis");
    MoreLikeThisQueryBuilder queryBuilder = QueryBuilders.moreLikeThisQuery("text").ids("1").include(true).minTermFreq(1).minDocFreq(1);
    SearchResponse mltResponse = httpClient.search(Requests.searchRequest("test").types("type1").source(new SearchSourceBuilder().query(queryBuilder))).get();
    assertHitCount(mltResponse, 3l);
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
protected void checkServerStatus() {
    ClusterHealthStatus status = getHealthStatus();

    // Check the current status of the ES cluster.
    if (ClusterHealthStatus.RED.equals(status)) {
        LOGGER.info("ES cluster status is " + status + ". Waiting for ES recovery.");

        // Waits at most 30 seconds to make sure the cluster health is at least yellow.
        getClient().admin().cluster().prepareHealth()
                .setWaitForYellowStatus()
                .setTimeout("30s")
                .execute().actionGet();
    }

    // Check the cluster health for a final time.
    status = getHealthStatus();
    LOGGER.info("ES cluster status is " + status);

    // If we are still in red status, then we cannot proceed.
    if (ClusterHealthStatus.RED.equals(status)) {
        throw new RuntimeException("ES cluster health status is RED. Server is not able to start.");
    }

}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
public void init(final ActionListener<Void> listener) {
    client.admin().cluster().prepareHealth().setWaitForYellowStatus()
            .execute(new ActionListener<ClusterHealthResponse>() {
                @Override
                public void onResponse(final ClusterHealthResponse response) {
                    if (response.getStatus() == ClusterHealthStatus.RED) {
                        listener.onFailure(new AuthException(
                                RestStatus.SERVICE_UNAVAILABLE,
                                "This cluster is not ready."));
                    } else {
                        createConstraintIndexIfNotExist(listener);
                    }
                }

                @Override
                public void onFailure(final Throwable e) {
                    listener.onFailure(e);
                }
            });
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
@BeforeMethod
protected void createIndex() {
    logger.info("creating index [" + INDEX + "]");
    CreateIndexRequest createIndexRequest = createIndexRequest(INDEX);
    String settings = getSettings();
    if (settings != null)
        createIndexRequest.settings(getSettings());
    String mapping = getMapping();
    if (mapping != null)
        createIndexRequest.mapping(TYPE, getMapping());
    assertThat("Index creation", node.client().admin().indices().create(createIndexRequest).actionGet().isAcknowledged());
    logger.info("Running Cluster Health");
    ClusterHealthResponse clusterHealth = node.client().admin().cluster().health(clusterHealthRequest().waitForGreenStatus()).actionGet();
    logger.info("Done Cluster Health, status " + clusterHealth.getStatus());
    assertThat(clusterHealth.isTimedOut(), equalTo(false));
    assertThat(clusterHealth.getStatus(), equalTo(ClusterHealthStatus.GREEN));
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
protected void checkServerStatus() {
    ClusterHealthStatus status = getHealthStatus();
 
    // Check the current status of ES.
    if (ClusterHealthStatus.RED.equals(status)) {
        LOGGER.info("ES status is " + status + ". Waiting for ES recovery.");
 
        // Waits at most 10 minutes to make sure the cluster health is at least yellow.
        getClient().admin().cluster().prepareHealth()
                .setWaitForYellowStatus()
                .setTimeout("600s")
                .execute().actionGet();
    }
 
    // Check the cluster health for a final time.
    status = getHealthStatus();
    LOGGER.info("ES cluster status is " + status);
 
    // If we are still in red status, then we cannot proceed.
    if (ClusterHealthStatus.RED.equals(status)) {
        throw new RuntimeException("ES health status is RED. Server is not able to start.");
    }
 
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
/**
   * Waits for all relocating shards to become active and the cluster has reached the given health status
   * using the cluster health API.
   */
  public ClusterHealthStatus waitForRelocation(ClusterHealthStatus status) {
    ClusterHealthRequest request = Requests.clusterHealthRequest().waitForRelocatingShards(0);
    if (status != null) {
      request.waitForStatus(status);
    }
    ClusterHealthResponse actionGet = adminClient.cluster()
        .health(request).actionGet();
    if (actionGet.isTimedOut()) {
//      logger.info("waitForRelocation timed out (status={}), cluster state:\n{}\n{}", status, adminClient.cluster().prepareState().get().getState().prettyPrint(), adminClient.cluster().preparePendingClusterTasks().get().prettyPrint());
      assertThat("timed out waiting for relocation", actionGet.isTimedOut(), equalTo(false));
    }
    if (status != null) {
      assertThat(actionGet.getStatus(), equalTo(status));
    }
    return actionGet.getStatus();
  }
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
/**
 * Ensures the cluster has a yellow state via the cluster health API.
 */
public ClusterHealthStatus ensureYellow(final String... indices) {
    final ClusterHealthResponse actionGet = client()
            .admin()
            .cluster()
            .health(Requests.clusterHealthRequest(indices).waitForRelocatingShards(0).waitForYellowStatus()
                    .waitForEvents(Priority.LANGUID)).actionGet();
    if (actionGet.isTimedOut()) {
        logger.info("ensureYellow timed out, cluster state:\n{}\n{}", client().admin().cluster().prepareState().get().getState()
                .prettyPrint(), client().admin().cluster().preparePendingClusterTasks().get().prettyPrint());
        assertThat("timed out waiting for yellow", actionGet.isTimedOut(), equalTo(false));
    }
    logger.debug("indices {} are yellow", indices.length == 0 ? "[_all]" : indices);
    return actionGet.getStatus();
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
@Test
public void redResponseMapsToDown() {
	PlainActionFuture<ClusterHealthResponse> responseFuture = new PlainActionFuture<ClusterHealthResponse>();
	responseFuture.onResponse(new StubClusterHealthResponse(ClusterHealthStatus.RED));
	given(this.cluster.health(any(ClusterHealthRequest.class)))
			.willReturn(responseFuture);
	assertThat(this.indicator.health().getStatus(), is(Status.DOWN));
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
@Test
public void yellowResponseMapsToUp() {
	PlainActionFuture<ClusterHealthResponse> responseFuture = new PlainActionFuture<ClusterHealthResponse>();
	responseFuture
			.onResponse(new StubClusterHealthResponse(ClusterHealthStatus.YELLOW));
	given(this.cluster.health(any(ClusterHealthRequest.class)))
			.willReturn(responseFuture);
	assertThat(this.indicator.health().getStatus(), is(Status.UP));
}
 

import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus; //導入依賴的package包/類
private StubClusterHealthResponse(ClusterHealthStatus status) {
	super("test-cluster", new String[0],
			new ClusterState(null, 0, null, RoutingTable.builder().build(),
					DiscoveryNodes.builder().build(),
					ClusterBlocks.builder().build(), null));
	this.status = status;
}