Adding Feature Flag to RoboCopy in WindowsMachineFileCopyV2 (#20746)

* Adding Feature Flag to RoboCopy in WindowsMachineFileCopyV2 * updating task version
microsoft · Dec 31, 2024 · 93a10ac · 93a10ac
1 parent f7d65f3
commit 93a10ac
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 5 deletions.
diff --git a/Tasks/WindowsMachineFileCopyV2/RoboCopyJob.ps1 b/Tasks/WindowsMachineFileCopyV2/RoboCopyJob.ps1
@@ -1,4 +1,9 @@
-$CopyJob = {
+$featureFlags = @{
+    ModifyNumberOfRetriesInRobocopy = [System.Convert]::ToBoolean($env:MODIFY_NUMBER_OF_RETRIES_IN_ROBOCOPY)
+} 
+$ModifyRoboCopyRetries= $featureFlags.ModifyNumberOfRetriesInRobocopy
+
+$CopyJob = {
 param (
     [string]$fqdn, 
     [string]$sourcePath,
@@ -101,7 +106,12 @@ param (
 
     function Clean-Target
     {
-        $cleanupArgument = "/NOCOPY /E /PURGE" 
+        if(-not $ModifyRoboCopyRetries){
+          $cleanupArgument = "/NOCOPY /E /PURGE" 
+        }
+        else{
+          $cleanupArgument = "/NOCOPY /E /PURGE /R:3 " 
+        }
         $guid = [GUID]::NewGuid()
         $tempDirectory = "$scriptRoot\temp$guid" 
         New-Item -ItemType Directory -Force -Path $tempDirectory         
@@ -128,7 +138,12 @@ param (
         [switch]$fileCopy
         )
     {
-        $robocopyParameters = "/COPY:DAT "
+        if(-not $ModifyRoboCopyRetries){
+          $robocopyParameters = "/COPY:DAT "
+        }
+        else{
+          $robocopyParameters = "/COPY:DAT /R:3 "
+        }
 
         if(-not $fileCopy.IsPresent)
         {

diff --git a/Tasks/WindowsMachineFileCopyV2/task.json b/Tasks/WindowsMachineFileCopyV2/task.json
@@ -13,7 +13,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 2,
-    "Minor": 242,
+    "Minor": 251,
     "Patch": 0
   },
   "releaseNotes": "What's new in Version 2.0: <br/>&nbsp;&nbsp;Proxy support is being added. <br/>&nbsp;&nbsp; Removed support of legacy DTL machines.",

diff --git a/Tasks/WindowsMachineFileCopyV2/task.loc.json b/Tasks/WindowsMachineFileCopyV2/task.loc.json
@@ -13,7 +13,7 @@
   "author": "Microsoft Corporation",
   "version": {
     "Major": 2,
-    "Minor": 242,
+    "Minor": 251,
     "Patch": 0
   },
   "releaseNotes": "ms-resource:loc.releaseNotes",