親にアタッチされたタスクを作成する拡張メソッドを作成しようとしています。
延長コード:
internal static class TaskHelpers
{
public static Task AttachToParrent(this Task task)
{
var tsc = new TaskCompletionSource<Task>(task.CreationOptions & TaskCreationOptions.AttachedToParent);
task.ContinueWith(_ => tsc.TrySetResult(task), TaskContinuationOptions.OnlyOnRanToCompletion);
task.ContinueWith(t => tsc.TrySetException(task.Exception ?? new AggregateException(new ApplicationException("Unknown"))), TaskContinuationOptions.OnlyOnFaulted);
task.ContinueWith(t => tsc.TrySetCanceled(), TaskContinuationOptions.OnlyOnCanceled);
return tsc.Task.Unwrap();
}
}
テストコード:
static void Main(string[] args)
{
var cancellationTokenSource = new CancellationTokenSource();
var task1 = Task.Factory.StartNew(() =>
{
var task2 = Task.Factory.StartNew(() =>
{
for (int i = 0; i < 5; i++)
{
if (!cancellationTokenSource.IsCancellationRequested)
{
Console.WriteLine(i);
Thread.Sleep(5000);
}
else
{
Console.WriteLine("Loop cancelled.");
break;
}
}
}, cancellationTokenSource.Token).AttachToParrent();
var task3 = Task.Factory.StartNew(() =>
{
for (int i = 0; i < 5; i++)
{
if (!cancellationTokenSource.IsCancellationRequested)
{
Console.WriteLine(i*10);
Thread.Sleep(5000);
}
else
{
Console.WriteLine("Loop cancelled.");
break;
}
}
}, cancellationTokenSource.Token).AttachToParrent();
}, cancellationTokenSource.Token);
Console.WriteLine("Enter to cancel");
Console.ReadKey();
cancellationTokenSource.Cancel();
Console.WriteLine("Waiting To Be Cancelled");
task1.Wait();
Console.WriteLine("Task Cancelled");
Console.ReadKey();
}
親タスクは、内部タスクの完了を待たずにすぐにキャンセルされます。入力として、親ディスパッチャータスクで実行しているタスクを取得しています。親にアタッチされたタスクを実行したい。