Task扩展方法取消操作

     /// <summary>
     /// 任务扩展，传入取消操作
     /// </summary>
     public static class TaskExtensionDemo
     {
         //因为没有非泛型的TaskCompleteSource类
         private struct Void { }

         public static async Task<TResult> WithCancellation<TResult>(this Task<TResult> originalTask, CancellationToken cancellationToken)
         {
             var cancelTask = new TaskCompletionSource<Void>();

             // 注册一个将在取消此 System.Threading.CancellationToken 时调用的委托。
             using (cancellationToken.Register(t => ((TaskCompletionSource<Void>)t).TrySetResult(new Void()), cancelTask))
             {
                 Task any = await Task.WhenAny(originalTask, cancelTask.Task);

                 if (any == cancelTask.Task)
                 {
                     cancellationToken.ThrowIfCancellationRequested();
                 }
             }

             return await originalTask;
         }

         public static async Task Go()
         {
             //在指定的延迟（以毫秒为单位）后将被取消。
             var cts = new CancellationTokenSource();
             var ct = cts.Token;

             try
             {
                 Task<Int64> t = new Task<Int64>(() =>
                 {
                     //Task.Delay不阻塞线程，任务不会被取消（5秒后早执行完了）
                     //Task.Delay(10000);

                     //阻塞线程，10秒后收到取消通知
                     Thread.Sleep();
                     return ;
                 });
                 t.Start();

                 //调用扩展方法取消任务
                 await t.WithCancellation(ct);

                 Console.WriteLine("Task completed");
             }
             catch (OperationCanceledException)
             {
                 Console.WriteLine("Task cancelled");
             }
         }
     }