如何在取消异步任务时同步等待

Question

我有一个异步运行某些工作包的通用类。它有可能取消所有任务的执行并同步等待取消的任务完成。在新事务（用户做某事）开始之前触发取消。这是必要的，因为异步任务和新事务会更改相同的对象，但异步任务会在假定事务之前的状态时执行此操作。

这是为什么这种行为如此重要的示例代码：

private void Transaction()
{
    asyncExecution.AbortAllAsyncWork();
    DoTransaction();
}

此方法被同步调用，DoTransaction 更改对象，这些对象在异步任务中也发生更改。例如列表可以在迭代时更改。

以前，我在传递同步任务调度程序的任务上使用 ContinueWith 方法实现了这种行为。总而言之，这很难理解，而且看起来有点脏。因此，我想知道是否可以使用新的 async-await 功能实现相同的行为。这里的问题在于here 描述的死锁。到目前为止出现死锁问题的代码：

public void RunAsync<TWork, TResult>(IIncrementalAsyncExecutable<TWork, TResult> executable, TWork initialWork) where TWork : class
{
    Task workingTask = RunAsyncInternal(executable, initialWork, CancellationTokenSource);
    if (IsRunning(workingTask))
    {
        workingTasks.Add(workingTask);
    }
}

private async Task RunAsyncInternal<TWork, TResult>(IIncrementalAsyncExecutable<TWork, TResult> executable, TWork initialWork, CancellationTokenSource tokenSource) where TWork : class
{
    while (!executable.WorkDone)
    {
        TResult result = await Task.Run(() => executable.CalculateNextStep(initialWork));
        executable.SyncResult(result);
        if (tokenSource.IsCancellationRequested)
        {
            return;
        }
    }
}

public void AbortAllAsyncWork()
{
    CancellationTokenSource.Cancel();
    foreach (Task workingTask in workingTasks)
    {
        if (IsRunning(workingTask))
        {
            workingTask.Wait(); // here is the deadlock problem
        }
    }
}

是否有可能通过新的 async-await 功能实现此行为而不会出现死锁？

Answer 1

我有一个异步运行某些工作包的通用类。

为此使用内置类型（例如 TPL 数据流）要容易得多。他们已经完成了所有艰苦的工作。

问题是，应用程序是同步设计的。

是的，但请注意问题出在应用程序的设计中。等待任务完成本身就是一个异步操作，最佳的解决方案肯定是Yuval's。

之前我在传递同步任务调度程序的任务上使用 ContinueWith 方法实现了此行为。

我不明白这怎么可能避免你看到的僵局。

我想找到一个不需要重构的解决方案。

您真正要问的是“我如何进行异步同步”。 best answer is "don't"。但是您可以使用一些技巧来强制它在某些情况下工作。 没有始终有效的通用解决方案。

这些技巧是：只需阻塞，将操作推送到后台线程，然后运行嵌套消息循环。在Stephen Toub's blog 上有更详细的描述。

Answer 2

如果每个Task 都将被取消，您只需在它们上await 并抓住OperationCanceledException：

public async Task AbortAllAsyncWork()
{
    CancellationTokenSource.Cancel();
    foreach (Task workingTask in workingTasks.Keys)
    {
        if (IsRunning(workingTask))
        {
            try
            {
                await workingTask;
            }
            catch (OperationCanceledException oce)
            {
                // Do something usefull
            }
        }
    }
}

或者，您可以简单地 await Task.WhenAll 处理所有任务：

await Task.WhenAll(workingTasks.Keys)

假设Keys 是IEnumerable<Task>，则返回的Task 将处于Canceled 状态。