-
Notifications
You must be signed in to change notification settings - Fork 369
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Add stats api. The api provides basic information on the plugin usage and performance, including, - overall status of the plugin health. - statistics on the feature stores. - statistics on the internal caches used by the plugin. * Refactor the getRequest method. * Fix case-change in IndexMetadata as part of 7.8 upgrade. * Resolve Review comments - Refactor to remove the LTRStats dependency from RestLTRStats - Remove client and clusterService instance variables from LtrQueryParserPlugin. - Minor Refactoring * Remove StoreUtils - move the operations inside the stat suppliers. * Use aggregates with multi search requests. * add rest-api spec and test. * add stats section into advanced functionality doc. * Addressed review comments.
- Loading branch information
Rabi Panda
authored
Aug 28, 2020
1 parent
e6a2877
commit 0455051
Showing
20 changed files
with
1,542 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
180 changes: 180 additions & 0 deletions
180
src/main/java/com/o19s/es/ltr/action/LTRStatsAction.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,180 @@ | ||
| package com.o19s.es.ltr.action; | ||
|
|
||
| import org.elasticsearch.action.ActionRequestBuilder; | ||
| import org.elasticsearch.action.ActionType; | ||
| import org.elasticsearch.action.FailedNodeException; | ||
| import org.elasticsearch.action.support.nodes.BaseNodeRequest; | ||
| import org.elasticsearch.action.support.nodes.BaseNodeResponse; | ||
| import org.elasticsearch.action.support.nodes.BaseNodesRequest; | ||
| import org.elasticsearch.action.support.nodes.BaseNodesResponse; | ||
| import org.elasticsearch.client.ElasticsearchClient; | ||
| import org.elasticsearch.cluster.ClusterName; | ||
| import org.elasticsearch.cluster.node.DiscoveryNode; | ||
| import org.elasticsearch.common.io.stream.StreamInput; | ||
| import org.elasticsearch.common.io.stream.StreamOutput; | ||
| import org.elasticsearch.common.xcontent.ToXContent; | ||
| import org.elasticsearch.common.xcontent.ToXContentFragment; | ||
| import org.elasticsearch.common.xcontent.XContentBuilder; | ||
|
|
||
| import java.io.IOException; | ||
| import java.util.HashSet; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.Set; | ||
|
|
||
| public class LTRStatsAction extends ActionType<LTRStatsAction.LTRStatsNodesResponse> { | ||
| public static final String NAME = "cluster:admin/ltr/stats"; | ||
| public static final LTRStatsAction INSTANCE = new LTRStatsAction(); | ||
|
|
||
| public LTRStatsAction() { | ||
| super(NAME, LTRStatsNodesResponse::new); | ||
| } | ||
|
|
||
| public static class LTRStatsRequestBuilder | ||
| extends ActionRequestBuilder<LTRStatsNodesRequest, LTRStatsNodesResponse> { | ||
| private static final String[] nodeIds = null; | ||
|
|
||
| public LTRStatsRequestBuilder(ElasticsearchClient client) { | ||
| super(client, INSTANCE, new LTRStatsNodesRequest(nodeIds)); | ||
| } | ||
| } | ||
|
|
||
| public static class LTRStatsNodeRequest extends BaseNodeRequest { | ||
| private final LTRStatsNodesRequest nodesRequest; | ||
|
|
||
| public LTRStatsNodeRequest(LTRStatsNodesRequest nodesRequest) { | ||
| this.nodesRequest = nodesRequest; | ||
| } | ||
|
|
||
| public LTRStatsNodeRequest(StreamInput in) throws IOException { | ||
| super(in); | ||
| nodesRequest = new LTRStatsNodesRequest(in); | ||
| } | ||
|
|
||
| public LTRStatsNodesRequest getLTRStatsNodesRequest() { | ||
| return nodesRequest; | ||
| } | ||
|
|
||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| super.writeTo(out); | ||
| nodesRequest.writeTo(out); | ||
| } | ||
| } | ||
|
|
||
| public static class LTRStatsNodeResponse extends BaseNodeResponse implements ToXContentFragment { | ||
|
|
||
| private final Map<String, Object> statsMap; | ||
|
|
||
| LTRStatsNodeResponse(StreamInput in) throws IOException { | ||
| super(in); | ||
| this.statsMap = in.readMap(StreamInput::readString, StreamInput::readGenericValue); | ||
| } | ||
|
|
||
| LTRStatsNodeResponse(DiscoveryNode node, Map<String, Object> statsToValues) { | ||
| super(node); | ||
| this.statsMap = statsToValues; | ||
| } | ||
|
|
||
| public Map<String, Object> getStatsMap() { | ||
| return statsMap; | ||
| } | ||
|
|
||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| super.writeTo(out); | ||
| out.writeMap(statsMap, StreamOutput::writeString, StreamOutput::writeGenericValue); | ||
| } | ||
|
|
||
| public XContentBuilder toXContent(XContentBuilder builder, ToXContent.Params params) throws IOException { | ||
| for (Map.Entry<String, Object> stat : statsMap.entrySet()) { | ||
| builder.field(stat.getKey(), stat.getValue()); | ||
| } | ||
|
|
||
| return builder; | ||
| } | ||
| } | ||
|
|
||
| public static class LTRStatsNodesRequest extends BaseNodesRequest<LTRStatsNodesRequest> { | ||
|
|
||
| public static final String ALL_STATS_KEY = "_all"; | ||
|
|
||
| private Set<String> statsToBeRetrieved; | ||
|
|
||
| public LTRStatsNodesRequest(StreamInput in) throws IOException { | ||
| super(in); | ||
| statsToBeRetrieved = in.readSet(StreamInput::readString); | ||
| } | ||
|
|
||
| public LTRStatsNodesRequest(String[] nodeIds) { | ||
| super(nodeIds); | ||
| statsToBeRetrieved = new HashSet<>(); | ||
| } | ||
|
|
||
| public void setStatsToBeRetrieved(Set<String> statsToBeRetrieved) { | ||
| this.statsToBeRetrieved = statsToBeRetrieved; | ||
| } | ||
|
|
||
| public Set<String> getStatsToBeRetrieved() { | ||
| return statsToBeRetrieved; | ||
| } | ||
|
|
||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| super.writeTo(out); | ||
| out.writeStringCollection(statsToBeRetrieved); | ||
| } | ||
| } | ||
|
|
||
| public static class LTRStatsNodesResponse extends BaseNodesResponse<LTRStatsNodeResponse> implements ToXContent { | ||
| private static final String NODES_KEY = "nodes"; | ||
| private final Map<String, Object> clusterStats; | ||
|
|
||
| public LTRStatsNodesResponse(StreamInput in) throws IOException { | ||
| super(in); | ||
| clusterStats = in.readMap(); | ||
| } | ||
|
|
||
| public LTRStatsNodesResponse(ClusterName clusterName, List<LTRStatsNodeResponse> nodeResponses, | ||
| List<FailedNodeException> failures, Map<String, Object> clusterStats) { | ||
| super(clusterName, nodeResponses, failures); | ||
| this.clusterStats = clusterStats; | ||
| } | ||
|
|
||
| public Map<String, Object> getClusterStats() { | ||
| return clusterStats; | ||
| } | ||
|
|
||
| @Override | ||
| protected List<LTRStatsNodeResponse> readNodesFrom(StreamInput in) throws IOException { | ||
| return in.readList(LTRStatsNodeResponse::new); | ||
| } | ||
|
|
||
| @Override | ||
| protected void writeNodesTo(StreamOutput out, List<LTRStatsNodeResponse> nodeResponses) throws IOException { | ||
| out.writeList(nodeResponses); | ||
| } | ||
|
|
||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| super.writeTo(out); | ||
| out.writeMap(clusterStats); | ||
| } | ||
|
|
||
| @Override | ||
| public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException { | ||
| for (Map.Entry<String, Object> clusterStat : clusterStats.entrySet()) { | ||
| builder.field(clusterStat.getKey(), clusterStat.getValue()); | ||
| } | ||
|
|
||
| builder.startObject(NODES_KEY); | ||
| for (LTRStatsNodeResponse ltrStats : getNodes()) { | ||
| builder.startObject(ltrStats.getNode().getId()); | ||
| ltrStats.toXContent(builder, params); | ||
| builder.endObject(); | ||
| } | ||
| builder.endObject(); | ||
| return builder; | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.