在 CosmosDB 中使用 CosmosClient 进行分页

Question

我正在尝试使用 SDK v3 CosmosClient 而不是旧的 DocumentClient。

来实现分页

原因是 DocumentClient 似乎没有很好地翻译包含空间函数的 LINQ 查询（即：使用 Within() 时，我会从 DocumentClient 得到一个错误，说明方法是未实施）。

分页与 DocumentClient.CreateDocumentQuery<T> 配合使用效果很好：

var query = DocumentClient.CreateDocumentQuery<T>(UriFactory.CreateDocumentCollectionUri("master", "features"), feedOptions)
                .Where(t => t.Type == typeof(T).Name)
                .Where(pred)
                .AsDocumentQuery();

string queryContinuationToken = null;
var page = await query.ExecuteNextAsync<T>();
if (query.HasMoreResults)
    queryContinuationToken = page.ResponseContinuation;

我不知道在哪里可以使用 CosmosClient 及其 Container class:

收集延续令牌

QueryRequestOptions options = new QueryRequestOptions();
options.MaxItemCount = maxRecords;

FeedIterator<T> feed;

if (continuationToken == "")
    feed = Container.GetItemLinqQueryable<T>(true, null, options).Where(x => x.Type == typeof(T).Name).Where(pred).ToFeedIterator();
else
    feed = Container.GetItemLinqQueryable<T>(true, continuationToken, options).Where(x => x.Type == typeof(T).Name).Where(pred).ToFeedIterator();

FeedIterator 似乎有一些成员 IDocumentQuery 有（比如 HasMoreResults），但我在任何地方都找不到延续标记。

我错过了什么？

Answer 1

好的，这是我实现的 Where 方法。乍一看似乎可行。

如果您这样做 var f = feed.ReadNextAsync()，您将不会获得 FeedResponse 类型的对象，从而阻止您访问令牌。您需要显式声明 f 类型 FeedResponse<T>

public async Task<(IEnumerable<T> Results, string ContinuationToken)> Where<T>(Expression<Func<T, bool>> pred, int maxRecords = 0, string partitionKey = "", string continuationToken = "") where T : IDocumentModel
{

    QueryRequestOptions options = new QueryRequestOptions();

    if (partitionKey != "")
        options.PartitionKey = new PartitionKey(partitionKey);


    if (maxRecords == 0)
    {
        return (Container.GetItemLinqQueryable<T>(true, null, options).Where(x => x.Type == typeof(T).Name).Where(pred), "");
    }
    else
    {
        options.MaxItemCount = maxRecords;
        string token = "";
        FeedIterator<T> feed;
        List<T> res = new List<T>();

        if (continuationToken == "")
            feed = Container.GetItemLinqQueryable<T>(true, null, options).Where(x => x.Type == typeof(T).Name).Where(pred).ToFeedIterator();
        else
            feed = Container.GetItemLinqQueryable<T>(true, continuationToken, options).Where(x => x.Type == typeof(T).Name).Where(pred).ToFeedIterator();

        Microsoft.Azure.Cosmos.FeedResponse<T> f = await feed.ReadNextAsync();
        token = f.ContinuationToken;

        foreach (var item in f)
        {
            res.Add(item);
        }

        return (res, token);
    }

}

Answer 2

对于 Cosmos SDK 的 3.12.0 版，以下功能按预期工作，几乎可以替代旧的 DocumentQuery。

原始DocumentClient方法对比：

IDocumentQuery<ToDoItem> query = client.CreateDocumentQuery<ToDoItem>(collectionUri)
            .Where(t => t.Description.Contains(searchterm))
            .AsDocumentQuery();

while (query.HasMoreResults)
{
  foreach (ToDoItem result in await query.ExecuteNextAsync())
  {
    log.LogInformation(result.Description);
  }
}

使用 CosmosClient 这变成：

var database = client.GetDatabase("ToDoItems");
var container = database.GetContainer("Items");

var query = container.GetItemLinqQueryable<ToDoItem>()
     .Where(t => t.Description.Contains(searchTerm))
     .ToFeedIterator();

while (query.HasMoreResults)
{
  foreach (ToDoItem result in await query.ReadNextAsync())
  {
    log.LogInformation(result.Description);
  }    
}

所以您的查询现在是 FeedIterator，您可以对其调用 HasMoreResults 和 ReadNextAsync。

诚然，这不会让您访问 FeedIterator 上的诊断、请求费用等，但它会清楚地翻阅结果。

Answer 3

IQueryable<returnVModel> query;
var requestOptions = new QueryRequestOptions
{
 MaxItemCount = 20
};
if (Token == "" || Token == null)
{
      query = Container.GetItemLinqQueryable<returnVModel>(false, null, requestOptions).Where(x => x.id == id);
 }
 else
 {
      query = Container.GetItemLinqQueryable<returnVModel>(false, Token, requestOptions).Where(x => x.id == id);
 }
 var ct = new CancellationTokenSource();
 var totalCount = await query.CountAsync(ct.Token); //Total Count
 var feedIterator = query.ToFeedIterator();
 var queryResults = new List<returnVModel>();
 FeedResponse<returnVModel> feedResults = await feedIterator.ReadNextAsync(ct.Token);
 queryResults.AddRange(feedResults); // Output
 var PaginationToken = feedResults.ContinuationToken //Token

第一次我们需要将令牌作为 null 传递，从下一页开始传递我们在上一个输出中收到的令牌。

分页在 v3 中运行良好。

在 CosmosDB 中使用 CosmosClient 进行分页

Paging with CosmosClient in CosmosDB

c#

azure-cosmosdb

asp.net-core-3.0