如何将CosmosDB FeedIterator结果转换为IAsyncEnumerable或IEnumerable？

问题描述

要以简单的可重用方式将CosmosDb FeedIterator中的数据转换为IEnumerable，而不必到处编写循环和迭代，这有点棘手。

official Microsoft example不如我喜欢的整齐：

using (FeedIterator setIterator = container.GetItemLinqQueryable<Book>()
                     .Where(b => b.Title == "War and Peace")
                     .ToFeedIterator())
{
    //Asynchronous query execution
    while (setIterator.HasMoreResults)
    {
        foreach(var item in await setIterator.ReadNextAsync())
        {
            Console.WriteLine(item.Price);
        }
    }
}

我想要一个可重复使用的线条，但找不到，于是我写了一个。

尤其是在.NET API中使用FeedIterator时，因此我编写了一个扩展方法将FeedIterator转换为IAsyncEnumerable，这样您就可以使用

(在C#8中，您可以从API返回IAsyncEnumerable，但如果您像我一样需要与netStandard2.0兼容，可以将IAsyncEnumerable转换为常规IEnumerable)

推荐答案

我是这样解决的：

扩展方式：

public static class CosmosDbExtensions
{
    /// <summary>
    /// Convert a feed iterator to IAsyncEnumerable
    /// </summary>
    /// <typeparam name="TModel"></typeparam>
    /// <param name="setIterator"></param>
    /// <returns></returns>
    public static async IAsyncEnumerable<TModel> ToAsyncEnumerable<TModel>(this FeedIterator<TModel> setIterator)
    {
        while (setIterator.HasMoreResults)
            foreach (var item in await setIterator.ReadNextAsync())
            {
                yield return item;
            }
    }
}

返回IAsyncEnumerable：

 public virtual IAsyncEnumerable<TModel> Query()
    {
        Container container = GetContainer(); // Your code to get container

        return  container.GetItemLinqQueryable<TModel>().ToFeedIterator().ToAsyncEnumerable();
    }

返回IEnumerable：

public virtual async Task<IEnumerable<TModel>> Query()
{

    Container container = GetContainer(); // Your code to get container

    using (var feedIterator = container.GetItemLinqQueryable<TModel>().ToFeedIterator())
    {
        return await feedIterator.ToAsyncEnumerable().ToListAsync();
    }
}

(有关this thread中System.Linq.Async包的.ToListAsync()的更多信息)

这篇关于如何将CosmosDB FeedIterator结果转换为IAsyncEnumerable或IEnumerable？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！