PR Fixes - Make enable / disable endpoint cancellable

Author: lukewhiting

modules/streams/src/main/java/org/elasticsearch/rest/streams/logs/LogsStreamsActivationToggleAction.java

@@ -9,15 +9,18 @@
 
 package org.elasticsearch.rest.streams.logs;
 
-import org.elasticsearch.action.ActionRequestValidationException;
 import org.elasticsearch.action.ActionType;
 import org.elasticsearch.action.support.master.AcknowledgedRequest;
 import org.elasticsearch.action.support.master.AcknowledgedResponse;
 import org.elasticsearch.common.io.stream.StreamInput;
 import org.elasticsearch.common.io.stream.StreamOutput;
 import org.elasticsearch.core.TimeValue;
+import org.elasticsearch.tasks.CancellableTask;
+import org.elasticsearch.tasks.Task;
+import org.elasticsearch.tasks.TaskId;
 
 import java.io.IOException;
+import java.util.Map;
 
 public class LogsStreamsActivationToggleAction {
 
@@ -43,11 +46,6 @@ public void writeTo(StreamOutput out) throws IOException {
             out.writeBoolean(enable);
         }
 
-        @Override
-        public ActionRequestValidationException validate() {
-            return null;
-        }
-
         @Override
         public String toString() {
             return "LogsStreamsActivationToggleAction.Request{" + "enable=" + enable + '}';
@@ -56,5 +54,10 @@ public String toString() {
         public boolean shouldEnable() {
             return enable;
         }
+
+        @Override
+        public Task createTask(String localNodeId, long id, String type, String action, TaskId parentTaskId, Map<String, String> headers) {
+            return new CancellableTask(id, type, action, "Logs streams activation toggle request", parentTaskId, headers);
+        }
     }
 }

modules/streams/src/main/java/org/elasticsearch/rest/streams/logs/RestSetLogStreamsEnabledAction.java

@@ -15,6 +15,7 @@
 import org.elasticsearch.rest.RestUtils;
 import org.elasticsearch.rest.Scope;
 import org.elasticsearch.rest.ServerlessScope;
+import org.elasticsearch.rest.action.RestCancellableNodeClient;
 import org.elasticsearch.rest.action.RestToXContentListener;
 
 import java.util.List;
@@ -48,7 +49,7 @@ protected RestChannelConsumer prepareRequest(RestRequest request, NodeClient cli
             enabled
         );
 
-        return restChannel -> client.execute(
+        return restChannel -> new RestCancellableNodeClient(client, request.getHttpChannel()).execute(
             LogsStreamsActivationToggleAction.INSTANCE,
             activationRequest,
             new RestToXContentListener<>(restChannel)