Merge pull request #3 from shsagir/ad-hoc

Add snippets for zonal migration

Author: shsagir

docs/snippets/migrate-cluster-to-multiple-availability-zone/configure-zones.cs

@@ -0,0 +1,39 @@
+using Azure.Core;
+using Azure.ResourceManager.Kusto;
+using Azure.ResourceManager.Kusto.Models;
+using ArmClient = Azure.ResourceManager.ArmClient;
+using ClientSecretCredential = Azure.Identity.ClientSecretCredential;
+using WaitUntil = Azure.WaitUntil;
+
+var tenantId = "{tenantId}";
+var clientId = "{clientId}";
+var clientSecret = "{clientSecret}";
+var subscriptionId = "{subscriptionId}";
+var resourceGroupName = "{resourceGroupName}";
+var clusterName = "{clusterName}";
+
+var credentials = new ClientSecretCredential(tenantId, clientId, clientSecret);
+var resourceManagementClient = new ArmClient(credentials, subscriptionId);
+var resourceIdentifier = new ResourceIdentifier($"/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Kusto/Clusters/{clusterName}");
+var cluster = resourceManagementClient.GetKustoClusterResource(resourceIdentifier);
+
+var kustoClusterPatch = new KustoClusterPatch(AzureLocation.NorthEurope);
+kustoClusterPatch.Zones.Clear();
+kustoClusterPatch.Zones.Add("1");
+kustoClusterPatch.Zones.Add("2");
+kustoClusterPatch.Zones.Add("3");
+
+var armOperation = await cluster.UpdateAsync(WaitUntil.Started, kustoClusterPatch).ConfigureAwait(false);
+
+var response = armOperation.UpdateStatus();
+Console.WriteLine($"ClientRequestId: {response.ClientRequestId}");
+
+while (true)
+{
+  Console.WriteLine($"{DateTime.UtcNow:o} {response.Status, -5} {response.ReasonPhrase}");
+  if (armOperation.HasCompleted)
+    break;
+
+  await Task.Delay(60000).ConfigureAwait(false);
+  response = await armOperation.UpdateStatusAsync().ConfigureAwait(false);
+}

docs/snippets/migrate-cluster-to-multiple-availability-zone/configure-zones.json

@@ -0,0 +1,46 @@
+{
+  "$schema": "https://schema.management.azure.com/schemas/2019-04-01/deploymentTemplate.json#",
+  "contentVersion": "1.0.0.0",
+  "parameters": {
+      "clusters_clusterName_name": {
+          "defaultValue": "clusterName",
+          "type": "string"
+      }
+  },
+  "variables": {},
+  "resources": [
+      {
+          "type": "Microsoft.Kusto/clusters",
+          "apiVersion": "2023-05-02",
+          "name": "[parameters('clusters_clusterName_name')]",
+          "location": "North Europe",
+          "sku": {
+              "name": "{skuName}",
+              "tier": "Basic",
+              "capacity": 1
+          },
+          "zones": ["1","2","3"],
+          "identity": {
+              "type": "SystemAssigned"
+          },
+          "properties": {
+              "trustedExternalTenants": [],
+              "enableDiskEncryption": false,
+              "enableStreamingIngest": false,
+              "languageExtensions": {
+                  "value": []
+              },
+              "enablePurge": false,
+              "enableDoubleEncryption": false,
+              "engineType": "V3",
+              "acceptedAudiences": [],
+              "restrictOutboundNetworkAccess": "Disabled",
+              "allowedFqdnList": [],
+              "publicNetworkAccess": "Enabled",
+              "allowedIpRangeList": [],
+              "enableAutoStop": true,
+              "publicIPType": "IPv4"
+          }
+      }
+  ]
+}

docs/snippets/migrate-cluster-to-multiple-availability-zone/configure-zones.py

@@ -0,0 +1,22 @@
+from azure.identity import DefaultAzureCredential
+from azure.mgmt.kusto import KustoManagementClient
+from azure.mgmt.kusto import models
+import time
+from azure.core.exceptions import HttpResponseError 
+
+subscription_id = "{subscriptionId}"
+resource_group_name = "{resourceGroupName}"
+cluster_name = "{clusterName}"
+
+client = KustoManagementClient(DefaultAzureCredential(), subscription_id)
+lro_poller = client.clusters.begin_update(resource_group_name, cluster_name, models.ClusterUpdate.from_dict({"zones": ["1", "2", "3"]}))
+
+while (not(lro_poller.done())):
+    time.sleep(60)
+
+print (f"status: {lro_poller.status()}")
+
+try:
+    lro_poller.result()
+except HttpResponseError as e:
+    print (f"Exception: {e}")