如何使用Cosmos DB Table API查询指定DateTime范围的数据

问题原因 #

• 报错Method 'Visit' in type 'QueryTokenVisitor' from assembly 'Microsoft.Azure.Cosmos.Table, Version=1.0.8.0' does not have an implementation.是Microsoft.Azure.Cosmos.Table 1.0.8版本的已知Bug，该SDK已停止官方维护，执行分区键范围查询时会触发该异常。
• 分区键不支持范围查询是原生Azure Table Storage的限制，Cosmos DB Table API原生支持分区键范围查询，无需调整数据存储结构。

解决方案 #

方案1：迁移到官方维护的新SDK Azure.Data.Tables（优先推荐） #

这是微软当前迭代维护的Table API专用SDK，兼容性和Bug修复更完善，替换后的代码示例如下：

using Azure.Data.Tables;

public async Task<List<TEntity>> GetEntityWithDateTimePartitionKeyAsync<TEntity>(long startDateTimeTicks, long endDateTimeTicks, string tableName) where TEntity : class, ITableEntity, new()
{
    var results = new List<TEntity>();
    if (endDateTimeTicks <= startDateTimeTicks) return results;
    
    // 连接字符串替换为自身Cosmos DB Table实例的连接字符串
    var tableClient = new TableClient("Cosmos DB Table连接字符串", tableName);
    await tableClient.CreateIfNotExistsAsync();
    
    // 构造范围查询过滤条件
    var filter = TableClient.CreateQueryFilter($"PartitionKey ge {startDateTimeTicks.ToString()} and PartitionKey lt {endDateTimeTicks.ToString()}");
    
    // 分页拉取全量匹配数据
    await foreach (var page in tableClient.QueryAsync<TEntity>(filter).AsPages())
    {
        results.AddRange(page.Values);
    }

    return results;
}

// 实体类调整为实现ITableEntity接口即可
public class TestItem : ITableEntity
{
    public string PartitionKey { get; set; }
    public string RowKey { get; set; }
    public DateTimeOffset? Timestamp { get; set; }
    public ETag ETag { get; set; }
}

// 调用逻辑和原有代码完全兼容
var entityList = await GetEntityWithDateTimePartitionKeyAsync<TestItem>(637671350058030000 , 637671350058036000, "TestTable");

方案2：不更换SDK的临时修复 #

如果暂时无法迁移SDK，将Microsoft.Azure.Cosmos.Table升级到1.0.14及以上的最新稳定版即可，该版本已经修复了分区键范围查询的相关Bug，原有代码无需做任何修改即可正常运行。

注意事项 #

• 写入PartitionKey时的字符串格式需要和查询时参数转换的字符串格式完全一致，避免因字符排序规则不匹配导致查询结果偏差。
• 半小时定时拉取的场景下，可以将上次拉取的结束时间刻度单独存储，每次拉取完成后更新该值，避免重复拉取或漏数据。

内容的提问来源于stack exchange，提问作者user1748546