small fix

sillsdev · Sep 5, 2024 · 986b97e · 986b97e
1 parent 9a0cb81
commit 986b97e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Serval/src/Serval.Shared/Utils/TaskEx.cs b/src/Serval/src/Serval.Shared/Utils/TaskEx.cs
@@ -35,13 +35,14 @@ public static async Task Timeout(
             var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
             Task task = action(cts.Token);
             Task delayTask = Task.Delay(timeout, cancellationToken);
-            await Task.WhenAny(task, delayTask);
+            Task completedTask = await Task.WhenAny(task, delayTask);
             if (delayTask.Status == TaskStatus.RanToCompletion)
             {
                 if (throwOnTimeout)
                     throw new TimeoutException($"Operation timed out after {timeout}");
                 cts.Cancel();
             }
+            await completedTask;
         }
     }