Using async/await is so fun but in the same time there are plenty of situations which the awaiting part can stuck forever, specially when there is any involvement of programming over the network. Other than that there are situations which the app must not wait more than a timeout for a response.

Therefore, the ability to set a timeout for a asynchronous operation is a must, by using CancellationToken one can achieve this and this post will demonstrate an easier method of using these CancellationTokens by applying an extension method for the Task object type.

Implementation

NuGet Installation

.Net CLI
dotnet add package easyAsyncCancel

Package Manager
Install-Package easyAsyncCancel

Usage

Let's say there is a Task which it can take unknown amount of time to finish

private async Task Task_NetworkBound()
{
    await new HttpClient().GetStringAsync("https://dotnetfoundation.org");
}

In order for the task to get canceled after 1 seconds

await Task_NetworkBound().CancelAfter(1000);
OR
await Task_NetworkBound().CancelAfter(1000,"Custom OperationCanceledException message");

Test it

[TestFixture]
public class TaskCancellationTest
{
    private async Task Task_NetworkBound()
    {
        await new HttpClient().GetStringAsync("https://dotnetfoundation.org");
    }

    private async Task<string> Task_NetworkBound_T()
    {
        return await new HttpClient().GetStringAsync("https://dotnetfoundation.org");
    }

    [Test]
    public void Test_CancelWithMilliseconds()
    {
        Assert.ThrowsAsync<OperationCanceledException>(
            async () => { await Task_NetworkBound().CancelAfter(1000); });

        Assert.ThrowsAsync<OperationCanceledException>(
            async () => { await Task_NetworkBound_T().CancelAfter(1000); });
    }


    [Test]
    public void Test_CancelWithToken()
    {
        Assert.ThrowsAsync<OperationCanceledException>(async () =>
        {
            var cts = new CancellationTokenSource();
            cts.CancelAfter(1000);
            await Task_NetworkBound().CancelAfter(cts.Token);
        });


        Assert.ThrowsAsync<OperationCanceledException>(async () =>
        {
            var cts = new CancellationTokenSource();
            cts.CancelAfter(1000);
            await Task_NetworkBound_T().CancelAfter(cts.Token);
        });
    }
}

Full source code here

Happy coding :)