异步写入 IAsyncEnumerable

Question

我正在使用 IAsyncEnumerable 从 API 中返回分页结果，如下所示：

public async IAsyncEnumerable<ApiObject> GetResults(){
  int startIndex = 0;
  PaginatedResponse<ApiObject> response;
  do {
    response = await _client.GetResults(_endpoint, startIndex, _pageSize);
    foreach (var obj in response.Items){
      yield return obj;
    }
    startIndex = response.StartIndex + response.Count;
  } while (response.StartIndex + response.Count < response.Total);
}

我想要的是并行发送多个页面的请求，并将每个页面返回到IAsyncEnumerable。顺序无关紧要。

我正在尝试做的事情可能吗？我可以让多个异步任务写入我的方法返回的同一个IAsyncEnumerable 吗？我能想到的最好的办法是让任务写入一个共同的IList，然后在所有任务完成后，遍历该 IList 并返回所有元素：

public async Task<IAsyncEnumerable<ApiObject>> GetResults(){
  int totalPages = _client.MagicGetTotalPagesMethod();
  IList<ApiObject> results = new List<ApiObject>();
  var tasks = Enumerable.Range(0,totalPages).Select(async x => {
    // my goal is to somehow return from here, rather than aggregating into the IList and returning after all tasks are done
    results.AddRange(await _client.GetResults(_endpoint, x*_pageSize, _pageSize));
  });
  await Task.WhenAll(tasks);
  foreach (var result in results){
    yield return result;
  }
}

这个部分通过一次发出多个请求解决了这个问题，但是我们仍然需要等待所有请求都返回，然后这个GetResults 的消费者才能使用结果。 p>

Answer 1

您可以使用the ability to order a sequence of tasks by completion 使这种方法变得非常简单。

var tasks = Enumerable.Range(0,totalPages).Select(x =>
    client.GetResults(_endpoint, x*_pageSize, _pageSize));

foreach(var task in tasks.Order())
    foreach(var item in await task)
        yield return item;