Как объединить TaskCompletionSource и CancellationTokenSource?

Question

У меня есть такой код (упрощенный здесь), который ожидает отделочные задачи:

var task_completion_source = new TaskCompletionSource<bool>(); 
observable.Subscribe(b => 
    { 
     if (b) 
      task_completion_source.SetResult(true); 
    }); 
await task_completion_source.Task;  

Идея заключается в том, чтобы подписаться и ждать true в потоке булевы. Это завершает «задачу», и я могу двигаться дальше await.

Однако я хотел был бы отменить - но не подписку, но ожидающую. Я хотел бы передать маркер отмены (как-то) на task_completion_source, поэтому, когда я отменяю токен, await будет двигаться дальше.

Как это сделать?

Обновление: CancellationTokenSource является внешним в этом коде, и все, что у меня есть, это его токен.

Answer 1

Если я вас правильно понимаю, вы можете сделать это следующим образом:

CancellationToken ct; 
ct.Register(() => 
{ 
    // this callback will be executed when token is cancelled 
    task_comletion_source.TrySetCanceled(); 
}); 

Обратите внимание, что он будет бросать исключение на вашем ЖДУТ, которые вы должны справиться.

Answer 2

Я рекомендую вам не строить это самостоятельно. Существует ряд крайних случаев вокруг маркеров отмены, которые являются утомительными для получения права. Например, если регистрация, возвращенная с Register, никогда не удаляется, вы можете получить утечку ресурсов.

Вместо этого, вы можете использовать метод Task.WaitAsync расширения от моего AsyncEx.Tasks library:

var task_completion_source = new TaskCompletionSource<bool>(); 
observable.Subscribe(b => 
{ 
    if (b) 
    task_completion_source.SetResult(true); 
}); 
await task_completion_source.Task.WaitAsync(cancellationToken); 

На стороне записки, я настоятельно рекомендую вам использовать ToTask вместо явного TaskCompletionSource. Опять же, ToTask отлично подходит для вас.