From 94a3187a6fb7a673946a215869c154048603389b Mon Sep 17 00:00:00 2001
From: zheyanyu <zheyanyu@amazon.com>
Date: Thu, 10 Jun 2021 14:11:58 -0400
Subject: [PATCH] chore: script for add alias to existing index

---
 scripts/elasticsearch-operations.js | 49 ++++++++++++++++++++++++++---
 1 file changed, 44 insertions(+), 5 deletions(-)

diff --git a/scripts/elasticsearch-operations.js b/scripts/elasticsearch-operations.js
index 57890ec9..862d2b67 100644
--- a/scripts/elasticsearch-operations.js
+++ b/scripts/elasticsearch-operations.js
@@ -1,8 +1,8 @@
-/* 
- *  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
- *  SPDX-License-Identifier: Apache-2.0
- */
-
+/*
+ *  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *  SPDX-License-Identifier: Apache-2.0
+ */
+
 // Running ES Script
 // ACCESS_KEY=<ACCESS_KEY> SECRET_KEY=<SECRET_KEY> ES_DOMAIN_ENDPOINT=<ES_DOMAIN_ENDPOINT> node elasticsearch-operations.js <region> <function to execute> <optional additional params>
 //
@@ -110,6 +110,43 @@ const deleteIndex = async () => {
     }
 };
 
+const addAlias = async () => {
+    // Get all indices
+    const response = await es.cat.indices({ format: 'json' });
+    const indices = response.body.map(indexDetail => indexDetail.index);
+
+    // Check indices for alias
+    const checkAlias = indices.map(indexName => {
+        return es.indices.existsAlias({
+            index: indexName,
+            name: `${indexName}-alias`,
+        });
+    });
+    const checkAliasResponse = await Promise.all(checkAlias);
+
+    // Filter out indices without alias
+    const aliasesToAdd = checkAliasResponse
+        .map((checkAliasResult, index) => {
+            return { hasAlias: checkAliasResult.body, indexName: indices[index] };
+        })
+        .filter(checkResult => !checkResult.hasAlias);
+
+    // Add alias if needed
+    if (aliasesToAdd.length === 0) {
+        console.log('All indices have alias created, nothing to do.');
+    } else {
+        console.log(
+            `Adding aliases for indices: `,
+            aliasesToAdd.map(aliasToAdd => aliasToAdd.indexName),
+        );
+        const addAliases = aliasesToAdd.map(checkResult => {
+            return es.indices.putAlias({ index: checkResult.indexName, name: `${checkResult.indexName}-alias` });
+        });
+        await Promise.all(addAliases);
+        console.log('Aliases added.');
+    }
+};
+
 const functionToExecute = process.argv[3];
 if (functionToExecute === 'search') {
     const name = process.argv[4];
@@ -122,4 +159,6 @@ if (functionToExecute === 'search') {
     deleteIndex();
 } else if (functionToExecute === 'getIndexMapping') {
     getIndexMapping();
+} else if (functionToExecute === 'addAlias') {
+    addAlias();
 }