TaskScheduler.Default不总是保证任务将在池线程上执行吗?

时间:2014-08-03 04:42:42

标签: c# .net task-parallel-library async-await task

TaskScheduler.Default 是否始终保证任务将在池线程上执行?

在修复错误时,我发现至少有一个案例没有。它可以像这样再现(一个由实际代码制作的人为例子):

var tcs = new TaskCompletionSource<bool>();
var sc = SynchronizationContext.Current;
sc.Post(_ => tcs.SetResult(true), null);
await tcs.Task.ContinueWith(_ =>
    {
        // breaks here
        Debug.Assert(Thread.CurrentThread.IsThreadPoolThread);
    }, 
    CancellationToken.None, 
    TaskContinuationOptions.ExecuteSynchronously, TaskScheduler.Default);

还有其他案件吗?

此外,是否有一种优雅的方法可以确保同步执行ContinueWith操作,如果先前任务已在池线程上完成,或者排队等待线程池(我知道我可以使用{{ 1}}在QueueUserWorkItem行动中,但我不喜欢它。)

已编辑,我想我可以实现自己的ContinueWith,并检查我是否已经在TaskScheduler内的线程池线程中,以控制它。

1 个答案:

答案 0 :(得分:5)

我认为发生这种情况的原因是因为您使用了TaskContinuationOptions.ExecuteSynchronously。来自doco:

  

ExecuteSynchronously指定延续任务应该是   同步执行。指定此选项后,继续   将在导致前期任务的同一线程上运行   过渡到最终状态。如果先行者已经完成   当创建延续时,继续将在   线程创建延续。只是非常短暂的运行   延续应该同步执行。

如果先前任务在线程池线程以外的线程上完成,那么继续也将在该线程上运行。所以我想在那种情况下不会发生任何调度。另一种情况可能是延续任务已经完成,也将同步运行。

<强>更新

为了达到你在问题的第二部分所提出的要求,我认为你需要一个定制的等待者。有关详细信息,请参阅this articl e,但类似的内容可能对您有用:

public static class Extensions
{
    public static ThreadPoolTaskAwaiter WithThreadPool(this Task task)
    {
        return new ThreadPoolTaskAwaiter(task);
    }

    public class ThreadPoolTaskAwaiter : INotifyCompletion
    {
        private readonly TaskAwaiter m_awaiter;

        public ThreadPoolTaskAwaiter(Task task)
        {
            if (task == null) throw new ArgumentNullException("task");
            m_awaiter = task.GetAwaiter();
        }

        public ThreadPoolTaskAwaiter GetAwaiter() { return this; }

        public bool IsCompleted { get { return m_awaiter.IsCompleted; } }

        public void OnCompleted(Action continuation)
        {
            if (Thread.CurrentThread.IsThreadPoolThread)
            {
                continuation();
            }
            else
            {
                Task.Run(continuation);
            }                
        }

        public void GetResult()
        {
            m_awaiter.GetResult();
        }
    }
}

然后你就这样使用它:

public static async Task Execute()
{          
    await Task.Delay(500).WithThreadPool();

    // does not break here
    if (!Thread.CurrentThread.IsThreadPoolThread)
    {
        Debugger.Break();
    }
}