Initial commit (#47268)

Azure · Nov 22, 2024 · 4d42d7b · 4d42d7b
1 parent a0363ff
commit 4d42d7b
Show file tree

Hide file tree

Showing 5 changed files with 877 additions and 10 deletions.
diff --git a/sdk/storage/Azure.Storage.DataMovement/src/DataTransferInternalState.cs b/sdk/storage/Azure.Storage.DataMovement/src/DataTransferInternalState.cs
@@ -109,7 +109,7 @@ public bool SetTransferState(DataTransferState state)
         public bool SetSkippedItemsState() => _status.SetSkippedItem();
 
         internal bool CanPause()
-            => DataTransferState.InProgress == _status.State;
+            => DataTransferState.InProgress == _status.State || DataTransferState.Queued == _status.State;
 
         public async Task PauseIfRunningAsync(CancellationToken cancellationToken)
         {

diff --git a/sdk/storage/Azure.Storage.DataMovement/src/TransferJobInternal.cs b/sdk/storage/Azure.Storage.DataMovement/src/TransferJobInternal.cs
@@ -283,7 +283,12 @@ public virtual async IAsyncEnumerable<JobPartInternal> ProcessJobToJobPartAsync(
         {
             try
             {
-                await OnJobStateChangedAsync(DataTransferState.InProgress).ConfigureAwait(false);
+                // only change state to 'InProgress' if it was 'Queued'
+                // (this is to prevent 'Pausing' to be changed)
+                if (_dataTransfer.TransferStatus.State == DataTransferState.Queued)
+                {
+                    await OnJobStateChangedAsync(DataTransferState.InProgress).ConfigureAwait(false);
+                }
             }
             catch (Exception ex)
             {