added kudu warm-up for arm

Author: Shikha Jha

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

@@ -7425,6 +7425,25 @@ def _build_onedeploy_arm_url(params):
     return params.cmd.cli_ctx.cloud.endpoints.resource_manager + base_url
 
 
+def _build_onedeploy_arm_url_with_instance(params, instance_id):
+    from azure.cli.core.commands.client_factory import get_subscription_id
+    client = web_client_factory(params.cmd.cli_ctx)
+    sub_id = get_subscription_id(params.cmd.cli_ctx)
+    if not params.slot:
+        base_url = (
+            f"subscriptions/{sub_id}/resourceGroups/{params.resource_group_name}/providers/Microsoft.Web/sites/"
+            f"{params.webapp_name}/instances/{instance_id}/extensions/"
+            f"onedeploy?api-version={client.DEFAULT_API_VERSION}"
+        )
+    else:
+        base_url = (
+            f"subscriptions/{sub_id}/resourceGroups/{params.resource_group_name}/providers/Microsoft.Web/sites/"
+            f"{params.webapp_name}/slots/{params.slot}/instances/{instance_id}/extensions/onedeploy"
+            f"?api-version={client.DEFAULT_API_VERSION}"
+        )
+    return params.cmd.cli_ctx.cloud.endpoints.resource_manager + base_url
+
+
 def _build_deploymentstatus_url(cmd, resource_group_name, webapp_name, slot, deployment_id):
     from azure.cli.core.commands.client_factory import get_subscription_id
     client = web_client_factory(cmd.cli_ctx)
@@ -7593,16 +7612,19 @@ def _make_onedeploy_request(params):
     # For debugging purposes only, you can change the async deployment into a sync deployment by polling the API status
     # For that, set poll_async_deployment_for_debugging=True
     logger.info("Deployment API: %s", deploy_url)
+    logger.warning("Warming up Kudu before deployment.")
+    cookies = _warmup_kudu_and_get_cookie_internal(params.cmd, params.resource_group_name,
+                                                   params.webapp_name, params.slot)
+    cookies_present = True
+    if cookies is None:
+        cookies_present = False
+        logger.info("Failed to fetch affinity cookie for Kudu. "
+                    "Deployment will proceed without pre-warming a Kudu instance.")
     if not params.src_url:  # use SCM endpoint
         # if linux webapp and not function app, then warmup kudu and use warmed up kudu for deployment
         if params.is_linux_webapp and not params.is_functionapp and params.enable_kudu_warmup:
             try:
-                logger.warning("Warming up Kudu before deployment.")
-                cookies = _warmup_kudu_and_get_cookie_internal(params.cmd, params.resource_group_name,
-                                                               params.webapp_name, params.slot)
-                if cookies is None:
-                    logger.info("Failed to fetch affinity cookie for Kudu. "
-                                "Deployment will proceed without pre-warming a Kudu instance.")
+                if not cookies_present:
                     response = requests.post(deploy_url, data=body, headers=headers,
                                              verify=not should_disable_connection_verify())
                 else:
@@ -7618,7 +7640,19 @@ def _make_onedeploy_request(params):
                                      verify=not should_disable_connection_verify())
         poll_async_deployment_for_debugging = True
     else:
-        response = send_raw_request(params.cmd.cli_ctx, "PUT", deploy_url, body=body)
+        if params.is_linux_webapp and not params.is_functionapp and params.enable_kudu_warmup:
+            try:
+                if not cookies_present:
+                    response = send_raw_request(params.cmd.cli_ctx, "PUT", deploy_url, body=body)
+                else:
+                    deploy_arm_url = _build_onedeploy_arm_url_with_instance(params, cookies.get("ARRAffinity"))
+                    response = send_raw_request(params.cmd.cli_ctx, "PUT", deploy_arm_url, body=body)
+            except Exception as ex:  # pylint: disable=broad-except
+                logger.info("Failed to deploy using instances endpoint. "
+                            "Deployment will proceed without pre-warming a Kudu instance. Ex: %s", ex)
+                response = send_raw_request(params.cmd.cli_ctx, "PUT", deploy_url, body=body)
+        else:
+            send_raw_request(params.cmd.cli_ctx, "PUT", deploy_url, body=body)
         poll_async_deployment_for_debugging = False
 
     # check the status of deployment