From a1e7d2caedbad860b0c0879b906d52289e9614a5 Mon Sep 17 00:00:00 2001
From: Yisheng Zhou <zhouys93@hotmail.com>
Date: Tue, 14 May 2024 11:23:44 -0700
Subject: [PATCH] Add ParallelReplaceNodeAction

---
 .../generic/GenericClusterWideAction.java     | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/orion-server/src/main/java/com/pinterest/orion/core/actions/generic/GenericClusterWideAction.java b/orion-server/src/main/java/com/pinterest/orion/core/actions/generic/GenericClusterWideAction.java
index 3187ca68..44feb9f7 100644
--- a/orion-server/src/main/java/com/pinterest/orion/core/actions/generic/GenericClusterWideAction.java
+++ b/orion-server/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");