-
Notifications
You must be signed in to change notification settings - Fork 25
/
TaskExtensions.cs
48 lines (39 loc) · 1.61 KB
/
TaskExtensions.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
using System;
using System.Threading;
using System.Threading.Tasks;
namespace LiquidProjections.Testing
{
internal static class TaskExtensions
{
public static Task WithWaitCancellation(this Task task, CancellationToken cancellationToken)
{
var taskCompletionSource = new TaskCompletionSource<bool>();
CancellationTokenRegistration registration = cancellationToken.Register(CancelTask, taskCompletionSource);
task.ContinueWith(ContinueTask, Tuple.Create(taskCompletionSource, registration), CancellationToken.None,
TaskContinuationOptions.ExecuteSynchronously, TaskScheduler.Default);
return taskCompletionSource.Task;
}
private static void CancelTask(object state)
{
var taskCompletionSource = (TaskCompletionSource<bool>)state;
taskCompletionSource.TrySetCanceled();
}
private static void ContinueTask(Task task, object state)
{
var tcsAndRegistration = (Tuple<TaskCompletionSource<bool>, CancellationTokenRegistration>)state;
if (task.IsFaulted && (task.Exception != null))
{
tcsAndRegistration.Item1.TrySetException(task.Exception.InnerException);
}
if (task.IsCanceled)
{
tcsAndRegistration.Item1.TrySetCanceled();
}
if (task.IsCompleted)
{
tcsAndRegistration.Item1.TrySetResult(false);
}
tcsAndRegistration.Item2.Dispose();
}
}
}