[HDInsight]Support compute isolation and return private ip address

src/azure-cli/azure/cli/command_modules/hdinsight/_help.py

@@ -150,6 +150,11 @@
         -p "HttpPassword1234!" --storage-account MyStorageAccount \\
         --subnet "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/rg/providers/Microsoft.Network/virtualNetworks/fakevnet/subnets/default" \\
         --resource-provider-connection Outbound --enable-private-link
+  - name: Create a cluster with Compute Isolation feature.
+    text: |-
+        az hdinsight create -t spark --version 3.6 -g MyResourceGroup -n MyCluster \\
+        -p "HttpPassword1234!" --storage-account MyStorageAccount \\
+        --enable-compute-isolation --workernode-size "Standard_E8S_V3" --headnode-size "Standard_E8S_V3"
 """
 
 helps['hdinsight list'] = """

src/azure-cli/azure/cli/command_modules/hdinsight/_params.py

@@ -224,6 +224,12 @@ def load_arguments(self, _):
         c.argument('enable_private_link', arg_group='Private Link', arg_type=get_three_state_flag(),
                    help='Indicate whether enable the private link or not.')
 
+        # compute isolation
+        c.argument('enable_compute_isolation', options_list=['--enable-compute-isolation', '--compute-isolation'],
+                   arg_group="Compute Isolation", arg_type=get_three_state_flag(),
+                   help='Indicate whether enable compute isolation or not.')
+        c.argument('host_sku', arg_group='Compute Isolation', help="The dedicated host sku of compute isolation.")
+
         # resize
         with self.argument_context('hdinsight resize') as c:
             c.argument('target_instance_count', options_list=['--workernode-count', '-c'],

src/azure-cli/azure/cli/command_modules/hdinsight/custom.py

@@ -36,8 +36,8 @@ def create_cluster(cmd, client, cluster_name, resource_group_name, cluster_type,
                    autoscale_type=None, autoscale_min_workernode_count=None, autoscale_max_workernode_count=None,
                    timezone=None, days=None, time=None, autoscale_workernode_count=None,
                    encryption_at_host=None, esp=False, idbroker=False,
-                   resource_provider_connection=None, enable_private_link=None,
-                   no_validation_timeout=False):
+                   resource_provider_connection=None, enable_private_link=None, enable_compute_isolation=None,
+                   host_sku=None, no_validation_timeout=False):
     from .util import build_identities_info, build_virtual_network_profile, parse_domain_name, \
         get_storage_account_endpoint, validate_esp_cluster_create_params
     from azure.mgmt.hdinsight.models import ClusterCreateParametersExtended, ClusterCreateProperties, OSType, \
@@ -46,7 +46,7 @@ def create_cluster(cmd, client, cluster_name, resource_group_name, cluster_type,
         DirectoryType, DiskEncryptionProperties, Tier, SshProfile, SshPublicKey, \
         KafkaRestProperties, ClientGroupInfo, EncryptionInTransitProperties, \
         Autoscale, AutoscaleCapacity, AutoscaleRecurrence, AutoscaleSchedule, AutoscaleTimeAndCapacity, \
-        NetworkProperties, PrivateLink
+        NetworkProperties, PrivateLink, ComputeIsolationProperties
 
     validate_esp_cluster_create_params(esp, cluster_name, resource_group_name, cluster_type,
                                        subnet, domain, cluster_admin_account, assign_identity,
@@ -350,6 +350,12 @@ def create_cluster(cmd, client, cluster_name, resource_group_name, cluster_type,
         private_link=PrivateLink.enabled if enable_private_link is True else PrivateLink.disabled
     )
 
+    # compute isolation
+    compute_isolation_properties = enable_compute_isolation and ComputeIsolationProperties(
+        enable_compute_isolation=enable_compute_isolation,
+        host_sku=host_sku
+    )
+
     create_params = ClusterCreateParametersExtended(
         location=location,
         tags=tags,
@@ -373,7 +379,8 @@ def create_cluster(cmd, client, cluster_name, resource_group_name, cluster_type,
             kafka_rest_properties=kafka_rest_properties,
             min_supported_tls_version=minimal_tls_version,
             encryption_in_transit_properties=encryption_in_transit_properties,
-            network_properties=network_properties
+            network_properties=network_properties,
+            compute_isolation_properties=compute_isolation_properties
         ),
         identity=cluster_identity
     )