[PLAT-12950]Allow aborting Support bundle task

Summary:
Allow aborting support bundle creation task.
Bundles are marked as `Aborted` and cleaned up by the GC.

Test Plan: Verified that cancellation of support bundle creation task works and the left behind files are cleaned up by the GC.

Reviewers: #yba-api-review!, nsingh, skurapati

Reviewed By: skurapati

Subscribers: anijhawan, sanketh, yugaware

Differential Revision: https://phorge.dev.yugabyte.com/D40634

Author: asharma-yb

managed/src/main/java/com/yugabyte/yw/commissioner/SupportBundleCleanup.java

@@ -65,13 +65,14 @@ void scheduleRunner() {
   public synchronized void deleteSupportBundleIfOld(SupportBundle supportBundle)
       throws ParseException {
     int default_delete_days = config.getInt("yb.support_bundle.retention_days");
-
-    if (supportBundle.getStatus() == SupportBundleStatusType.Failed) {
+    SupportBundleStatusType status = supportBundle.getStatus();
+    if (status == SupportBundleStatusType.Failed || status == SupportBundleStatusType.Aborted) {
       supportBundleUtil.deleteSupportBundle(supportBundle);
 
       log.info(
-          "Automatically deleted Support Bundle with UUID: {}, with status = Failed",
-          supportBundle.getBundleUUID());
+          "Automatically deleted Support Bundle with UUID: {}, with status = {}",
+          supportBundle.getBundleUUID(),
+          status);
     } else if (supportBundle.getStatus() == SupportBundleStatusType.Running) {
       return;
     } else {

managed/src/main/java/com/yugabyte/yw/commissioner/tasks/CreateSupportBundle.java

@@ -16,6 +16,7 @@
 import com.typesafe.config.Config;
 import com.yugabyte.yw.commissioner.AbstractTaskBase;
 import com.yugabyte.yw.commissioner.BaseTaskDependencies;
+import com.yugabyte.yw.commissioner.ITask.Abortable;
 import com.yugabyte.yw.commissioner.TaskExecutor.SubTaskGroup;
 import com.yugabyte.yw.commissioner.tasks.params.SupportBundleTaskParams;
 import com.yugabyte.yw.commissioner.tasks.subtasks.CheckNodeReachable;
@@ -34,6 +35,7 @@
 import com.yugabyte.yw.models.Customer;
 import com.yugabyte.yw.models.SupportBundle;
 import com.yugabyte.yw.models.SupportBundle.SupportBundleStatusType;
+import com.yugabyte.yw.models.TaskInfo;
 import com.yugabyte.yw.models.Universe;
 import com.yugabyte.yw.models.helpers.BundleDetails;
 import com.yugabyte.yw.models.helpers.NodeDetails;
@@ -52,6 +54,7 @@
 import play.libs.Json;
 
 @Slf4j
+@Abortable
 public class CreateSupportBundle extends AbstractTaskBase {
 
   @Inject private UniverseInfoHandler universeInfoHandler;
@@ -86,11 +89,17 @@ public void run() {
       operatorStatusUpdater.markSupportBundleFinished(
           supportBundle, taskParams().getKubernetesResourceDetails(), gzipPath);
     } catch (Exception e) {
-      taskParams().supportBundle.setStatus(SupportBundleStatusType.Failed);
-      operatorStatusUpdater.markSupportBundleFailed(
-          supportBundle, taskParams().getKubernetesResourceDetails());
-      Throwables.throwIfUnchecked(e);
-      throw new RuntimeException(e);
+      TaskInfo taskInfo = getRunnableTask().getTaskInfo();
+      if (taskInfo.getTaskState().equals(TaskInfo.State.Abort)) {
+        log.info("Marking support bundle with UUID: {} as aborted.", supportBundle.getBundleUUID());
+        supportBundle.setStatus(SupportBundleStatusType.Aborted);
+      } else {
+        supportBundle.setStatus(SupportBundleStatusType.Failed);
+        operatorStatusUpdater.markSupportBundleFailed(
+            supportBundle, taskParams().getKubernetesResourceDetails());
+        Throwables.throwIfUnchecked(e);
+        throw new RuntimeException(e);
+      }
     } finally {
       supportBundle.update();
     }

managed/src/main/java/com/yugabyte/yw/models/SupportBundle.java

@@ -82,7 +82,8 @@ public class SupportBundle extends Model {
   public enum SupportBundleStatusType {
     Running("Running"),
     Success("Success"),
-    Failed("Failed");
+    Failed("Failed"),
+    Aborted("Aborted");
 
     private final String status;
 

managed/src/main/resources/db/migration/default_/common/V387__Alter_Support_Bundle.sql

@@ -0,0 +1,5 @@
+-- Copyright (c) YugaByte, Inc.
+
+alter table if exists support_bundle drop constraint if exists ck_support_bundle_status;
+
+alter table if exists support_bundle add constraint ck_support_bundle_status check (status in ('Running','Success','Failed','Aborted'));

managed/src/main/resources/swagger-strict.json

@@ -12708,7 +12708,7 @@
           "type" : "string"
         },
         "status" : {
-          "enum" : [ "Running", "Success", "Failed" ],
+          "enum" : [ "Running", "Success", "Failed", "Aborted" ],
           "type" : "string"
         }
       },

managed/src/main/resources/swagger.json

@@ -12825,7 +12825,7 @@
           "type" : "string"
         },
         "status" : {
-          "enum" : [ "Running", "Success", "Failed" ],
+          "enum" : [ "Running", "Success", "Failed", "Aborted" ],
           "type" : "string"
         }
       },