如何优化CosmosDB中RU消耗过高的Count查询？

针对你提供的查询：

SELECT count(1)
FROM c
where c.creationDateTs > c.dueDateTs
AND c.partitionKey="DocReadTests_assignments"

RU消耗过高的核心原因是：查询条件涉及两个属性的直接比较（creationDateTs > dueDateTs），这种跨字段的逻辑判断无法通过复合索引或默认单字段索引优化——CosmosDB的索引只能加速基于字段本身的等值/范围过滤，无法直接处理字段间的比较运算，因此查询需要遍历分区内所有100多万条记录，导致RU飙升。

以下是具体优化手段：

1. 新增预处理计算字段（最优方案） #

在文档写入或更新时，提前计算并存储一个布尔型字段（比如isCreationAfterDue），其值为creationDateTs > dueDateTs的结果。例如写入文档时设置：

{
  "partitionKey": "DocReadTests_assignments",
  "creationDateTs": 1690000000,
  "dueDateTs": 1680000000,
  "isCreationAfterDue": true
}

修改后的查询变为：

SELECT count(1)
FROM c
WHERE c.partitionKey="DocReadTests_assignments"
AND c.isCreationAfterDue = true

此时默认索引即可覆盖该查询，CosmosDB会直接通过索引过滤出符合条件的文档，无需遍历全分区，RU消耗会大幅下降。

2. 验证索引配置 #

如果使用自定义索引策略，确保isCreationAfterDue字段被包含在索引路径中（默认索引策略已覆盖所有字段，无需额外配置）。

3. 备选方案（无法修改文档结构时） #

若无法新增字段，可使用分区内存储过程执行统计逻辑。存储过程在CosmosDB服务器端运行，避免客户端与服务端的数据传输开销，相比直接查询能降低部分RU，但性能提升效果不如预处理字段方案。

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