Azure Cosmos Client非ID单条查询的正确实现及相关疑问

问题背景 #

使用Azure Cosmos Client时，通过非ID条件查询单条数据，原始代码用FeedIterator执行Linq查询，判断setIterator.HasMoreResults为false时返回NotFoundResult，但仍出现空引用异常。修改代码后增加了查询结果的null判断，现在有两个疑问：

1. 已知最多返回一条结果，是否可跳过HasMoreResults检查？
2. 无法通过ID访问的情况下，是否有更优的实现方案？

原始代码 #

using var setIterator = container.GetItemLinqQueryable<Baureihe>(requestOptions: RequestOptions)
    .Where(b => b.MarkeSlug == markeSlug.ToLower() && b.Slug == baureiheSlug.ToLower())  
    .ToFeedIterator();  

if (setIterator.HasMoreResults is false)  
{  
    return new NotFoundResult();  
}  

var response = await setIterator.ReadNextAsync();  
var baureihe = response.First();

修改后的代码片段 #

var baureihe = response.First();
if (baureihe == null)  
{  
    return new NotFoundResult();  
}

疑问解答 #

1. 已知最多返回一条结果，是否可跳过HasMoreResults检查？ #

绝对不能跳过，而且HasMoreResults的判断逻辑本身就不适合用来检测是否有数据返回。

HasMoreResults的作用是判断FeedIterator是否还有下一页结果需要读取，它不代表当前页是否有数据。哪怕HasMoreResults为true，调用ReadNextAsync()返回的结果集也可能是空的（比如查询条件匹配不到任何数据）。

原始代码的问题就在于：错误认为HasMoreResults为false就没有数据，但实际场景中，当查询无匹配数据时，HasMoreResults可能为true，执行ReadNextAsync()后返回空结果集，此时调用First()就会抛出空引用异常。

正确的处理逻辑应该是：

• 执行ReadNextAsync()获取结果集
• 检查结果集是否有元素，再安全获取第一条数据

2. 无法通过ID访问的情况下，是否有更优的实现方案？ #

有两种更简洁高效的方案：

方案一：使用Linq扩展方法FirstOrDefaultAsync

借助Microsoft.Azure.Cosmos.Linq包提供的FirstOrDefaultAsync扩展方法，可以直接简化代码，无需手动处理FeedIterator：

var baureihe = await container.GetItemLinqQueryable<Baureihe>(requestOptions: RequestOptions)
    .Where(b => b.MarkeSlug == markeSlug.ToLower() && b.Slug == baureiheSlug.ToLower())
    .FirstOrDefaultAsync();

if (baureihe == null)
{
    return new NotFoundResult();
}

这个方法会自动处理迭代器逻辑，无匹配数据时返回null，避免手动操作的繁琐和异常风险。

方案二：直接使用SQL查询

如果更习惯SQL语法，也可以用GetItemQueryIterator执行SQL查询，同样可以简化逻辑：

var queryText = @"
    SELECT * FROM c 
    WHERE c.MarkeSlug = @markeSlug AND c.Slug = @baureiheSlug
";
var parameters = new QueryDefinition()
    .WithParameter("@markeSlug", markeSlug.ToLower())
    .WithParameter("@baureiheSlug", baureiheSlug.ToLower());

using var iterator = container.GetItemQueryIterator<Baureihe>(queryText, parameters: parameters);
var response = await iterator.ReadNextAsync();
var baureihe = response.FirstOrDefault();

if (baureihe == null)
{
    return new NotFoundResult();
}

关键性能优化提示 #

不管用哪种方案，必须为查询条件中的MarkeSlug和Slug创建复合索引，否则Cosmos会执行全容器扫描，性能极差。可以在容器的索引策略中添加如下配置：

{
  "indexingMode": "consistent",
  "includedPaths": [
    {
      "path": "/MarkeSlug/?",
      "indexes": [
        {
          "kind": "Range",
          "dataType": "String",
          "precision": -1
        }
      ]
    },
    {
      "path": "/Slug/?",
      "indexes": [
        {
          "kind": "Range",
          "dataType": "String",
          "precision": -1
        }
      ]
    }
  ]
}

另外，如果MarkeSlug + Slug的组合是业务唯一的，建议在容器中设置唯一键约束，从底层保证数据唯一性，避免查询返回多条数据的情况。

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