当前的 SynchronizationContext 不能用作 TaskScheduler

Question

我正在使用Tasks 在我的 ViewModel 中运行长时间运行的服务器调用，结果使用TaskScheduler.FromSyncronizationContext() 编组回Dispatcher。例如：

var context = TaskScheduler.FromCurrentSynchronizationContext();
this.Message = "Loading...";
Task task = Task.Factory.StartNew(() => { ... })
            .ContinueWith(x => this.Message = "Completed"
                          , context);

当我执行应用程序时，这工作正常。但是当我在Resharper 上运行我的NUnit 测试时，我在调用FromCurrentSynchronizationContext 时收到错误消息：

当前的 SynchronizationContext 不能用作 TaskScheduler。

我猜这是因为测试是在工作线程上运行的。如何确保测试在主线程上运行？欢迎任何其他建议。

Answer 1

我已经结合了多种解决方案来保证 SynchronizationContext 工作：

using System;
using System.Threading;
using System.Threading.Tasks;

public class CustomSynchronizationContext : SynchronizationContext
{
    public override void Post(SendOrPostCallback action, object state)
    {
        SendOrPostCallback actionWrap = (object state2) =>
        {
            SynchronizationContext.SetSynchronizationContext(new CustomSynchronizationContext());
            action.Invoke(state2);
        };
        var callback = new WaitCallback(actionWrap.Invoke);
        ThreadPool.QueueUserWorkItem(callback, state);
    }
    public override SynchronizationContext CreateCopy()
    {
        return new CustomSynchronizationContext();
    }
    public override void Send(SendOrPostCallback d, object state)
    {
        base.Send(d, state);
    }
    public override void OperationStarted()
    {
        base.OperationStarted();
    }
    public override void OperationCompleted()
    {
        base.OperationCompleted();
    }

    public static TaskScheduler GetSynchronizationContext() {
      TaskScheduler taskScheduler = null;

      try
      {
        taskScheduler = TaskScheduler.FromCurrentSynchronizationContext();
      } catch {}

      if (taskScheduler == null) {
        try
        {
          taskScheduler = TaskScheduler.Current;
        } catch {}
      }

      if (taskScheduler == null) {
        try
        {
          var context = new CustomSynchronizationContext();
          SynchronizationContext.SetSynchronizationContext(context);
          taskScheduler = TaskScheduler.FromCurrentSynchronizationContext();
        } catch {}
      }

      return taskScheduler;
    }
}

用法：

var context = CustomSynchronizationContext.GetSynchronizationContext();

if (context != null) 
{
    Task.Factory
      .StartNew(() => { ... })
      .ContinueWith(x => { ... }, context);
}
else 
{
    Task.Factory
      .StartNew(() => { ... })
      .ContinueWith(x => { ... });
}

Answer 2

Ritch Melton 的解决方案对我不起作用。这是因为我的TestInitialize 函数是异步的，我的测试也是如此，所以每个await 都会丢失当前的SynchronizationContext。这是因为正如 MSDN 指出的那样，SynchronizationContext 类是“哑”的，只是将所有工作排入线程池。

对我有用的实际上只是在没有SynchronizationContext 时跳过FromCurrentSynchronizationContext 调用（即，如果当前上下文为null）。如果没有 UI 线程，我一开始就不需要同步。

TaskScheduler syncContextScheduler;
if (SynchronizationContext.Current != null)
{
    syncContextScheduler = TaskScheduler.FromCurrentSynchronizationContext();
}
else
{
    // If there is no SyncContext for this thread (e.g. we are in a unit test
    // or console scenario instead of running in an app), then just use the
    // default scheduler because there is no UI thread to sync with.
    syncContextScheduler = TaskScheduler.Current;
}

我发现这个解决方案比其他解决方案更简单，其中：

• 将TaskScheduler 传递给 ViewModel（通过依赖注入）
• 创建一个测试 SynchronizationContext 和一个“假” UI 线程以运行测试 - 对我来说更麻烦的是值得

我失去了一些线程细微差别，但我没有明确测试我的 OnPropertyChanged 回调是否在特定线程上触发，所以我可以接受。无论如何，使用new SynchronizationContext() 的其他答案并没有真正为该目标做得更好。

Answer 3

您需要提供一个 SynchronizationContext。我是这样处理的：

[SetUp]
public void TestSetUp()
{
  SynchronizationContext.SetSynchronizationContext(new SynchronizationContext());
}