Add async param to update and create

Author: amberwang113

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

@@ -126,13 +126,15 @@ def load_arguments(self, _):
                                                                                  'the App Service plan that hosts it.')
         c.argument('zone_redundant', options_list=['--zone-redundant', '-z'], help='Enable zone redundancy for high availability. Minimum instance count is 2.')
         c.argument('tags', arg_type=tags_type)
+        c.argument('async_scaling_enabled', action='store_true', help='Enable async scaling for the app service plan. This will create an async operation if there are insufficient workers to scale synchronously.')
 
     with self.argument_context('appservice plan update') as c:
         c.argument('sku', arg_type=sku_arg_type)
         c.argument('elastic_scale', arg_type=get_three_state_flag(), is_preview=True, help='Enable or disable automatic scaling. Set to "true" to enable elastic scale for this plan, or "false" to disable elastic scale for this plan. The SKU must be a Premium V2 SKU (P1V2, P2V2, P3V2) or a Premium V3 SKU (P1V3, P2V3, P3V3)')
         c.argument('max_elastic_worker_count', options_list=['--max-elastic-worker-count', '-m'], type=int, is_preview=True, help='Maximum number of instances that the plan can scale out to. The plan must be an elastic scale plan.')
         c.argument('number_of_workers', type=int, help='Number of workers to be allocated.')
         c.ignore('allow_pending_state')
+        c.argument('async_scaling_enabled', arg_type=get_three_state_flag(), help='Enable async scaling for the app service plan. This will create an async operation if there are insufficient workers to scale synchronously.')
 
     with self.argument_context('appservice plan delete') as c:
         c.argument('name', arg_type=name_arg_type, help='The name of the app service plan',

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

@@ -3965,7 +3965,7 @@ def _enable_zone_redundant(plan_def, sku_def, number_of_workers):
 
 def create_app_service_plan(cmd, resource_group_name, name, is_linux, hyper_v, per_site_scaling=False,
                             app_service_environment=None, sku='B1', number_of_workers=None, location=None,
-                            tags=None, no_wait=False, zone_redundant=False):
+                            tags=None, no_wait=False, zone_redundant=False, async_scaling_enabled=None):
     HostingEnvironmentProfile, SkuDescription, AppServicePlan = cmd.get_models(
         'HostingEnvironmentProfile', 'SkuDescription', 'AppServicePlan')
 
@@ -4004,7 +4004,8 @@ def create_app_service_plan(cmd, resource_group_name, name, is_linux, hyper_v, p
     sku_def = SkuDescription(tier=get_sku_tier(sku), name=_normalize_sku(sku), capacity=number_of_workers)
     plan_def = AppServicePlan(location=location, tags=tags, sku=sku_def,
                               reserved=(is_linux or None), hyper_v=(hyper_v or None),
-                              per_site_scaling=per_site_scaling, hosting_environment_profile=ase_def)
+                              per_site_scaling=per_site_scaling, hosting_environment_profile=ase_def,
+                              async_scaling_enabled=async_scaling_enabled)
 
     if sku.upper() in ['WS1', 'WS2', 'WS3']:
         existing_plan = get_resource_if_exists(client.app_service_plans,
@@ -4022,11 +4023,17 @@ def create_app_service_plan(cmd, resource_group_name, name, is_linux, hyper_v, p
 
 
 def update_app_service_plan(cmd, instance, sku=None, number_of_workers=None, elastic_scale=None,
-                            max_elastic_worker_count=None):
-    if number_of_workers is None and sku is None and elastic_scale is None and max_elastic_worker_count is None:
+                            max_elastic_worker_count=None, async_scaling_enabled=None):
+    if (number_of_workers is None and sku is None and
+    elastic_scale is None and max_elastic_worker_count is None and
+    async_scaling_enabled is None):
         safe_params = cmd.cli_ctx.data['safe_params']
         if '--set' not in safe_params:
-            args = ["--number-of-workers", "--sku", "--elastic-scale", "--max-elastic-worker-count"]
+            args = ["--number-of-workers", 
+                    "--sku", 
+                    "--elastic-scale", 
+                    "--max-elastic-worker-count", 
+                    "--async-scaling-enabled"]
             logger.warning('Nothing to update. Set one of the following parameters to make an update: %s', str(args))
     sku_def = instance.sku
     if sku is not None:
@@ -4062,10 +4069,12 @@ def update_app_service_plan(cmd, instance, sku=None, number_of_workers=None, ela
         use_additional_properties(instance)
         instance.additional_properties["properties"]["maximumElasticWorkerCount"] = max_elastic_worker_count
 
+    if async_scaling_enabled is not None:
+        instance.async_scaling_enabled = async_scaling_enabled
+
     instance.sku = sku_def
     return instance
 
-
 def show_plan(cmd, resource_group_name, name):
     from azure.cli.core.commands.client_factory import get_subscription_id
     client = web_client_factory(cmd.cli_ctx)