CancellationToken.WhenCancelled()

CancellationTokenExtensions.cs

using System.Threading;
using System.Threading.Tasks;


public static class CancellationTokenExtensions
{
    public static async Task WhenCancelled(this CancellationToken cancellationToken)
    {
        TaskCompletionSource<bool> taskCompletionSource = new TaskCompletionSource<bool>();

        using (cancellationToken.Register(() =>
        {
            taskCompletionSource.TrySetResult(true);
        }))
        {
            await taskCompletionSource.Task;
        }
    }
}