From 2e3fbfd88bc712b2e9322d9b7813085f9dbcaf1e Mon Sep 17 00:00:00 2001
From: Sachin Kale <kalsac@amazon.com>
Date: Tue, 24 Oct 2023 13:22:55 +0530
Subject: [PATCH] Change to forceAllocate

Signed-off-by: Sachin Kale <kalsac@amazon.com>
---
 .../remotestore/RemoteStoreRestoreIT.java     | 10 +++----
 .../restore/RestoreRemoteStoreRequest.java    | 28 +++++++++----------
 .../cluster/routing/RecoverySource.java       | 22 +++++++--------
 .../recovery/RemoteStoreRestoreService.java   | 10 +++----
 .../opensearch/index/shard/StoreRecovery.java |  2 +-
 .../cluster/RestRestoreRemoteStoreAction.java |  2 +-
 6 files changed, 37 insertions(+), 37 deletions(-)

diff --git a/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteStoreRestoreIT.java b/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteStoreRestoreIT.java
index cbe7f3406998c..e202b9f6d34a7 100644
--- a/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteStoreRestoreIT.java
+++ b/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteStoreRestoreIT.java
@@ -38,7 +38,7 @@
 import static org.opensearch.test.hamcrest.OpenSearchAssertions.assertHitCount;
 import static org.hamcrest.Matchers.greaterThan;
 
-@OpenSearchIntegTestCase.ClusterScope(scope = OpenSearchIntegTestCase.Scope.SUITE, numDataNodes = 0)
+@OpenSearchIntegTestCase.ClusterScope(scope = OpenSearchIntegTestCase.Scope.TEST, numDataNodes = 0)
 public class RemoteStoreRestoreIT extends BaseRemoteStoreRestoreIT {
 
     /**
@@ -236,7 +236,7 @@ public void testRestoreFlowNoRedIndex() throws Exception {
         verifyRestoredData(indexStats, INDEX_NAME);
     }
 
-    public void testRestoreFlowWithForceEmptyTranslogNoOp() throws Exception {
+    public void testRestoreFlowWithForceAllocateNoOp() throws Exception {
         prepareCluster(1, 3, INDEX_NAME, 0, 1);
         Map<String, Long> indexStats = indexData(randomIntBetween(2, 3), randomBoolean(), INDEX_NAME);
 
@@ -252,7 +252,7 @@ public void testRestoreFlowWithForceEmptyTranslogNoOp() throws Exception {
         client().admin()
             .cluster()
             .restoreRemoteStore(
-                new RestoreRemoteStoreRequest().indices(INDEX_NAME).restoreAllShards(true).forceEmptyTranslog(true),
+                new RestoreRemoteStoreRequest().indices(INDEX_NAME).restoreAllShards(true).forceAllocate(true),
                 PlainActionFuture.newFuture()
             );
         ensureGreen(INDEX_NAME);
@@ -267,7 +267,7 @@ public void testRestoreFlowWithForceEmptyTranslogNoOp() throws Exception {
         );
     }
 
-    public void testRestoreFlowWithForceEmptyTranslog() throws Exception {
+    public void testRestoreFlowWithForceAllocate() throws Exception {
         prepareCluster(1, 3, INDEX_NAME, 0, 1);
         Map<String, Long> indexStats = indexData(randomIntBetween(2, 3), randomBoolean(), INDEX_NAME);
 
@@ -292,7 +292,7 @@ public void testRestoreFlowWithForceEmptyTranslog() throws Exception {
         client().admin()
             .cluster()
             .restoreRemoteStore(
-                new RestoreRemoteStoreRequest().indices(INDEX_NAME).restoreAllShards(true).forceEmptyTranslog(true),
+                new RestoreRemoteStoreRequest().indices(INDEX_NAME).restoreAllShards(true).forceAllocate(true),
                 PlainActionFuture.newFuture()
             );
         ensureGreen(INDEX_NAME);
diff --git a/server/src/main/java/org/opensearch/action/admin/cluster/remotestore/restore/RestoreRemoteStoreRequest.java b/server/src/main/java/org/opensearch/action/admin/cluster/remotestore/restore/RestoreRemoteStoreRequest.java
index 60928d641c2ab..c3e8834941e1c 100644
--- a/server/src/main/java/org/opensearch/action/admin/cluster/remotestore/restore/RestoreRemoteStoreRequest.java
+++ b/server/src/main/java/org/opensearch/action/admin/cluster/remotestore/restore/RestoreRemoteStoreRequest.java
@@ -38,7 +38,7 @@ public class RestoreRemoteStoreRequest extends ClusterManagerNodeRequest<Restore
     private String[] indices = Strings.EMPTY_ARRAY;
     private Boolean waitForCompletion = false;
     private Boolean restoreAllShards = false;
-    private Boolean forceEmptyTranslog = false;
+    private Boolean forceAllocate = false;
 
     public RestoreRemoteStoreRequest() {}
 
@@ -48,7 +48,7 @@ public RestoreRemoteStoreRequest(StreamInput in) throws IOException {
         waitForCompletion = in.readOptionalBoolean();
         restoreAllShards = in.readOptionalBoolean();
         if (in.getVersion().onOrAfter(Version.V_3_0_0)) {
-            forceEmptyTranslog = in.readOptionalBoolean();
+            forceAllocate = in.readOptionalBoolean();
         }
     }
 
@@ -59,7 +59,7 @@ public void writeTo(StreamOutput out) throws IOException {
         out.writeOptionalBoolean(waitForCompletion);
         out.writeOptionalBoolean(restoreAllShards);
         if (out.getVersion().onOrAfter(Version.V_3_0_0)) {
-            out.writeOptionalBoolean(forceEmptyTranslog);
+            out.writeOptionalBoolean(forceAllocate);
         }
     }
 
@@ -151,24 +151,24 @@ public boolean restoreAllShards() {
     }
 
     /**
-     * Set the value for forceEmptyTranslog, denoting whether to create empty translog if remote translog does not have any data.
+     * Set the value for forceAllocate, denoting whether to create empty store/translog if remote segment store/translog does not have any data.
      *
-     * @param forceEmptyTranslog If true and if remote translog does not have any data, the operation will create empty translog on local
-     *                           If false, the operation will always try to fetch data from remote translog and will fail if remote translog is empty.
+     * @param forceAllocate If true and if remote segment store/translog does not have any data, the operation will create empty store/translog on local
+     *                      If false, the operation will always try to fetch data from remote segment store/translog and will fail if either one is empty.
      * @return this request
      */
-    public RestoreRemoteStoreRequest forceEmptyTranslog(boolean forceEmptyTranslog) {
-        this.forceEmptyTranslog = forceEmptyTranslog;
+    public RestoreRemoteStoreRequest forceAllocate(boolean forceAllocate) {
+        this.forceAllocate = forceAllocate;
         return this;
     }
 
     /**
-     * Returns forceEmptyTranslog setting
+     * Returns forceAllocate setting
      *
-     * @return true if the operation will create empty translog on local when remote translog is empty
+     * @return true if the operation will create empty store/translog on local when remote segment store/translog is empty
      */
-    public boolean forceEmptyTranslog() {
-        return forceEmptyTranslog;
+    public boolean forceAllocate() {
+        return forceAllocate;
     }
 
     /**
@@ -222,13 +222,13 @@ public boolean equals(Object o) {
         RestoreRemoteStoreRequest that = (RestoreRemoteStoreRequest) o;
         return waitForCompletion == that.waitForCompletion
             && restoreAllShards == that.restoreAllShards
-            && forceEmptyTranslog == that.forceEmptyTranslog
+            && forceAllocate == that.forceAllocate
             && Arrays.equals(indices, that.indices);
     }
 
     @Override
     public int hashCode() {
-        int result = Objects.hash(waitForCompletion, restoreAllShards, forceEmptyTranslog);
+        int result = Objects.hash(waitForCompletion, restoreAllShards, forceAllocate);
         result = 31 * result + Arrays.hashCode(indices);
         return result;
     }
diff --git a/server/src/main/java/org/opensearch/cluster/routing/RecoverySource.java b/server/src/main/java/org/opensearch/cluster/routing/RecoverySource.java
index a9a1cf8d9ff90..0235f53174e8a 100644
--- a/server/src/main/java/org/opensearch/cluster/routing/RecoverySource.java
+++ b/server/src/main/java/org/opensearch/cluster/routing/RecoverySource.java
@@ -419,17 +419,17 @@ public static class RemoteStoreRecoverySource extends RecoverySource {
         private final String restoreUUID;
         private final IndexId index;
         private final Version version;
-        private final boolean forceEmptyTranslog;
+        private final boolean forceAllocate;
 
         public RemoteStoreRecoverySource(String restoreUUID, Version version, IndexId indexId) {
             this(restoreUUID, version, indexId, false);
         }
 
-        public RemoteStoreRecoverySource(String restoreUUID, Version version, IndexId indexId, boolean forceEmptyTranslog) {
+        public RemoteStoreRecoverySource(String restoreUUID, Version version, IndexId indexId, boolean forceAllocate) {
             this.restoreUUID = restoreUUID;
             this.version = Objects.requireNonNull(version);
             this.index = Objects.requireNonNull(indexId);
-            this.forceEmptyTranslog = forceEmptyTranslog;
+            this.forceAllocate = forceAllocate;
         }
 
         RemoteStoreRecoverySource(StreamInput in) throws IOException {
@@ -437,9 +437,9 @@ public RemoteStoreRecoverySource(String restoreUUID, Version version, IndexId in
             version = in.readVersion();
             index = new IndexId(in);
             if (in.getVersion().onOrAfter(Version.V_3_0_0)) {
-                forceEmptyTranslog = in.readBoolean();
+                forceAllocate = in.readBoolean();
             } else {
-                forceEmptyTranslog = false;
+                forceAllocate = false;
             }
         }
 
@@ -461,8 +461,8 @@ public Version version() {
             return version;
         }
 
-        public boolean forceEmptyTranslog() {
-            return forceEmptyTranslog;
+        public boolean forceAllocate() {
+            return forceAllocate;
         }
 
         @Override
@@ -471,7 +471,7 @@ protected void writeAdditionalFields(StreamOutput out) throws IOException {
             out.writeVersion(version);
             index.writeTo(out);
             if (out.getVersion().onOrAfter(Version.V_3_0_0)) {
-                out.writeBoolean(forceEmptyTranslog);
+                out.writeBoolean(forceAllocate);
             }
         }
 
@@ -485,7 +485,7 @@ public void addAdditionalFields(XContentBuilder builder, ToXContent.Params param
             builder.field("version", version.toString())
                 .field("index", index.getName())
                 .field("restoreUUID", restoreUUID)
-                .field("forceEmptyTranslog", forceEmptyTranslog);
+                .field("forceAllocate", forceAllocate);
         }
 
         @Override
@@ -506,12 +506,12 @@ public boolean equals(Object o) {
             return restoreUUID.equals(that.restoreUUID)
                 && index.equals(that.index)
                 && version.equals(that.version)
-                && forceEmptyTranslog == that.forceEmptyTranslog;
+                && forceAllocate == that.forceAllocate;
         }
 
         @Override
         public int hashCode() {
-            return Objects.hash(restoreUUID, index, version, forceEmptyTranslog);
+            return Objects.hash(restoreUUID, index, version, forceAllocate);
         }
 
         // TODO: This override should be removed/be updated to return "true",
diff --git a/server/src/main/java/org/opensearch/index/recovery/RemoteStoreRestoreService.java b/server/src/main/java/org/opensearch/index/recovery/RemoteStoreRestoreService.java
index b1d27949d4518..2d53d4f962dea 100644
--- a/server/src/main/java/org/opensearch/index/recovery/RemoteStoreRestoreService.java
+++ b/server/src/main/java/org/opensearch/index/recovery/RemoteStoreRestoreService.java
@@ -103,7 +103,7 @@ public ClusterState execute(ClusterState currentState) {
                     currentState,
                     null,
                     request.restoreAllShards(),
-                    request.forceEmptyTranslog(),
+                    request.forceAllocate(),
                     request.indices()
                 );
                 restoreUUID = remoteRestoreResult.getRestoreUUID();
@@ -141,7 +141,7 @@ public RemoteRestoreResult restore(
         ClusterState currentState,
         @Nullable String restoreClusterUUID,
         boolean restoreAllShards,
-        boolean forceEmptyTranslog,
+        boolean forceAllocate,
         String[] indexNames
     ) {
         Map<String, Tuple<Boolean, IndexMetadata>> indexMetadataMap = new HashMap<>();
@@ -183,7 +183,7 @@ public RemoteRestoreResult restore(
                 }
             }
         }
-        return executeRestore(currentState, indexMetadataMap, restoreAllShards, remoteMetadata, forceEmptyTranslog);
+        return executeRestore(currentState, indexMetadataMap, restoreAllShards, remoteMetadata, forceAllocate);
     }
 
     /**
@@ -198,7 +198,7 @@ private RemoteRestoreResult executeRestore(
         Map<String, Tuple<Boolean, IndexMetadata>> indexMetadataMap,
         boolean restoreAllShards,
         Metadata remoteMetadata,
-        boolean forceEmptyTranslog
+        boolean forceAllocate
     ) {
         final String restoreUUID = UUIDs.randomBase64UUID();
         List<String> indicesToBeRestored = new ArrayList<>();
@@ -237,7 +237,7 @@ private RemoteRestoreResult executeRestore(
                     restoreUUID,
                     updatedIndexMetadata.getCreationVersion(),
                     indexId,
-                    forceEmptyTranslog
+                    forceAllocate
                 );
 
                 rtBuilder.addAsRemoteStoreRestore(updatedIndexMetadata, recoverySource, indexShardRoutingTableMap, restoreAllShards);
diff --git a/server/src/main/java/org/opensearch/index/shard/StoreRecovery.java b/server/src/main/java/org/opensearch/index/shard/StoreRecovery.java
index 7d0cbaf9986fa..71d12514969ae 100644
--- a/server/src/main/java/org/opensearch/index/shard/StoreRecovery.java
+++ b/server/src/main/java/org/opensearch/index/shard/StoreRecovery.java
@@ -554,7 +554,7 @@ private void recoverFromRemoteStore(IndexShard indexShard) throws IndexShardReco
                     store.createEmpty(indexShard.indexSettings().getIndexVersionCreated().luceneVersion, translogHeader.getTranslogUUID());
                 }
             } else if (remoteSegmentEmpty == false && remoteTranslogEmpty) {
-                if (((RecoverySource.RemoteStoreRecoverySource) indexShard.shardRouting.recoverySource()).forceEmptyTranslog()) {
+                if (((RecoverySource.RemoteStoreRecoverySource) indexShard.shardRouting.recoverySource()).forceAllocate()) {
                     bootstrapFromLastCommit(indexShard, store);
                 }
             }
diff --git a/server/src/main/java/org/opensearch/rest/action/admin/cluster/RestRestoreRemoteStoreAction.java b/server/src/main/java/org/opensearch/rest/action/admin/cluster/RestRestoreRemoteStoreAction.java
index 73dfa4ba0a656..5bd0d8dfba663 100644
--- a/server/src/main/java/org/opensearch/rest/action/admin/cluster/RestRestoreRemoteStoreAction.java
+++ b/server/src/main/java/org/opensearch/rest/action/admin/cluster/RestRestoreRemoteStoreAction.java
@@ -45,7 +45,7 @@ public RestChannelConsumer prepareRequest(final RestRequest request, final NodeC
         );
         restoreRemoteStoreRequest.waitForCompletion(request.paramAsBoolean("wait_for_completion", false));
         restoreRemoteStoreRequest.restoreAllShards(request.paramAsBoolean("restore_all_shards", false));
-        restoreRemoteStoreRequest.forceEmptyTranslog(request.paramAsBoolean("force_empty_translog", false));
+        restoreRemoteStoreRequest.forceAllocate(request.paramAsBoolean("force_allocate", false));
         request.applyContentParser(p -> restoreRemoteStoreRequest.source(p.mapOrdered()));
         return channel -> client.admin().cluster().restoreRemoteStore(restoreRemoteStoreRequest, new RestToXContentListener<>(channel));
     }