如何創建發佈到給定SynchronizationContext的任務？

Question

鑑於我已經擁有的SynchronizationContext（並且基本上是特定主題的窗口），我如何創建發佈到此上下文的Task？

作爲參考，這裏是一個非常基本的示範如何設置SynchronizationContext。

public class SomeDispatcher : SynchronizationContext 
{ 
    SomeDispatcher() { 

     new Thread(() => { 

      SynchronizationContext.SetSynchronizationContext(this); 

      // Dispatching loop (among other things) 

     }).Start(); 
    } 

    override void Post(SendOrPostCallback d, object state) 
    { 
     // Add (d, state) to a dispatch queue; 
    } 
} 

這對於異步/等待那些已經在上下文中運行工作正常。

現在，我想能夠從外部上下文（例如從UI線程）發佈Task s，但似乎無法找到乾淨的方式來做到這一點。

一種方法是使用TaskCompletionSource<>。

Task StartTask(Action action) 
{ 
    var tcs = new TaskCompletionSource<object>(); 
    SaidDispatcher.Post(state => { 
     try 
     { 
      action.Invoke(); 
      tcs.SetResult(null); 
     } 
     catch (Exception ex) 
     { 
      tcs.SetException(ex); 
     } 
    }); 
    return tcs.Task; 
}); 

但這是重新發明輪子和一個重大的痛苦支持的變化，如StartNew(Func<TResult>)，StartNew(Func<Task<TResult>>)等

一個TaskFactory接口的SynchronizationContext可能是最好的，但我似乎無法實例化一個清潔：

TaskFactory CreateTaskFactory() 
{ 
    var original = SynchronizationContext.Current; 
    SynchronizationContext.SetSynchronizationContext(SomeDispatcher); // yuck! 
    try 
    { 
     return new TaskFactory(TaskScheduler.FromCurrentSynchronizationContext()); 
    } 
    finally 
    { 
     SynchronizationContext.SetSynchronizationContext(original); 
    } 
} 

（即具有臨時軟管當前線程的同步上下文似乎哈克。）

Answer 1

似乎默認SynchronizationContextTaskScheduler是

1. 內部
2. 只適用於當前同步上下文

但它的源代碼可用here，我們看到它的比較簡單，所以我們可以嘗試推出我們自己的調度程序，像這樣：

public sealed class MySynchronizationContextTaskScheduler : TaskScheduler { 
    private readonly SynchronizationContext _synchronizationContext; 

    public MySynchronizationContextTaskScheduler(SynchronizationContext context) { 
     _synchronizationContext = context; 
    } 

    [SecurityCritical] 
    protected override void QueueTask(Task task) { 
     _synchronizationContext.Post(PostCallback, task); 
    } 

    [SecurityCritical] 
    protected override bool TryExecuteTaskInline(Task task, bool taskWasPreviouslyQueued) { 
     if (SynchronizationContext.Current == _synchronizationContext) { 
      return TryExecuteTask(task); 
     } 
     else 
      return false; 
    } 

    [SecurityCritical] 
    protected override IEnumerable<Task> GetScheduledTasks() { 
     return null; 
    } 

    public override Int32 MaximumConcurrencyLevel 
    { 
     get { return 1; } 
    } 

    private void PostCallback(object obj) { 
     Task task = (Task) obj; 
     base.TryExecuteTask(task); 
    } 
} 

然後你的CreateTaskFactory變爲：

TaskFactory CreateTaskFactory() { 
    return new TaskFactory(new MySynchronizationContextTaskScheduler(SomeDispatcher)); 
} 

你創建任務：

var factory = CreateTaskFactory(); 
var task = factory.StartNew(...); 

Answer 2

Parallel Extensions Extras contains SynchronizationContextTaskScheduler這不正是你想要的東西。

如果您不想自己編譯PEE，請撥打there is an unofficial NuGet package for it。

請注意，您通常不需要這樣做，而且您要求提供這一點的事實可能表明您的設計存在缺陷。