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(5000);
            var ct = cts.Token;

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

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

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

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