CosmosDB 文档客户端 - 如何为查询隐式添加类型？

Question

我在 .NET Core 3.1 中为 Azure Cosmos DB 使用 DocumentClient。我的示例查询是：

var results = await _cosmosClient
.CreateDocumentQuery<Employee>(_documentCollectionUri, FeedOptionsProvider.DefaultOptions<Employee>())
.Where(x => x.IdentityId == identityId)
.AsDocumentQuery()
.GetResponse();

这段代码有很大的问题，因为如果数据库中有另一个文档具有通过条件，但是另一种类型，那么响应将包含该对象。我知道 lambda 表达式仅限于泛型类型的公共属性，但是是否有可能隐式添加检查类型正确性？数据库对象具有Discriminator 之类的属性，包含实体类型，但在 lambda 表达式中不可见。我想要的行为是检查此Discriminator 属性的条件，而对象模型中没有此属性。该怎么做？

Answer 1

您可以使用下面使用QueryDefinition Class 和container.GetItemQueryIterator 方法的方法：

/// <summary>
/// Run a query (using Azure Cosmos DB SQL syntax) against the container
/// </summary>
private async Task QueryItemsAsync()
{
    var sqlQueryText = "SELECT * FROM c WHERE c.Discriminator = 'Value' AND c.IdentityId = 'Id'";

    Console.WriteLine("Running query: {0}\n", sqlQueryText);

    QueryDefinition queryDefinition = new QueryDefinition(sqlQueryText);
    FeedIterator<Family> queryResultSetIterator = this.container.GetItemQueryIterator<Family>(queryDefinition);

    List<Family> families = new List<Family>();

    while (queryResultSetIterator.HasMoreResults)
    {
        FeedResponse<Family> currentResultSet = await queryResultSetIterator.ReadNextAsync();
        foreach (Family family in currentResultSet)
        {
            families.Add(family);
            Console.WriteLine("\tRead {0}\n", family);
        }
    }
}

在这里，您可以直接添加您的查询，该查询将基于Discriminator 字段进行过滤。

更多实现细节，可以访问：Tutorial: Build a .NET console app to manage data in Azure Cosmos DB SQL API account