fix: Resume session should complete task insertion into the queue

aneoconsulting · Nov 28, 2024 · 8d9671e · 8d9671e
1 parent 2e55acc
commit 8d9671e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Common/src/Storage/TaskLifeCycleHelper.cs b/Common/src/Storage/TaskLifeCycleHelper.cs
@@ -535,18 +535,20 @@ public static async Task<SessionData> ResumeAsync(ITaskTable        taskTable,
                                                          cancellationToken)
                                           .ConfigureAwait(false))
       {
+        cancellationToken.ThrowIfCancellationRequested();
+
         var taskIds = tasks.Select(task => task.TaskId)
                            .AsICollection();
 
         await taskTable.UpdateManyTasks(data => data.SessionId == sessionId && data.Status == TaskStatus.Paused && taskIds.Contains(data.TaskId),
                                         new UpdateDefinition<TaskData>().Set(data => data.Status,
                                                                              TaskStatus.Submitted),
-                                        cancellationToken)
+                                        CancellationToken.None)
                        .ConfigureAwait(false);
 
         await pushQueueStorage.PushMessagesAsync(tasks,
                                                  grouping.Key.PartitionId,
-                                                 cancellationToken)
+                                                 CancellationToken.None)
                               .ConfigureAwait(false);
       }
     }