Awaitable Cancellation Tokens

AwaitableCancellationTokenExtensions.cs

public static class AwaitableCancellationTokenExtensions
{
	public struct CancellationTokenAwaiter : ICriticalNotifyCompletion, INotifyCompletion
	{
		private readonly CancellationToken _cancellationToken;
		public CancellationTokenAwaiter(CancellationToken cancellationToken)
		{
			_cancellationToken = cancellationToken;
		}

		public void OnCompleted(Action continuation) => _cancellationToken.Register(continuation, false);
		public void UnsafeOnCompleted(Action continuation) => OnCompleted(continuation);
		public bool IsCompleted => _cancellationToken.IsCancellationRequested;
		public void GetResult() { }
	}

	public static CancellationTokenAwaiter GetAwaiter(this CancellationToken c) => new CancellationTokenAwaiter(c);
}

Enables awaiting CancellationToken

public async Task SomeMethod( string someArgument, CancellationToken cancellationToken ) {
  // ... 
  await cancellationToken;
}