Adding support for restart upgrades for plugins bwc

Signed-off-by: Vacha <vachshah@amazon.com>
opensearch-project · Aug 10, 2021 · 7216b78 · 7216b78
1 parent e616879
commit 7216b78
Showing 1 changed file with 22 additions and 16 deletions.
diff --git a/buildSrc/src/main/java/org/opensearch/gradle/testclusters/OpenSearchCluster.java b/buildSrc/src/main/java/org/opensearch/gradle/testclusters/OpenSearchCluster.java
@@ -392,32 +392,26 @@ public void goToNextVersion() {
         writeUnicastHostsFiles();
     }
 
+    public void upgradeAllNodesAndPluginsToNextVersion(List<Provider<RegularFile>> plugins) {
+        stop(false);
+        nodes.all(OpenSearchNode::goToNextVersion);
+        upgradePlugin(plugins);
+        start();
+        writeUnicastHostsFiles();
+    }
+
     public void fullRestart() {
         stop(false);
         start();
     }
 
     public void nextNodeToNextVersion() {
-        if (nodeIndex + 1 > nodes.size()) {
-            throw new TestClustersException("Ran out of nodes to take to the next version");
-        }
-        OpenSearchNode node = nodes.getByName(clusterName + "-" + nodeIndex);
-        node.stop(false);
-        node.goToNextVersion();
-        commonNodeConfig(node, null, null);
-        nodeIndex += 1;
+        OpenSearchNode node = upgradeNodeToNextVersion();
         node.start();
     }
 
     public void upgradeNodeAndPluginToNextVersion(List<Provider<RegularFile>> plugins) {
-        if (nodeIndex + 1 > nodes.size()) {
-            throw new TestClustersException("Ran out of nodes to take to the next version");
-        }
-        OpenSearchNode node = nodes.getByName(clusterName + "-" + nodeIndex);
-        node.stop(false);
-        node.goToNextVersion();
-        commonNodeConfig(node, null, null);
-        nodeIndex += 1;
+        OpenSearchNode node = upgradeNodeToNextVersion();
         node.upgradePlugin(plugins);
         node.start();
     }
@@ -453,6 +447,18 @@ private void writeUnicastHostsFiles() {
         });
     }
 
+    private OpenSearchNode upgradeNodeToNextVersion() {
+        if (nodeIndex + 1 > nodes.size()) {
+            throw new TestClustersException("Ran out of nodes to take to the next version");
+        }
+        OpenSearchNode node = nodes.getByName(clusterName + "-" + nodeIndex);
+        node.stop(false);
+        node.goToNextVersion();
+        commonNodeConfig(node, null, null);
+        nodeIndex += 1;
+        return node;
+    }
+
     @Override
     @Internal
     public String getHttpSocketURI() {