EF查询Cosmos DB时如何配置复合索引匹配谓词查询？

问题背景 #

使用EF查询Cosmos DB时，已配置的复合索引无法生效，但手动编写的相同逻辑SQL查询能正常使用该索引。EF生成的查询包含额外的type条件分支，导致索引不匹配。

EF代码 #

private readonly IRepository<AdministratorResolutionApproved> _repository;
public Administrator_Resolutions_Approved_List_GET(IRepositoryFactory repositoryFactory)
{
    _repository = repositoryFactory.RepositoryOf<AdministratorResolutionApproved>();
}

public async Task GetDataAsync(){
    Expression<Func<AdministratorResolutionApproved, bool>> predicate = item => item.Pk == "key-approved" && item.Ttl == -1;
    var result = await _repository.GetAsync(predicate);
}

EF生成的Cosmos查询 #

{"query":"SELECT VALUE root FROM root WHERE (((root[\"pk\"] = \"key-approved\") AND (root[\"ttl\"] = -1)) AND ((NOT IS_DEFINED(root[\"type\"])) OR (root[\"type\"] = \"AdministratorResolutionApproved\")))"}

当前配置的复合索引 #

"compositeIndexes": [
        [
            {
                "path": "/pk",
                "order": "ascending"
            },
            {
                "path": "/ttl",
                "order": "ascending"
            },
            {
                "path": "/type",
                "order": "ascending"
            }
        ]
    ]

生效的手动SQL查询 #

SELECT * FROM c 
where c.pk = "key-approved"
and c.ttl = -1
and c.type = "AdministratorResolutionApproved"

核心原因 #

EF默认会为实体添加鉴别器过滤，生成的查询包含(NOT IS_DEFINED(root["type"])) OR (root["type"] = "xxx")的分支条件，而你配置的复合索引仅匹配type明确等于指定值的场景，无法覆盖OR分支，导致索引不被使用。

解决方案 #

方案1：修改EF查询，明确指定type条件 #

在谓词中手动添加type的精确匹配，让EF生成的查询和手动SQL结构一致，直接匹配现有复合索引：

Expression<Func<AdministratorResolutionApproved, bool>> predicate = item => 
    item.Pk == "key-approved" && 
    item.Ttl == -1 &&
    item.Type == "AdministratorResolutionApproved"; // 新增精确匹配条件

修改后EF生成的查询会去掉OR分支，变为：

{"query":"SELECT VALUE root FROM root WHERE (((root[\"pk\"] = \"key-approved\") AND (root[\"ttl\"] = -1)) AND (root[\"type\"] = \"AdministratorResolutionApproved\"))"}

此时现有复合索引会正常生效。

方案2：关闭EF的鉴别器自动过滤（单类型集合适用） #

如果你的Cosmos DB集合中仅存储AdministratorResolutionApproved类型的数据，可以在EF模型配置中关闭鉴别器，避免生成额外的type过滤条件：

protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    modelBuilder.Entity<AdministratorResolutionApproved>()
        .HasNoDiscriminator();
}

这种方式不需要修改查询逻辑，现有复合索引即可生效，但仅适用于集合中无其他实体类型的场景。

方案3：调整复合索引适配OR分支（不推荐） #

Cosmos DB支持为OR条件的每个分支配置对应索引，但这种方式会增加索引维护成本，仅在无法修改EF查询或模型时考虑：

"compositeIndexes": [
    // 匹配type明确等于的场景
    [
        {"path": "/pk", "order": "ascending"},
        {"path": "/ttl", "order": "ascending"},
        {"path": "/type", "order": "ascending"}
    ],
    // 匹配type未定义的场景
    [
        {"path": "/pk", "order": "ascending"},
        {"path": "/ttl", "order": "ascending"},
        {"path": "/type", "order": "ascending"}
    ]
]

注意：这种方式需要确保两个分支的索引都能被查询命中，实际效果不如前两种方案直接。

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