Add retries for Build image in Azure (#577)

* Add retries for building image

* move run

* Remove line

* change

Author: motm32

package-lock.json

@@ -631,6 +631,7 @@
         "dotenv": "^16.0.0",
         "fs-extra": "^8.1.0",
         "semver": "^7.5.2",
+        "p-retry": "^4.6.2",
         "vscode-nls": "^4.1.1",
         "vscode-uri": "^3.0.2"
     },

package.json

@@ -5,8 +5,10 @@
 
 import { type DockerBuildRequest as AcrDockerBuildRequest } from "@azure/arm-containerregistry";
 import { AzExtFsExtra, GenericParentTreeItem, activityFailContext, activityFailIcon } from "@microsoft/vscode-azext-utils";
+import * as retry from 'p-retry';
 import * as path from 'path';
 import { type Progress } from "vscode";
+import { ext } from "../../../../extensionVariables";
 import { ExecuteActivityOutputStepBase, type ExecuteActivityOutput } from "../../../../utils/activity/ExecuteActivityOutputStepBase";
 import { createActivityChildContext } from "../../../../utils/activity/activityUtils";
 import { localize } from "../../../../utils/localize";
@@ -27,10 +29,20 @@ export class RunStep extends ExecuteActivityOutputStepBase<BuildImageInAzureImag
                 dockerFilePath: path.basename(context.dockerfilePath) /* Assume the dockerfile is always in the root of the source */
             };
 
-            const building: string = localize('buildingImage', 'Building image...');
-            progress.report({ message: building });
+            const retries = 3;
+            await retry(
+                async (currentAttempt: number): Promise<void> => {
+                    const message: string = currentAttempt === 1 ?
+                        localize('buildingImage', 'Building image...') :
+                        localize('buildingImageAttempt', 'Building image (Attempt {0}/{1})...', currentAttempt, retries + 1);
+                    progress.report({ message: message });
+                    ext.outputChannel.appendLog(message);
 
-            context.run = await context.client.registries.beginScheduleRunAndWait(context.resourceGroupName, context.registryName, runRequest);
+                    context.run = await context.client.registries.beginScheduleRunAndWait(context.resourceGroupName, context.registryName, runRequest);
+
+                },
+                { retries, minTimeout: 2 * 1000 }
+            );
         } finally {
             if (await AzExtFsExtra.pathExists(context.tarFilePath)) {
                 await AzExtFsExtra.deleteResource(context.tarFilePath);