I'm trying to create an extension method that will make any task attached to parent.
Extension code:
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();
}
}
Test Code:
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();
}
Parent task is cancelled immediately without waiting for inner tasks to complete. How can I solve it, as an input I'm getting a task which I'm execution in my parent dispatcher task. I want to execute any task as attached to parent.