Merge pull request #271 from pinterest/ParallelReplaceNodeAction

Add ParallelReplaceNodeAction
pinterest · May 14, 2024 · 403a00b · 403a00b
2 parents 3e6f76d + a1e7d2c
commit 403a00b
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/...rver/src/main/java/com/pinterest/orion/core/actions/generic/GenericClusterWideAction.java b/...rver/src/main/java/com/pinterest/orion/core/actions/generic/GenericClusterWideAction.java
@@ -528,6 +528,32 @@ public boolean failIfNoNodeIds() {
     }
   }
 
+  public static class ParallelReplaceNodeAction extends ParallelAction {
+    public ParallelReplaceNodeAction() {
+      super("Replace");
+    }
+
+    @Override
+    public Action getChildAction() {
+      return new NodeAction() {
+        @Override
+        public String getName() {
+          return "ReplaceNodeAction";
+        }
+      };
+    }
+
+    @Override
+    public boolean isCancelAllIfFailed() {
+      return true;
+    }
+
+    @Override
+    public boolean failIfNoNodeIds() {
+      return true;
+    }
+  }
+
   public static class ParallelRebootNodeAction extends ParallelAction {
     public ParallelRebootNodeAction() {
       super("Reboot");