Clean up

Author: damianh

libs/little-forker/src/LittleForker/TaskExtensions.cs

@@ -17,37 +17,33 @@ internal static Task<T> WithCancellation<T>(this Task<T> task, CancellationToken
 
     internal static async Task TimeoutAfter(this Task task, TimeSpan timeout)
     {
-        using (var timeoutCancellationTokenSource = new CancellationTokenSource())
-        {
-            var completedTask = await Task
-                .WhenAny(task, Task.Delay(timeout, timeoutCancellationTokenSource.Token))
-                .ConfigureAwait(false);
-
-            if (completedTask == task)
-            {
-                timeoutCancellationTokenSource.Cancel();
-                await task.ConfigureAwait(false);
-                return;
-            }
+        using var timeoutCancellationTokenSource = new CancellationTokenSource();
+        var completedTask = await Task
+            .WhenAny(task, Task.Delay(timeout, timeoutCancellationTokenSource.Token))
+            .ConfigureAwait(false);
 
-            throw new TimeoutException("The operation has timed out.");
+        if (completedTask == task)
+        {
+            timeoutCancellationTokenSource.Cancel();
+            await task.ConfigureAwait(false);
+            return;
         }
+
+        throw new TimeoutException("The operation has timed out.");
     }
 
     internal static async Task<T> TimeoutAfter<T>(this Task<T> task, TimeSpan timeout)
     {
-        using (var timeoutCancellationTokenSource = new CancellationTokenSource())
+        using var timeoutCancellationTokenSource = new CancellationTokenSource();
+        var completedTask = await Task
+            .WhenAny(task, Task.Delay(timeout, timeoutCancellationTokenSource.Token))
+            .ConfigureAwait(false);
+        if (completedTask == task)
         {
-            var completedTask = await Task
-                .WhenAny(task, Task.Delay(timeout, timeoutCancellationTokenSource.Token))
-                .ConfigureAwait(false);
-            if (completedTask == task)
-            {
-                timeoutCancellationTokenSource.Cancel();
-                return await task.ConfigureAwait(false);
-            }
-
-            throw new TimeoutException("The operation has timed out.");
+            timeoutCancellationTokenSource.Cancel();
+            return await task.ConfigureAwait(false);
         }
+
+        throw new TimeoutException("The operation has timed out.");
     }
 }