Changed the root rest endpoint ('/') to use cluster service

Instead of issuing a redundant cluster state request.

Closes #7899
This commit is contained in:
uboness 2014-09-30 15:55:03 +02:00
parent e53b2eede7
commit ddbeb910be
1 changed files with 40 additions and 48 deletions

View File

@ -22,16 +22,13 @@ package org.elasticsearch.rest.action.main;
import org.apache.lucene.util.Constants;
import org.elasticsearch.Build;
import org.elasticsearch.Version;
import org.elasticsearch.action.admin.cluster.state.ClusterStateRequest;
import org.elasticsearch.action.admin.cluster.state.ClusterStateResponse;
import org.elasticsearch.client.Client;
import org.elasticsearch.cluster.ClusterName;
import org.elasticsearch.cluster.ClusterService;
import org.elasticsearch.common.inject.Inject;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.unit.TimeValue;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.rest.*;
import org.elasticsearch.rest.action.support.RestResponseListener;
import static org.elasticsearch.rest.RestRequest.Method.GET;
import static org.elasticsearch.rest.RestRequest.Method.HEAD;
@ -43,32 +40,28 @@ public class RestMainAction extends BaseRestHandler {
private final Version version;
private final ClusterName clusterName;
private final ClusterService clusterService;
@Inject
public RestMainAction(Settings settings, Version version, RestController controller, ClusterName clusterName, Client client) {
public RestMainAction(Settings settings, Version version, RestController controller, ClusterName clusterName, Client client, ClusterService clusterService) {
super(settings, controller, client);
this.version = version;
this.clusterName = clusterName;
this.clusterService = clusterService;
controller.registerHandler(GET, "/", this);
controller.registerHandler(HEAD, "/", this);
}
@Override
public void handleRequest(final RestRequest request, final RestChannel channel, final Client client) {
ClusterStateRequest clusterStateRequest = new ClusterStateRequest();
clusterStateRequest.listenerThreaded(false);
clusterStateRequest.masterNodeTimeout(TimeValue.timeValueMillis(0));
clusterStateRequest.local(true);
clusterStateRequest.clear().blocks(true);
client.admin().cluster().state(clusterStateRequest, new RestResponseListener<ClusterStateResponse>(channel) {
@Override
public RestResponse buildResponse(ClusterStateResponse response) throws Exception {
public void handleRequest(final RestRequest request, RestChannel channel, final Client client) throws Exception {
RestStatus status = RestStatus.OK;
if (response.getState().blocks().hasGlobalBlock(RestStatus.SERVICE_UNAVAILABLE)) {
if (clusterService.state().blocks().hasGlobalBlock(RestStatus.SERVICE_UNAVAILABLE)) {
status = RestStatus.SERVICE_UNAVAILABLE;
}
if (request.method() == RestRequest.Method.HEAD) {
return new BytesRestResponse(status);
channel.sendResponse(new BytesRestResponse(status));
return;
}
XContentBuilder builder = channel.newBuilder();
@ -97,8 +90,7 @@ public class RestMainAction extends BaseRestHandler {
.endObject();
builder.field("tagline", "You Know, for Search");
builder.endObject();
return new BytesRestResponse(status, builder);
}
});
channel.sendResponse(new BytesRestResponse(status, builder));
}
}