add index parameter

Author: javanna

test/external-modules/seek-tracking-directory/src/main/java/org/elasticsearch/test/seektracker/RestSeekCountAction.java renamed to test/external-modules/seek-tracking-directory/src/main/java/org/elasticsearch/test/seektracker/RestSeekStatsAction.java

@@ -9,14 +9,15 @@
 package org.elasticsearch.test.seektracker;
 
 import org.elasticsearch.client.internal.node.NodeClient;
+import org.elasticsearch.common.Strings;
 import org.elasticsearch.rest.BaseRestHandler;
 import org.elasticsearch.rest.RestHandler;
 import org.elasticsearch.rest.RestRequest;
 import org.elasticsearch.rest.action.RestToXContentListener;
 
 import java.util.List;
 
-public class RestSeekCountAction extends BaseRestHandler {
+public class RestSeekStatsAction extends BaseRestHandler {
 
     @Override
     public String getName() {
@@ -25,11 +26,14 @@ public String getName() {
 
     @Override
     public List<Route> routes() {
-        return List.of(new RestHandler.Route(RestRequest.Method.GET, "/_seek_stats"));
+        return List.of(new RestHandler.Route(RestRequest.Method.GET, "/_seek_stats"),
+            new RestHandler.Route(RestRequest.Method.GET, "/{index}/_seek_stats"));
     }
 
     @Override
     protected RestChannelConsumer prepareRequest(RestRequest request, NodeClient client) {
-        return channel -> client.executeLocally(SeekStatsAction.INSTANCE, new SeekStatsRequest(), new RestToXContentListener<>(channel));
+        String[] indices = request.paramAsStringArray("index", Strings.EMPTY_ARRAY);
+        SeekStatsRequest seekStatsRequest = new SeekStatsRequest(indices);
+        return channel -> client.executeLocally(SeekStatsAction.INSTANCE, seekStatsRequest, new RestToXContentListener<>(channel));
     }
 }

test/external-modules/seek-tracking-directory/src/main/java/org/elasticsearch/test/seektracker/SeekStatsRequest.java

@@ -10,18 +10,34 @@
 
 import org.elasticsearch.action.ActionRequestValidationException;
 import org.elasticsearch.action.support.nodes.BaseNodesRequest;
+import org.elasticsearch.common.Strings;
 import org.elasticsearch.common.io.stream.StreamInput;
+import org.elasticsearch.common.io.stream.StreamOutput;
 
 import java.io.IOException;
 
 public class SeekStatsRequest extends BaseNodesRequest<SeekStatsRequest> {
 
-    public SeekStatsRequest(String... nodeIds) {
-        super(nodeIds);
+    private final String[] indices;
+
+    public SeekStatsRequest(String... indices) {
+        super(Strings.EMPTY_ARRAY);
+        this.indices = indices;
     }
 
     public SeekStatsRequest(StreamInput in) throws IOException {
         super(in);
+        this.indices = in.readStringArray();
+    }
+
+    @Override
+    public void writeTo(StreamOutput out) throws IOException {
+        super.writeTo(out);
+        out.writeStringArray(indices);
+    }
+
+    public String[] getIndices() {
+        return indices;
     }
 
     @Override

test/external-modules/seek-tracking-directory/src/main/java/org/elasticsearch/test/seektracker/SeekTrackerPlugin.java

@@ -81,7 +81,7 @@ public List<RestHandler> getRestHandlers(
         IndexNameExpressionResolver indexNameExpressionResolver,
         Supplier<DiscoveryNodes> nodesInCluster
     ) {
-        return Collections.singletonList(new RestSeekCountAction());
+        return Collections.singletonList(new RestSeekStatsAction());
     }
 
     @Override

test/external-modules/seek-tracking-directory/src/main/java/org/elasticsearch/test/seektracker/TransportSeekStatsAction.java

@@ -70,8 +70,15 @@ protected SeekStats newNodeResponse(StreamInput in, DiscoveryNode node) throws I
     @Override
     protected SeekStats nodeOperation(SeekStatsRequest request, Task task) {
         Map<String, Long> seeks = new HashMap<>();
-        for (Map.Entry<String, LongAdder> entry : seekStatsService.seeks.entrySet()) {
-            seeks.put(entry.getKey(), entry.getValue().longValue());
+        if (request.getIndices().length == 0) {
+            for (Map.Entry<String, LongAdder> entry : seekStatsService.seeks.entrySet()) {
+                seeks.put(entry.getKey(), entry.getValue().longValue());
+            }
+        } else {
+            for (String index : request.getIndices()) {
+                LongAdder longAdder = seekStatsService.seeks.get(index);
+                seeks.put(index, longAdder.longValue());
+            }
         }
         return new SeekStats(clusterService.localNode(), seeks);
     }