Adding a cancelled field to tell if a cancellable task is cancelled

client/rest-high-level/src/main/java/org/opensearch/client/tasks/TaskInfo.java

@@ -53,6 +53,7 @@ public class TaskInfo {
     private long startTime;
     private long runningTimeNanos;
     private boolean cancellable;
+    private boolean cancelled;
     private TaskId parentTaskId;
     private final Map<String, Object> status = new HashMap<>();
     private final Map<String, String> headers = new HashMap<>();
@@ -117,6 +118,14 @@ void setCancellable(boolean cancellable) {
         this.cancellable = cancellable;
     }
 
+    public boolean isCancelled() {
+        return cancelled;
+    }
+
+    void setCancelled(boolean cancelled) {
+        this.cancelled = cancelled;
+    }
+
     public TaskId getParentTaskId() {
         return parentTaskId;
     }
@@ -158,6 +167,7 @@ private void noOpParse(Object s) {}
         parser.declareLong(TaskInfo::setStartTime, new ParseField("start_time_in_millis"));
         parser.declareLong(TaskInfo::setRunningTimeNanos, new ParseField("running_time_in_nanos"));
         parser.declareBoolean(TaskInfo::setCancellable, new ParseField("cancellable"));
+        parser.declareBoolean(TaskInfo::setCancelled, new ParseField("cancelled"));
         parser.declareString(TaskInfo::setParentTaskId, new ParseField("parent_task_id"));
         parser.declareObject(TaskInfo::setHeaders, (p, c) -> p.mapStrings(), new ParseField("headers"));
         PARSER = (XContentParser p, Void v, String name) -> parser.parse(p, new TaskInfo(new TaskId(name)), null);
@@ -171,6 +181,7 @@ public boolean equals(Object o) {
         return getStartTime() == taskInfo.getStartTime()
             && getRunningTimeNanos() == taskInfo.getRunningTimeNanos()
             && isCancellable() == taskInfo.isCancellable()
+            && isCancelled() == taskInfo.isCancelled()
             && Objects.equals(getTaskId(), taskInfo.getTaskId())
             && Objects.equals(getType(), taskInfo.getType())
             && Objects.equals(getAction(), taskInfo.getAction())
@@ -190,6 +201,7 @@ public int hashCode() {
             getStartTime(),
             getRunningTimeNanos(),
             isCancellable(),
+            isCancelled(),
             getParentTaskId(),
             status,
             getHeaders()
@@ -216,6 +228,8 @@ public String toString() {
             + runningTimeNanos
             + ", cancellable="
             + cancellable
+            + ", cancelled="
+            + cancelled
             + ", parentTaskId="
             + parentTaskId
             + ", status="

client/rest-high-level/src/test/java/org/opensearch/client/core/tasks/GetTaskResponseTests.java

@@ -90,11 +90,24 @@ static TaskInfo randomTaskInfo() {
         long startTime = randomLong();
         long runningTimeNanos = randomLong();
         boolean cancellable = randomBoolean();
+        boolean cancelled = cancellable == true ? randomBoolean() : false;
         TaskId parentTaskId = randomBoolean() ? TaskId.EMPTY_TASK_ID : randomTaskId();
         Map<String, String> headers = randomBoolean()
             ? Collections.emptyMap()
             : Collections.singletonMap(randomAlphaOfLength(5), randomAlphaOfLength(5));
-        return new TaskInfo(taskId, type, action, description, status, startTime, runningTimeNanos, cancellable, parentTaskId, headers);
+        return new TaskInfo(
+            taskId,
+            type,
+            action,
+            description,
+            status,
+            startTime,
+            runningTimeNanos,
+            cancellable,
+            cancelled,
+            parentTaskId,
+            headers
+        );
     }
 
     private static TaskId randomTaskId() {

client/rest-high-level/src/test/java/org/opensearch/client/tasks/CancelTasksResponseTests.java

@@ -82,6 +82,8 @@ protected CancelTasksResponseTests.ByNodeCancelTasksResponse createServerTestIns
         }
 
         for (int i = 0; i < 4; i++) {
+            boolean cancellable = randomBoolean();
+            boolean cancelled = cancellable == true ? randomBoolean() : false;
             tasks.add(
                 new org.opensearch.tasks.TaskInfo(
                     new TaskId(NODE_ID, (long) i),
@@ -91,7 +93,8 @@ protected CancelTasksResponseTests.ByNodeCancelTasksResponse createServerTestIns
                     new FakeTaskStatus(randomAlphaOfLength(4), randomInt()),
                     randomLongBetween(1, 3),
                     randomIntBetween(5, 10),
-                    false,
+                    cancellable,
+                    cancelled,
                     new TaskId("node1", randomLong()),
                     Collections.singletonMap("x-header-of", "some-value")
                 )
@@ -128,6 +131,7 @@ protected void assertInstances(
             assertEquals(ti.getStartTime(), taskInfo.getStartTime());
             assertEquals(ti.getRunningTimeNanos(), taskInfo.getRunningTimeNanos());
             assertEquals(ti.isCancellable(), taskInfo.isCancellable());
+            assertEquals(ti.isCancelled(), taskInfo.isCancelled());
             assertEquals(ti.getParentTaskId().getNodeId(), taskInfo.getParentTaskId().getNodeId());
             assertEquals(ti.getParentTaskId().getId(), taskInfo.getParentTaskId().getId());
             FakeTaskStatus status = (FakeTaskStatus) ti.getStatus();

modules/reindex/src/test/java/org/opensearch/index/reindex/TransportRethrottleActionTests.java

@@ -129,6 +129,7 @@ public void testRethrottleSuccessfulResponse() {
                     0,
                     0,
                     true,
+                    false,
                     new TaskId("test", task.getId()),
                     Collections.emptyMap()
                 )
@@ -164,6 +165,7 @@ public void testRethrottleWithSomeSucceeded() {
                     0,
                     0,
                     true,
+                    false,
                     new TaskId("test", task.getId()),
                     Collections.emptyMap()
                 )

server/src/internalClusterTest/java/org/opensearch/action/admin/cluster/node/tasks/TasksIT.java

@@ -901,7 +901,19 @@ public void testNodeNotFoundButTaskFound() throws Exception {
         TaskResultsService resultsService = internalCluster().getInstance(TaskResultsService.class);
         resultsService.storeResult(
             new TaskResult(
-                new TaskInfo(new TaskId("fake", 1), "test", "test", "", null, 0, 0, false, TaskId.EMPTY_TASK_ID, Collections.emptyMap()),
+                new TaskInfo(
+                    new TaskId("fake", 1),
+                    "test",
+                    "test",
+                    "",
+                    null,
+                    0,
+                    0,
+                    false,
+                    false,
+                    TaskId.EMPTY_TASK_ID,
+                    Collections.emptyMap()
+                ),
                 new RuntimeException("test")
             ),
             new ActionListener<Void>() {

server/src/main/java/org/opensearch/tasks/Task.java

@@ -130,6 +130,7 @@ protected final TaskInfo taskInfo(String localNodeId, String description, Status
             startTime,
             System.nanoTime() - startTimeNanos,
             this instanceof CancellableTask,
+            this instanceof CancellableTask && ((CancellableTask) this).isCancelled(),
             parentTask,
             headers
         );

server/src/main/java/org/opensearch/tasks/TaskInfo.java

@@ -32,6 +32,7 @@
 
 package org.opensearch.tasks;
 
+import org.opensearch.Version;
 import org.opensearch.common.ParseField;
 import org.opensearch.common.Strings;
 import org.opensearch.common.bytes.BytesReference;
@@ -79,6 +80,8 @@ public final class TaskInfo implements Writeable, ToXContentFragment {
 
     private final boolean cancellable;
 
+    private final boolean cancelled;
+
     private final TaskId parentTaskId;
 
     private final Map<String, String> headers;
@@ -92,9 +95,13 @@ public TaskInfo(
         long startTime,
         long runningTimeNanos,
         boolean cancellable,
+        boolean cancelled,
         TaskId parentTaskId,
         Map<String, String> headers
     ) {
+        if (cancellable == false && cancelled == true) {
+            throw new IllegalArgumentException("task cannot be cancelled");
+        }
         this.taskId = taskId;
         this.type = type;
         this.action = action;
@@ -103,6 +110,7 @@ public TaskInfo(
         this.startTime = startTime;
         this.runningTimeNanos = runningTimeNanos;
         this.cancellable = cancellable;
+        this.cancelled = cancelled;
         this.parentTaskId = parentTaskId;
         this.headers = headers;
     }
@@ -119,6 +127,14 @@ public TaskInfo(StreamInput in) throws IOException {
         startTime = in.readLong();
         runningTimeNanos = in.readLong();
         cancellable = in.readBoolean();
+        if (in.getVersion().onOrAfter(Version.V_2_0_0)) {
+            cancelled = in.readBoolean();
+        } else {
+            cancelled = false;
+        }
+        if (cancellable == false && cancelled == true) {
+            throw new IllegalArgumentException("task cannot be cancelled");
+        }
         parentTaskId = TaskId.readFromStream(in);
         headers = in.readMap(StreamInput::readString, StreamInput::readString);
     }
@@ -133,6 +149,9 @@ public void writeTo(StreamOutput out) throws IOException {
         out.writeLong(startTime);
         out.writeLong(runningTimeNanos);
         out.writeBoolean(cancellable);
+        if (out.getVersion().onOrAfter(Version.V_2_0_0)) {
+            out.writeBoolean(cancelled);
+        }
         parentTaskId.writeTo(out);
         out.writeMap(headers, StreamOutput::writeString, StreamOutput::writeString);
     }
@@ -186,6 +205,13 @@ public boolean isCancellable() {
         return cancellable;
     }
 
+    /**
+     * Returns true if the task has been cancelled
+     */
+    public boolean isCancelled() {
+        return cancelled;
+    }
+
     /**
      * Returns the parent task id
      */
@@ -218,6 +244,7 @@ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws
         }
         builder.field("running_time_in_nanos", runningTimeNanos);
         builder.field("cancellable", cancellable);
+        builder.field("cancelled", cancelled);
         if (parentTaskId.isSet()) {
             builder.field("parent_task_id", parentTaskId.toString());
         }
@@ -243,6 +270,7 @@ public static TaskInfo fromXContent(XContentParser parser) {
         long startTime = (Long) a[i++];
         long runningTimeNanos = (Long) a[i++];
         boolean cancellable = (Boolean) a[i++];
+        boolean cancelled = a[i++] == Boolean.TRUE;
         String parentTaskIdString = (String) a[i++];
         @SuppressWarnings("unchecked")
         Map<String, String> headers = (Map<String, String>) a[i++];
@@ -252,7 +280,19 @@ public static TaskInfo fromXContent(XContentParser parser) {
         }
         RawTaskStatus status = statusBytes == null ? null : new RawTaskStatus(statusBytes);
         TaskId parentTaskId = parentTaskIdString == null ? TaskId.EMPTY_TASK_ID : new TaskId(parentTaskIdString);
-        return new TaskInfo(id, type, action, description, status, startTime, runningTimeNanos, cancellable, parentTaskId, headers);
+        return new TaskInfo(
+            id,
+            type,
+            action,
+            description,
+            status,
+            startTime,
+            runningTimeNanos,
+            cancellable,
+            cancelled,
+            parentTaskId,
+            headers
+        );
     });
     static {
         // Note for the future: this has to be backwards and forwards compatible with all changes to the task storage format
@@ -266,6 +306,7 @@ public static TaskInfo fromXContent(XContentParser parser) {
         PARSER.declareLong(constructorArg(), new ParseField("start_time_in_millis"));
         PARSER.declareLong(constructorArg(), new ParseField("running_time_in_nanos"));
         PARSER.declareBoolean(constructorArg(), new ParseField("cancellable"));
+        PARSER.declareBoolean(optionalConstructorArg(), new ParseField("cancelled"));
         PARSER.declareString(optionalConstructorArg(), new ParseField("parent_task_id"));
         PARSER.declareObject(optionalConstructorArg(), (p, c) -> p.mapStrings(), new ParseField("headers"));
     }
@@ -290,12 +331,25 @@ public boolean equals(Object obj) {
             && Objects.equals(runningTimeNanos, other.runningTimeNanos)
             && Objects.equals(parentTaskId, other.parentTaskId)
             && Objects.equals(cancellable, other.cancellable)
+            && Objects.equals(cancelled, other.cancelled)
             && Objects.equals(status, other.status)
             && Objects.equals(headers, other.headers);
     }
 
     @Override
     public int hashCode() {
-        return Objects.hash(taskId, type, action, description, startTime, runningTimeNanos, parentTaskId, cancellable, status, headers);
+        return Objects.hash(
+            taskId,
+            type,
+            action,
+            description,
+            startTime,
+            runningTimeNanos,
+            parentTaskId,
+            cancellable,
+            cancelled,
+            status,
+            headers
+        );
     }
 }

server/src/main/resources/org/opensearch/tasks/task-index-mapping.json

@@ -16,6 +16,9 @@
           "cancellable": {
             "type": "boolean"
           },
+          "cancelled": {
+            "type": "boolean"
+          },
           "id": {
             "type": "long"
           },