TaskCompletionSourceWithCancellation.cs

Go to the documentation of this file.

namespace System.Threading.Tasks;
 
internal sealed class TaskCompletionSourceWithCancellation<T> : TaskCompletionSource<T>
{
    public TaskCompletionSourceWithCancellation()
        : base(TaskCreationOptions.RunContinuationsAsynchronously)
    {
    }
 
    public async ValueTask<T> WaitWithCancellationAsync(CancellationToken cancellationToken)
    {
        using (cancellationToken.UnsafeRegister(delegate(object s, CancellationToken cancellationToken)
        {
            ((TaskCompletionSourceWithCancellation<T>)s).TrySetCanceled(cancellationToken);
        }, this))
        {
            return await base.Task.ConfigureAwait(continueOnCapturedContext: false);
        }
    }
 
    public T WaitWithCancellation(CancellationToken cancellationToken)
    {
        using (cancellationToken.UnsafeRegister(delegate(object s, CancellationToken cancellationToken)
        {
            ((TaskCompletionSourceWithCancellation<T>)s).TrySetCanceled(cancellationToken);
        }, this))
        {
            return base.Task.GetAwaiter().GetResult();
        }
    }
 
    public ValueTask<T> WaitWithCancellationAsync(bool async, CancellationToken cancellationToken)
    {
        if (!async)
        {
            return new ValueTask<T>(WaitWithCancellation(cancellationToken));
        }
        return WaitWithCancellationAsync(cancellationToken);
    }
}