Examples with NodesUsageRequest org.opensearch.action.admin.cluster.node.usage.NodesUsageRequest used on opensource projects

Example 1 with NodesUsageRequest

use of org.opensearch.action.admin.cluster.node.usage.NodesUsageRequest in project OpenSearch by opensearch-project.

the class RestNodesUsageAction method prepareRequest.

@Override
protected RestChannelConsumer prepareRequest(RestRequest request, NodeClient client) throws IOException {
    String[] nodesIds = Strings.splitStringByCommaToArray(request.param("nodeId"));
    Set<String> metrics = Strings.tokenizeByCommaToSet(request.param("metric", "_all"));
    NodesUsageRequest nodesUsageRequest = new NodesUsageRequest(nodesIds);
    nodesUsageRequest.timeout(request.param("timeout"));
    if (metrics.size() == 1 && metrics.contains("_all")) {
        nodesUsageRequest.all();
    } else if (metrics.contains("_all")) {
        throw new IllegalArgumentException(String.format(Locale.ROOT, "request [%s] contains _all and individual metrics [%s]", request.path(), request.param("metric")));
    } else {
        nodesUsageRequest.clear();
        nodesUsageRequest.restActions(metrics.contains("rest_actions"));
        nodesUsageRequest.aggregations(metrics.contains("aggregations"));
    }
    return channel -> client.admin().cluster().nodesUsage(nodesUsageRequest, new RestBuilderListener<NodesUsageResponse>(channel) {

        @Override
        public RestResponse buildResponse(NodesUsageResponse response, XContentBuilder builder) throws Exception {
            builder.startObject();
            RestActions.buildNodesHeader(builder, channel.request(), response);
            builder.field("cluster_name", response.getClusterName().value());
            response.toXContent(builder, channel.request());
            builder.endObject();
            return new BytesRestResponse(RestStatus.OK, builder);
        }
    });
}