如何使用 IAsyncEnumerable 进行并行化

Question

我有一种情况，我正在启动 Task-s，我希望他们的结果以某种方式尽快在数据结构中通过管道传输/排队，而不关心他们的顺序。

IAsyncEnumerable 适合这种情况吗？

public async Task BigMethod()
{
    Queue<int> queue = new Queue<int>();
    foreach (var item in RunJobsAsync())
    {
        queue.Enqueue(item);
    }
    //await foreach(var item in await RunIAsyncJobsAsync())
    // {
    //     queue.Enqueue(item);
    // }
    // [some more code]
}

没有IAsyncEnumerable

public async Task<IEnumerable<int> RunJobsAsync()
{    
    List<Task<int>> tasks = new List<Task<int>>();
    foreach(var x in Enumerable.Range(0,100))
    {
        tasks.Add(Task.Run(async()=> await someMethodAsync()));
    }
    await tasks.WhenAll(tasks);
    return tasks.Select(x=>x.Result);
}

与IAsyncEnumerable

public async IAsyncEnumerable<int> RunIAsyncJobsAsync()
{
    foreach (var x in Enumerable.Range(0, 100))
    {
        yield return await Task.Run(async () => await someMethodAsync());
    }
}

IAsyncEnumerable 是否有任何性能提升，因为我最终希望在所有 async 调用完成之前不进一步使用算法，但我希望它们并行执行，而不是等待所有其中顺序，我想等待最长的完成。

P.S在这种情况下，我需要ConcurrentQueue/locking 吗？

Answer 1

不，IAsyncEnumerable 不适合这种情况。由于在完成所有异步操作之前您不想进一步移动，因此您不会利用 IAsyncEnumerables 主要资产。即：一旦结果可用，就立即流式传输结果。返回Task<IEnumerable<int>>，甚至更好的是Task<int[]>，更简单、更可取。特别是如果您还想同时启动操作，这是IAsyncEnumerable 默认不做的事情，而是quite tricky 来实现。