Update Program.cs

Azure · Jan 6, 2022 · bcf2b81 · bcf2b81
1 parent 7385654
commit bcf2b81
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/Microsoft.Azure.Cosmos.Samples/Usage/ReEncryption/Program.cs b/Microsoft.Azure.Cosmos.Samples/Usage/ReEncryption/Program.cs
@@ -204,7 +204,10 @@ private static async Task CreateAndRunReEncryptionTasks()
 
             Console.WriteLine("\n ReEncryption in progress. Press esc key to exit. \n");
 
-            Task reEncryptionProgress = Program.CheckReEncryptionProgressOrCancelReEncryptionTasksAsync(sourceContainer, targetContainer, cancellationTokenSource);
+            Task reEncryptionProgress = Program.CheckReEncryptionProgressOrCancelReEncryptionTasksAsync(
+                sourceContainer,
+                targetContainer,
+                cancellationTokenSource);
 
             try
             {
@@ -285,7 +288,11 @@ private static async Task ExecuteReEncrytionAsync(
             ReEncryptionResponseMessage responseMessage;
             do
             {
-                responseMessage = await Program.ReEncryptNextAsync(sourceContainer, feedRange, continuationToken, cancellationToken);
+                responseMessage = await Program.ReEncryptNextAsync(
+                    sourceContainer,
+                    feedRange,
+                    continuationToken,
+                    cancellationToken);
 
                 if (responseMessage.ContinuationToken != null)
                 {
@@ -321,6 +328,7 @@ private static async Task<ReEncryptionResponseMessage> ReEncryptNextAsync(
             {
                 responseMessage = await iterator.EncryptNextAsync(cancellationToken);
                 File.WriteAllText(ContinuationTokenFile + sourceContainer.Id + feedRange.ToString(), responseMessage.ContinuationToken);
+
                 if (responseMessage.StatusCode == HttpStatusCode.NotModified)
                 {
                     break;
@@ -329,7 +337,10 @@ private static async Task<ReEncryptionResponseMessage> ReEncryptNextAsync(
 
             if (iterator.HasMoreResults == false)
             {
-                return new ReEncryptionResponseMessage(responseMessage, null, responseMessage.ReEncryptionBulkOperationResponse);
+                return new ReEncryptionResponseMessage(
+                    responseMessage: responseMessage,
+                    reEncryptionContinuationToken: null,
+                    reEncryptionBulkOperationResponse: responseMessage.ReEncryptionBulkOperationResponse);
             }
 
             return responseMessage;