Make indexSuffix a parameter

Signed-off-by: Mikayla Thompson <thomika@amazon.com>
mikaylathompson · Apr 19, 2024 · 384401e · 384401e
1 parent 4ec659b
commit 384401e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/RFS/src/main/java/com/rfs/ReindexFromSnapshot.java b/RFS/src/main/java/com/rfs/ReindexFromSnapshot.java
@@ -88,6 +88,9 @@ public static class Args {
 
         @Parameter(names = {"--log-level"}, description = "What log level you want.  Default: 'info'", required = false, converter = Logging.ArgsConverter.class)
         public Level logLevel = Level.INFO;
+
+        @Parameter(names = {"--index_suffix"}, description = "An optional suffix to add to index names as they're transfered. Default: none", required = false)
+        public String indexSuffix = "";
     }
 
     public static void main(String[] args) throws InterruptedException {
@@ -118,6 +121,7 @@ public static void main(String[] args) throws InterruptedException {
         List<String> componentTemplateWhitelist = arguments.componentTemplateWhitelist;
         MovementType movementType = arguments.movementType;
         Level logLevel = arguments.logLevel;
+        String indexSuffix = arguments.indexSuffix;
 
         Logging.setLevel(logLevel);
 
@@ -297,7 +301,7 @@ public static void main(String[] args) throws InterruptedException {
                 logger.info("==================================================================");
                 logger.info("Attempting to recreate the indices...");
                 for (IndexMetadata.Data indexMetadata : indexMetadatas) {
-                    String reindexName = indexMetadata.getName() + "_reindexed";
+                    String reindexName = indexMetadata.getName() + indexSuffix;
                     logger.info("Recreating index " + indexMetadata.getName() + " as " + reindexName + " on target...");
 
                     ObjectNode root = indexMetadata.toObjectNode();
@@ -355,7 +359,7 @@ public static void main(String[] args) throws InterruptedException {
                         logger.info("Documents read successfully");
 
                         for (Document document : documents) {
-                            String targetIndex = indexMetadata.getName() + "_reindexed";
+                            String targetIndex = indexMetadata.getName() + indexSuffix;
                             DocumentReindexer.reindex(targetIndex, document, targetConnection);
                         }
                     }