c# - 如何等待IAsyncEnumerable <>的所有结果？

Question

我正在修改C#8.0中的新IAsyncEnumerable<T>内容。假设我有一些我想使用的方法:

public IAsyncEnumerable<T> SomeBlackBoxFunctionAsync<T>(...) { ... }

我知道我可以将其与 await foreach...语法一起使用。但是，可以说，我的使用者需要从该函数获得所有结果，然后才能继续。在继续之前等待所有结果的最佳语法是什么？换句话说，我希望能够执行以下操作:

// but that extension - AllResultsAsync() - doesn't exist :-/
List<T> myList = await SomeBlackBoxFunctionAsync<T>().AllResultsAsync(); 

正确的方法是什么？

Answer 1

首先发出警告:按照定义，异步流可​​能永远不会结束，并且会继续产生结果，直到应用程序终止。例如，已经在SignalR或gRPC中使用了它。轮询循环也以这种方式工作。

在异步流上使用ToListAsync可能会产生意想不到的后果。

像这样的运算符(operator)已经可以通过System.Linq.Async包获得。

通过ToListAsync可以使用整个流。该代码*看似简单，但隐藏了一些有趣的问题:

public static ValueTask<List<TSource>> ToListAsync<TSource>(this IAsyncEnumerable<TSource> source, CancellationToken cancellationToken = default)
{
    if (source == null)
        throw Error.ArgumentNull(nameof(source));

    if (source is IAsyncIListProvider<TSource> listProvider)
        return listProvider.ToListAsync(cancellationToken);

    return Core(source, cancellationToken);

    static async ValueTask<List<TSource>> Core(IAsyncEnumerable<TSource> source, CancellationToken cancellationToken)
    {
        var list = new List<TSource>();

        await foreach (var item in source.WithCancellation(cancellationToken).ConfigureAwait(false))
        {
            list.Add(item);
        }

        return list;
    }
}

首先，它返回 ValueTask。其次，它确保观察到取消操作并使用 ConfigureAwait(false)来防止死锁。最后，如果源已经提供了自己的 ToListAsync实现，则运算符(operator)将遵循此实现。