Run XPack usage actions on local node (#122933)

These actions solely need the cluster state, they can run on any node.

Relates #101805

Author: nielsbauman

docs/changelog/122933.yaml

@@ -0,0 +1,5 @@
+pr: 122933
+summary: Run XPack usage actions on local node
+area: Stats
+type: enhancement
+issues: []

server/src/main/java/org/elasticsearch/action/support/local/TransportLocalClusterStateAction.java

@@ -66,6 +66,9 @@ protected abstract void localClusterStateOperation(Task task, Request request, C
 
     @Override
     protected final void doExecute(Task task, Request request, ActionListener<Response> listener) {
+        if (task != null) {
+            request.setParentTask(clusterService.localNode().getId(), task.getId());
+        }
         final var state = clusterService.state();
         final var clusterBlockException = checkBlock(request, state);
         if (clusterBlockException != null) {

x-pack/plugin/analytics/src/main/java/org/elasticsearch/xpack/analytics/action/AnalyticsUsageTransportAction.java

@@ -38,7 +38,7 @@ public AnalyticsUsageTransportAction(
     }
 
     @Override
-    protected void masterOperation(
+    protected void localClusterStateOperation(
         Task task,
         XPackUsageRequest request,
         ClusterState state,

x-pack/plugin/analytics/src/test/java/org/elasticsearch/xpack/analytics/action/AnalyticsInfoTransportActionTests.java

@@ -73,7 +73,7 @@ public void testAvailable() throws Exception {
             client
         );
         PlainActionFuture<XPackUsageFeatureResponse> future = new PlainActionFuture<>();
-        usageAction.masterOperation(task, null, clusterState, future);
+        usageAction.localClusterStateOperation(task, null, clusterState, future);
         XPackFeatureUsage usage = future.get().getUsage();
         assertThat(usage.available(), is(true));
 
@@ -100,7 +100,7 @@ public void testEnabled() throws Exception {
             client
         );
         PlainActionFuture<XPackUsageFeatureResponse> future = new PlainActionFuture<>();
-        usageAction.masterOperation(task, null, clusterState, future);
+        usageAction.localClusterStateOperation(task, null, clusterState, future);
         XPackFeatureUsage usage = future.get().getUsage();
         assertTrue(usage.enabled());
 

x-pack/plugin/ccr/src/main/java/org/elasticsearch/xpack/ccr/CCRUsageTransportAction.java

@@ -48,7 +48,7 @@ public CCRUsageTransportAction(
     }
 
     @Override
-    protected void masterOperation(
+    protected void localClusterStateOperation(
         Task task,
         XPackUsageRequest request,
         ClusterState state,

x-pack/plugin/ccr/src/test/java/org/elasticsearch/xpack/ccr/CCRInfoTransportActionTests.java

@@ -152,7 +152,7 @@ public void testUsageStats() throws Exception {
             licenseState
         );
         PlainActionFuture<XPackUsageFeatureResponse> future = new PlainActionFuture<>();
-        usageAction.masterOperation(null, null, clusterState, future);
+        usageAction.localClusterStateOperation(null, null, clusterState, future);
         CCRInfoTransportAction.Usage ccrUsage = (CCRInfoTransportAction.Usage) future.get().getUsage();
         assertThat(ccrUsage.enabled(), equalTo(true));
         assertThat(ccrUsage.available(), equalTo(false));

x-pack/plugin/core/src/internalClusterTest/java/org/elasticsearch/xpack/core/rest/action/XPackUsageRestCancellationIT.java

@@ -150,7 +150,7 @@ public BlockingXPackUsageAction(
         }
 
         @Override
-        protected void masterOperation(
+        protected void localClusterStateOperation(
             Task task,
             XPackUsageRequest request,
             ClusterState state,
@@ -185,7 +185,7 @@ public NonBlockingXPackUsageAction(
         }
 
         @Override
-        protected void masterOperation(
+        protected void localClusterStateOperation(
             Task task,
             XPackUsageRequest request,
             ClusterState state,

x-pack/plugin/core/src/main/java/org/elasticsearch/protocol/xpack/XPackUsageRequest.java

@@ -7,7 +7,7 @@
 package org.elasticsearch.protocol.xpack;
 
 import org.elasticsearch.action.ActionRequestValidationException;
-import org.elasticsearch.action.support.master.MasterNodeRequest;
+import org.elasticsearch.action.support.local.LocalClusterStateRequest;
 import org.elasticsearch.common.io.stream.StreamInput;
 import org.elasticsearch.core.TimeValue;
 import org.elasticsearch.tasks.CancellableTask;
@@ -17,14 +17,14 @@
 import java.io.IOException;
 import java.util.Map;
 
-public class XPackUsageRequest extends MasterNodeRequest<XPackUsageRequest> {
+public class XPackUsageRequest extends LocalClusterStateRequest {
 
     public XPackUsageRequest(TimeValue masterNodeTimeout) {
         super(masterNodeTimeout);
     }
 
     public XPackUsageRequest(StreamInput in) throws IOException {
-        super(in);
+        super(in, false);
     }
 
     @Override

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/HealthApiUsageTransportAction.java

@@ -42,7 +42,7 @@ public HealthApiUsageTransportAction(
     }
 
     @Override
-    protected void masterOperation(
+    protected void localClusterStateOperation(
         Task task,
         XPackUsageRequest request,
         ClusterState state,

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/RemoteClusterUsageTransportAction.java

@@ -23,6 +23,8 @@
 
 public class RemoteClusterUsageTransportAction extends XPackUsageFeatureTransportAction {
 
+    private TransportService transportService;
+
     @Inject
     public RemoteClusterUsageTransportAction(
         TransportService transportService,
@@ -31,10 +33,11 @@ public RemoteClusterUsageTransportAction(
         ActionFilters actionFilters
     ) {
         super(XPackUsageFeatureAction.REMOTE_CLUSTERS.name(), transportService, clusterService, threadPool, actionFilters);
+        this.transportService = transportService;
     }
 
     @Override
-    protected void masterOperation(
+    protected void localClusterStateOperation(
         Task task,
         XPackUsageRequest request,
         ClusterState state,