From 7ffa7a9b23bcc351936e207ea5d95d7ee9eaa83a Mon Sep 17 00:00:00 2001
From: Charles Gagnon <chgagnon@microsoft.com>
Date: Wed, 17 Mar 2021 10:12:26 -0700
Subject: [PATCH] Add support for replicas option when deploying MIAA (#14758)

---
 .../arcDeployment/deploy.sql.existing.arc.ipynb   |  2 +-
 extensions/arc/package.json                       | 15 +++++++++++++++
 extensions/arc/package.nls.json                   |  2 ++
 3 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/extensions/arc/notebooks/arcDeployment/deploy.sql.existing.arc.ipynb b/extensions/arc/notebooks/arcDeployment/deploy.sql.existing.arc.ipynb
index e6d45a14f3..ac052d08c0 100644
--- a/extensions/arc/notebooks/arcDeployment/deploy.sql.existing.arc.ipynb
+++ b/extensions/arc/notebooks/arcDeployment/deploy.sql.existing.arc.ipynb
@@ -138,7 +138,7 @@
                 "\n",
                 "os.environ[\"AZDATA_USERNAME\"] = sql_username\n",
                 "os.environ[\"AZDATA_PASSWORD\"] = os.environ[\"AZDATA_NB_VAR_SQL_PASSWORD\"]\n",
-                "cmd = f'azdata arc sql mi create -n {sql_instance_name} -scd {sql_storage_class_data} -scl {sql_storage_class_logs}{cores_request_option}{cores_limit_option}{memory_request_option}{memory_limit_option}'\n",
+                "cmd = f'azdata arc sql mi create -n {sql_instance_name} -scd {sql_storage_class_data} -scl {sql_storage_class_logs} --replicas {sql_replicas}{cores_request_option}{cores_limit_option}{memory_request_option}{memory_limit_option}'\n",
                 "out=run_command()"
             ],
             "metadata": {
diff --git a/extensions/arc/package.json b/extensions/arc/package.json
index 9c32e97eaa..c6cfbb5dc7 100644
--- a/extensions/arc/package.json
+++ b/extensions/arc/package.json
@@ -900,6 +900,21 @@
                   {
                     "title": "%arc.sql.instance.settings.section.title%",
                     "fields": [
+                      {
+                        "type": "options",
+                        "label": "%arc.sql.replicas.label%",
+                        "description": "%arc.sql.replicas.description%",
+                        "required": true,
+                        "variableName": "AZDATA_NB_VAR_SQL_REPLICAS",
+                        "options": {
+                          "values": [
+                            "1",
+                            "3"
+                          ],
+                          "defaultValue": "1",
+                          "optionsType": "radio"
+                        }
+                      },
                       {
                         "label": "%arc.storage-class.data.label%",
                         "description": "%arc.sql.storage-class.data.description%",
diff --git a/extensions/arc/package.nls.json b/extensions/arc/package.nls.json
index fa7c97ce44..02949302b4 100644
--- a/extensions/arc/package.nls.json
+++ b/extensions/arc/package.nls.json
@@ -84,6 +84,8 @@
 	"arc.sql.invalid.instance.name": "Instance name must consist of lower case alphanumeric characters or '-', start with a letter, end with an alphanumeric character, and be 13 characters or fewer in length.",
 	"arc.storage-class.dc.label": "Storage Class",
 	"arc.sql.storage-class.dc.description": "The storage class to be used for all data and logs persistent volumes for all data controller pods that require them.",
+	"arc.sql.replicas.label": "Replicas",
+	"arc.sql.replicas.description": "The number of replicas to be deployed for high availability purpose",
 	"arc.storage-class.data.label": "Storage Class (Data)",
 	"arc.sql.storage-class.data.description": "The storage class to be used for data (.mdf)",
 	"arc.postgres.storage-class.data.description": "The storage class to be used for data persistent volumes",