awaitを使用してカスタムTaskSchedulerでタスクを実行するにはどうすればよいですか？

Question

Task<T>に返信する方法があります。awaitを自由に使えます。私はそれらのタスクをデフォルトのものの代わりにカスタムTaskSchedulerで実行させたいと思っています。

var task = GetTaskAsync(); 
await task; 

私は新しいTaskFactory (new CustomScheduler())を作成し、それからStartNew()を行うことができます知っているが、StartNew()行動をとり、Taskを作成し、私はすでに（TaskCompletionSourceで舞台裏で返される）Task

を持っています

awaitのために自分自身のTaskSchedulerを指定するにはどうすればよいですか？

Answer 1

コメントの後、awaitが実行された後のコードのスケジューラを制御したいようです。

コンパイルは、デフォルトで現在のSynchronizationContextで実行される待機から継続を作成します。ですから、あなたのベストショットは、awaitを呼び出す前にSynchronizationContextを設定することです。

特定のコンテキストを待つ方法がいくつかあります。このような実装方法の詳細については、Jon SkeetのConfigure Await、特にSwitchToの部分を参照してください。

EDIT： TaskExのSwitchToメソッドが誤用されすぎて削除されました。理由はMSDN Forumを参照してください。

Answer 2

私は実際にはTask.Runをやってみたいと思っていますが、カスタムスケジューラを使っています。 StartNewは非同期メソッドでは直感的に動作しません。 Stephen Toubはthe differences between Task.Run and TaskFactory.StartNewについて素晴らしいブログ投稿をしています。

private static readonly TaskFactory myTaskFactory = new TaskFactory(
    CancellationToken.None, TaskCreationOptions.DenyChildAttach, 
    TaskContinuationOptions.None, new MyTaskScheduler()); 
private static Task RunOnMyScheduler(Func<Task> func) 
{ 
    return myTaskFactory.StartNew(func).Unwrap(); 
} 
private static Task<T> RunOnMyScheduler<T>(Func<Task<T>> func) 
{ 
    return myTaskFactory.StartNew(func).Unwrap(); 
} 
private static Task RunOnMyScheduler(Action func) 
{ 
    return myTaskFactory.StartNew(func); 
} 
private static Task<T> RunOnMyScheduler<T>(Func<T> func) 
{ 
    return myTaskFactory.StartNew(func); 
} 

その後、カスタムスケジューラの同期または非同期メソッドを実行することができます。

ので、独自のカスタムRunを作成するには、このような何かを行うことができます。

TaskCompletionSource<T>.Task

Answer 3

は、任意のアクションなしで構成され、スケジューラ は（Asynchronous Programming with the Reactive Framework and the Task Parallel Library — Part 3から）ContinueWith(...)の最初の呼び出し時に割り当てられます。

ありがたいことに、あなたはINotifyCompletionから独自のクラスの導出を実装して、タスクが（await anything;から）OnCompleted(Action continuation)方法で使用を開始する必要があり、スケジューラを設定するには、await SomeTask.ConfigureAwait(false)と同様のパターンで、それを使って少しのawaitの動作をカスタマイズすることができます。ここで

は、使用される：ここでは

TaskCompletionSource<object> source = new TaskCompletionSource<object>(); 

    public async Task Foo() { 
     // Force await to schedule the task on the supplied scheduler 
     await SomeAsyncTask().ConfigureScheduler(scheduler); 
    } 

    public Task SomeAsyncTask() { return source.Task; } 

はOnCompletedにおける重要な部分でのタスクの拡張メソッドを使用してConfigureSchedulerの単純な実装です：

public static class TaskExtension { 
    public static CustomTaskAwaitable ConfigureScheduler(this Task task, TaskScheduler scheduler) { 
     return new CustomTaskAwaitable(task, scheduler); 
    } 
} 

public struct CustomTaskAwaitable { 
    CustomTaskAwaiter awaitable; 

    public CustomTaskAwaitable(Task task, TaskScheduler scheduler) { 
     awaitable = new CustomTaskAwaiter(task, scheduler); 
    } 

    public CustomTaskAwaiter GetAwaiter() { return awaitable; } 

    public struct CustomTaskAwaiter : INotifyCompletion { 
     Task task; 
     TaskScheduler scheduler; 

     public CustomTaskAwaiter(Task task, TaskScheduler scheduler) { 
      this.task = task; 
      this.scheduler = scheduler; 
     } 

     public void OnCompleted(Action continuation) { 
      // ContinueWith sets the scheduler to use for the continuation action 
      task.ContinueWith(x => continuation(), scheduler); 
     } 

     public bool IsCompleted { get { return task.IsCompleted; } } 
     public void GetResult() { } 
    } 
} 

ここのようにコンパイルしますワーキングサンプルがありますコンソールアプリケーション：

using System; 
using System.Collections.Generic; 
using System.Runtime.CompilerServices; 
using System.Threading.Tasks; 

namespace Example { 
    class Program { 
     static TaskCompletionSource<object> source = new TaskCompletionSource<object>(); 
     static TaskScheduler scheduler = new CustomTaskScheduler(); 

     static void Main(string[] args) { 
      Console.WriteLine("Main Started"); 
      var task = Foo(); 
      Console.WriteLine("Main Continue "); 
      // Continue Foo() using CustomTaskScheduler 
      source.SetResult(null); 
      Console.WriteLine("Main Finished"); 
     } 

     public static async Task Foo() { 
      Console.WriteLine("Foo Started"); 
      // Force await to schedule the task on the supplied scheduler 
      await SomeAsyncTask().ConfigureScheduler(scheduler); 
      Console.WriteLine("Foo Finished"); 
     } 

     public static Task SomeAsyncTask() { return source.Task; } 
    } 

    public struct CustomTaskAwaitable { 
     CustomTaskAwaiter awaitable; 

     public CustomTaskAwaitable(Task task, TaskScheduler scheduler) { 
      awaitable = new CustomTaskAwaiter(task, scheduler); 
     } 

     public CustomTaskAwaiter GetAwaiter() { return awaitable; } 

     public struct CustomTaskAwaiter : INotifyCompletion { 
      Task task; 
      TaskScheduler scheduler; 

      public CustomTaskAwaiter(Task task, TaskScheduler scheduler) { 
       this.task = task; 
       this.scheduler = scheduler; 
      } 

      public void OnCompleted(Action continuation) { 
       // ContinueWith sets the scheduler to use for the continuation action 
       task.ContinueWith(x => continuation(), scheduler); 
      } 

      public bool IsCompleted { get { return task.IsCompleted; } } 
      public void GetResult() { } 
     } 
    } 

    public static class TaskExtension { 
     public static CustomTaskAwaitable ConfigureScheduler(this Task task, TaskScheduler scheduler) { 
      return new CustomTaskAwaitable(task, scheduler); 
     } 
    } 

    public class CustomTaskScheduler : TaskScheduler { 
     protected override IEnumerable<Task> GetScheduledTasks() { yield break; } 
     protected override bool TryExecuteTaskInline(Task task, bool taskWasPreviouslyQueued) { return false; } 
     protected override void QueueTask(Task task) { 
      TryExecuteTask(task); 
     } 
    } 
} 

Answer 4

このメソッドコールに適合することができますか：

await Task.Factory.StartNew(
     () => { /* to do what you need */ }, 
     CancellationToken.None, /* you can change as you need */ 
     TaskCreationOptions.None, /* you can change as you need */ 
     customScheduler);