From 1645e4230f2a0cf8b2021ef77f38f337101e48eb Mon Sep 17 00:00:00 2001 From: Andrew Raines Date: Thu, 4 Jul 2013 15:40:09 -0500 Subject: [PATCH] Add /_cat/nodes. --- .../rest/action/RestActionModule.java | 2 + .../rest/action/cat/RestNodesAction.java | 152 ++++++++++++++++++ 2 files changed, 154 insertions(+) create mode 100644 src/main/java/org/elasticsearch/rest/action/cat/RestNodesAction.java diff --git a/src/main/java/org/elasticsearch/rest/action/RestActionModule.java b/src/main/java/org/elasticsearch/rest/action/RestActionModule.java index 3c8c2938c48..5e3ed9e4e6e 100644 --- a/src/main/java/org/elasticsearch/rest/action/RestActionModule.java +++ b/src/main/java/org/elasticsearch/rest/action/RestActionModule.java @@ -68,6 +68,7 @@ import org.elasticsearch.rest.action.admin.indices.warmer.get.RestGetWarmerActio import org.elasticsearch.rest.action.admin.indices.warmer.put.RestPutWarmerAction; import org.elasticsearch.rest.action.bulk.RestBulkAction; import org.elasticsearch.rest.action.cat.RestMasterAction; +import org.elasticsearch.rest.action.cat.RestNodesAction; import org.elasticsearch.rest.action.cat.RestShardsAction; import org.elasticsearch.rest.action.count.RestCountAction; import org.elasticsearch.rest.action.delete.RestDeleteAction; @@ -185,5 +186,6 @@ public class RestActionModule extends AbstractModule { bind(RestShardsAction.class).asEagerSingleton(); bind(RestMasterAction.class).asEagerSingleton(); + bind(RestNodesAction.class).asEagerSingleton(); } } diff --git a/src/main/java/org/elasticsearch/rest/action/cat/RestNodesAction.java b/src/main/java/org/elasticsearch/rest/action/cat/RestNodesAction.java new file mode 100644 index 00000000000..afe0bb10bde --- /dev/null +++ b/src/main/java/org/elasticsearch/rest/action/cat/RestNodesAction.java @@ -0,0 +1,152 @@ +/* + * Licensed to ElasticSearch and Shay Banon under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. ElasticSearch licenses this + * file to you 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 + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * 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 org.elasticsearch.rest.action.cat; + +import org.elasticsearch.action.ActionListener; +import org.elasticsearch.action.admin.cluster.node.info.NodeInfo; +import org.elasticsearch.action.admin.cluster.node.info.NodesInfoRequest; +import org.elasticsearch.action.admin.cluster.node.info.NodesInfoResponse; +import org.elasticsearch.action.admin.cluster.node.stats.NodeStats; +import org.elasticsearch.action.admin.cluster.node.stats.NodesStatsRequest; +import org.elasticsearch.action.admin.cluster.node.stats.NodesStatsResponse; +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.node.DiscoveryNode; +import org.elasticsearch.common.Table; +import org.elasticsearch.common.inject.Inject; +import org.elasticsearch.common.settings.Settings; +import org.elasticsearch.common.transport.InetSocketTransportAddress; +import org.elasticsearch.rest.*; +import org.elasticsearch.rest.action.support.RestTable; + +import java.io.IOException; + +import static org.elasticsearch.rest.RestRequest.Method.GET; + +public class RestNodesAction extends BaseRestHandler { + + @Inject + public RestNodesAction(Settings settings, Client client, RestController controller) { + super(settings, client); + controller.registerHandler(GET, "/_cat/nodes", this); + } + + @Override + public void handleRequest(final RestRequest request, final RestChannel channel) { + final ClusterStateRequest clusterStateRequest = new ClusterStateRequest(); + clusterStateRequest.filterMetaData(true); + clusterStateRequest.local(request.paramAsBoolean("local", clusterStateRequest.local())); + clusterStateRequest.masterNodeTimeout(request.paramAsTime("master_timeout", clusterStateRequest.masterNodeTimeout())); + + client.admin().cluster().state(clusterStateRequest, new ActionListener() { + @Override + public void onResponse(final ClusterStateResponse clusterStateResponse) { + NodesInfoRequest nodesInfoRequest = new NodesInfoRequest(); + nodesInfoRequest.clear().jvm(true).os(true); + client.admin().cluster().nodesInfo(nodesInfoRequest, new ActionListener() { + @Override + public void onResponse(final NodesInfoResponse nodesInfoResponse) { + NodesStatsRequest nodesStatsRequest = new NodesStatsRequest(); + nodesStatsRequest.clear().jvm(true); + client.admin().cluster().nodesStats(nodesStatsRequest, new ActionListener() { + @Override + public void onResponse(NodesStatsResponse nodesStatsResponse) { + try { + channel.sendResponse(RestTable.buildResponse(buildTable(clusterStateResponse, nodesInfoResponse, nodesStatsResponse), request, channel)); + } catch (Throwable e) { + onFailure(e); + } + } + + @Override + public void onFailure(Throwable e) { + try { + channel.sendResponse(new XContentThrowableRestResponse(request, e)); + } catch (IOException e1) { + logger.error("Failed to send failure response", e1); + } + } + }); + } + @Override + public void onFailure(Throwable e) { + try { + channel.sendResponse(new XContentThrowableRestResponse(request, e)); + } catch (IOException e1) { + logger.error("Failed to send failure response", e1); + } + } + }); + } + + @Override + public void onFailure(Throwable e) { + try { + channel.sendResponse(new XContentThrowableRestResponse(request, e)); + } catch (IOException e1) { + logger.error("Failed to send failure response", e1); + } + } + }); + } + + private Table buildTable(ClusterStateResponse state, NodesInfoResponse nodesInfo, NodesStatsResponse nodesStats) { + Table table = new Table(); + table.startHeaders(); + table.addCell("id"); + table.addCell("ip"); + table.addCell("port"); + + // Considering supporting params like jvm, os, http to turn various columns on/off + table.addCell("jdk"); + table.addCell("heapmax", "text-align:right;"); + table.addCell("memmax", "text-align:right;"); + + table.addCell("uptime", "text-align:right;"); + table.addCell("data/client"); + table.addCell("master"); + table.addCell("node"); + table.endHeaders(); + + for (DiscoveryNode node : state.getState().nodes()) { + String masterId = state.getState().nodes().masterNodeId(); + NodeInfo info = nodesInfo.getNodesMap().get(node.id()); + NodeStats stats = nodesStats.getNodesMap().get(node.id()); + + table.startRow(); + + table.addCell(node.id()); + table.addCell(((InetSocketTransportAddress) node.address()).address().getAddress().getHostAddress()); + table.addCell(((InetSocketTransportAddress) node.address()).address().getPort()); + table.addCell(info.getJvm().version()); + table.addCell(info.getJvm().mem().heapMax()); + table.addCell(info.getOs().mem() == null ? null : info.getOs().mem().total()); // sigar fails to load in IntelliJ + table.addCell(stats.getJvm().uptime()); + table.addCell(node.clientNode() ? "c" : node.dataNode() ? "d" : null); + table.addCell(masterId.equals(node.id()) ? "*" : node.masterNode() ? "m" : null); + table.addCell(node.name()); + + table.endRow(); + } + + return table; + } +}