azure - 为什么 TableQuery 不支持我的 TakeLimit？

Question

我想使用简单的 TableQuery 从我的 Azure 表中获取前 n 行。但使用下面的代码，无论我对 Take 的限制如何，都会获取所有行。

我做错了什么？

int entryLimit = 5;

var table = GetFromHelperFunc();

TableQuery<MyEntity> query = new TableQuery<MyEntity>()
    .Where(TableQuery.GenerateFilterCondition("PartitionKey", QueryComparisons.Equal, "MyPK"))
    .Take(entryLimit);

List<FeedEntry> entryList = new List<FeedEntry>();
TableQuerySegment<FeedEntry> currentSegment = null;

while (currentSegment == null || currentSegment.ContinuationToken != null)
{
    currentSegment = table.ExecuteQuerySegmented(query, this.EntryResolver, currentSegment != null ? currentSegment.ContinuationToken : null);
    entryList.AddRange(currentSegment.Results);
}


Trace.WriteLine(entryList.Count) // <-- Why does this exceed my limit?

Answer 1

存储 SDK 上的 Take 方法不像在 LINQ 中那样工作。想象一下你做了这样的事情:

TableQuery<TableEntity> query = new TableQuery<TableEntity>()
                .Where(TableQuery.GenerateFilterCondition("PartitionKey", QueryComparisons.Equal, "temp"))
                .Take(5);
var result = table.ExecuteQuery(query);

当您开始迭代结果时，您最初只会获得 5 个项目。但在下面，如果您继续迭代结果，SDK 将继续查询表(并继续到下一个包含 5 个项目的“页面”)。

如果我的表中有 5000 个项目，此代码将输出所有 5000 个项目(并且在 SDK 下将执行 1000 个请求并每个请求获取 5 个项目):

TableQuery<TableEntity> query = new TableQuery<TableEntity>()
                .Where(TableQuery.GenerateFilterCondition("PartitionKey", QueryComparisons.Equal, "temp"))
                .Take(5);
var result = table.ExecuteQuery(query);
foreach (var item in result)
{
    Trace.WriteLine(item.RowKey);
}

以下代码将在 1 个请求中准确获取 5 个项目并在此停止:

TableQuery<TableEntity> query = new TableQuery<TableEntity>()
                .Where(TableQuery.GenerateFilterCondition("PartitionKey", QueryComparisons.Equal, "temp"))
                .Take(5);
var result = table.ExecuteQuery(query);
int index = 0;
foreach (var item in result)
{
    Console.WriteLine(item.RowKey);
    index++;
    if (index == 5)
        break;
}

实际上，Take() 方法设置页面大小或“占用计数”(TableQuery 上的 TakeCount 属性)。但如果您只需要 5 条记录，您仍然需要按时停止迭代。

在您的示例中，您应该修改 while 循环，使其在达到 TakeCount(通过调用 Take 设置)时停止:

while (entryList.Count < query.TakeCount && (currentSegment == null || currentSegment.ContinuationToken != null))
{
    currentSegment = table.ExecuteQuerySegmented(query, currentSegment != null ? currentSegment.ContinuationToken : null);
    entryList.AddRange(currentSegment.Results);
}