Как реализовать ленивый источник TaskCompletion?

Question

Если Задача, открытая моим объектом TaskCompletionSource, никогда не может быть вызвана, как я могу выполнить вычисление результата, если и до тех пор, пока кто-то не выполнит задачу?

Например, я хочу заблокировать другие асинхронные потоки выполнения до тех пор, пока не будет выдано сообщение ManualResetEvent, используя следующую функцию WaitOneAsync. Я завершаю TaskCompleationSource в обратном вызове ThreadPool.RegisterWaitForSingleObject, который происходит, когда сигнализируется WaitHandle. Но если никто не ждет задания, то я не хочу RegisterWaitForSingleObject (и я не хочу RegisterWaitForSingleObject, если задача ожидается после сообщения WaitHandle).

Как я могу изменить WaitOneAsync так, чтобы работа по вычислению результата в RegisterWaitForSingleObject выполнялась только после того, как кто-то ждет TaskCompleationSource.Task?

Я считаю, что ответ может лежать в обычае TaskAwaiter, как описано здесь Implement AsyncManualResetEvent using Lazy<T> to determine if the task has been awaited Скотт Чемберлена, но я совсем не могу получить из его примера в моем решении ... :(

public static async Task<T> WaitOneAsync<T>(this WaitHandle waitHandle, Func<T> result) { 

    var tcs = new TaskCompletionSource<T>(); 

    RegisteredWaitHandle rwh = null; 
    rwh = ThreadPool.RegisterWaitForSingleObject(
     waitObject: waitHandle, 
     callBack: (s, t) => { 
      rwh.Unregister(null); 
      tcs.TrySetResult(result()); 
     }, 
     state: null, 
     millisecondsTimeOutInterval: -1, 
     executeOnlyOnce: true 
    ); 

    return await tcs.Task; 
} 

Answer 1

Как сказал Урр, невозможно что-то сделать в ответ на Task, являющийся await ред. Но если вы в порядке с использованием пользовательских ожидающих, то вы можете.

Самый простой способ сделать это состоит в использовании AsyncLazy from Stephen Cleary's AsyncEx:

private static Task<T> WaitOneAsyncImpl<T>(WaitHandle waitHandle, Func<T> result) 
{ 
    if (waitHandle.WaitOne(0)) 
     return Task.FromResult(result()); 

    var tcs = new TaskCompletionSource<T>(); 

    RegisteredWaitHandle rwh = null; 
    rwh = ThreadPool.RegisterWaitForSingleObject(
     waitObject: waitHandle, 
     callBack: (s, t) => 
     { 
      rwh.Unregister(null); 
      tcs.TrySetResult(result()); 
     }, 
     state: null, 
     millisecondsTimeOutInterval: -1, 
     executeOnlyOnce: true 
    ); 

    return tcs.Task; 
} 

public static AsyncLazy<T> WaitOneAsync<T>(this WaitHandle waitHandle, Func<T> result) 
    => new AsyncLazy<T>(() => WaitOneAsyncImpl(waitHandle, result)); 

Answer 2

Это не возможно точно так же, как вы описали свои требования. TPL не предоставляет событие или обратный вызов, когда кто-то добавляет продолжение или ждет выполнения задачи.

Таким образом, вам необходимо структурировать API так, чтобы на самом деле были созданы только те задачи, которые необходимы. Что это?

public static Func<Task<T>> CreateWaitOneAsyncFactory<T>(this WaitHandle waitHandle, Func<T> result) { 
return() => WaitOneAsync(waitHandle, result); 
} 

Это возвращает фабрику задач вместо задачи. Это обман, но единственные возможные решения включают в себя обман такого рода.

Вы также можете вернуть свой заказ. Но это вообще не связано с задачами, и оно не учитывает возможности компиляции задач. В основном это концепция C#. Выявление их может привести к нечистым API.

---

Unrelated на ваш вопрос: Вы можете удалить await tcs.Task и вернуть эту задачу непосредственно. Кроме того, функция result не требуется. Верните a Task, который не имеет результата. Затем абоненты могут добавить результат, если они того пожелают. Это делает API WaitOneAsync чище.