Add support to download IndexMetadata from Remote Cluster State

Signed-off-by: bansvaru <bansvaru@amazon.com>
soosinha · Aug 10, 2023 · e8226d4 · e8226d4
1 parent 77cac81
commit e8226d4
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/server/src/main/java/org/opensearch/cluster/store/RemoteClusterStateService.java b/server/src/main/java/org/opensearch/cluster/store/RemoteClusterStateService.java
@@ -15,6 +15,8 @@
 import org.opensearch.common.blobstore.BlobMetadata;
 import org.opensearch.cluster.store.ClusterMetadataMarker.UploadedIndexMetadata;
 import org.opensearch.common.settings.Settings;
+import org.opensearch.core.common.io.stream.InputStreamStreamInput;
+import org.opensearch.core.common.io.stream.StreamInput;
 import org.opensearch.indices.IndicesService;
 import org.opensearch.repositories.RepositoriesService;
 import org.opensearch.repositories.Repository;
@@ -194,4 +196,15 @@ public String getLatestMarkerFileName(String clusterUUID, String clusterState) {
         }
         return null;
     }
+
+    public Map<String, IndexMetadata> getLatestIndexMetadata(String clusterUUID, String clusterName) throws IOException {
+        Map<String, IndexMetadata> remoteIndexMetadata = new HashMap<>();
+        ClusterMetadataMarker clusterMetadataMarker = getLatestClusterMetadataMarker(clusterUUID, clusterName);
+        for (Map.Entry<String, UploadedIndexMetadata> entry: clusterMetadataMarker.getIndices().entrySet()) {
+            IndexMetadata indexMetadata = IndexMetadata.readFrom(new InputStreamStreamInput(
+                getMarkerBlobContainer(clusterUUID, clusterName).readBlob(entry.getValue().getUploadedFilename())));
+            remoteIndexMetadata.put(entry.getKey(), indexMetadata);
+        }
+        return remoteIndexMetadata;
+    }
 }