EF 7与Cosmos DB实体关联查询失败求助

核心原因 #

Cosmos DB 是文档型数据库，本身不支持服务器端跨文档 Join，而 EF Core 7 对 Cosmos DB 的 Include 关联查询仅支持嵌入式文档场景，无法将跨文档的 Include 翻译为有效的 Cosmos SQL，所以会抛出无法翻译的错误。

解决方案（无需客户端求值） #

方案1：使用原生 Cosmos SQL 服务器端查询 #

直接编写 Cosmos SQL 在服务器端完成关联查询，避免客户端内存中 Join。

方式A：EF Core FromSqlRaw 实现

var ride = db.Rides
    .FromSqlRaw(@"
        SELECT r, v
        FROM Rides r
        JOIN v IN Rides
        WHERE r.PartitionKey = @partitionKey 
            AND r.RideId = @rideId 
            AND r.IsDeleted = false
            AND v.VehicleId = r.VehicleId
            AND v.PartitionKey = @partitionKey
            AND v.IsDeleted = false
    ", 
    new SqlParameter("@partitionKey", key), 
    new SqlParameter("@rideId", id))
    .Select(result => 
    {
        // 手动关联Vehicle到Ride
        result.Vehicle = db.Vehicles
            .FirstOrDefault(v => v.VehicleId == result.VehicleId && v.PartitionKey == key);
        return result;
    })
    .FirstOrDefault();

方式B：直接调用 Cosmos SDK（更灵活）

绕过 EF Core 的查询限制，直接用 Cosmos SDK 执行原生查询：

var cosmosClient = dbContext.Database.GetCosmosClient();
var container = cosmosClient.GetContainer("Rides");

var query = new QueryDefinition(@"
    SELECT r, v
    FROM Rides r
    JOIN v IN Rides
    WHERE r.PartitionKey = @partitionKey 
        AND r.RideId = @rideId 
        AND r.IsDeleted = false
        AND v.VehicleId = r.VehicleId
        AND v.PartitionKey = @partitionKey
        AND v.IsDeleted = false
")
.WithParameter("@partitionKey", key)
.WithParameter("@rideId", id);

using var iterator = container.GetItemQueryIterator<(Ride Ride, Vehicle Vehicle)>(query, partitionKey: new PartitionKey(key));
var response = await iterator.ReadNextAsync();
var rideWithVehicle = response.FirstOrDefault();

if (rideWithVehicle != null)
{
    rideWithVehicle.Ride.Vehicle = rideWithVehicle.Vehicle;
}

方案2：调整实体设计（嵌入式文档） #

如果业务允许，将 Vehicle 嵌入到 Ride 实体中，这样查询 Ride 时直接就能获取关联的 Vehicle 数据，无需跨文档关联：

public class Ride: AuditEntity
{
    [Key]
    public Guid RideId { get; set; }

    // 移除单独的VehicleId，直接嵌入Vehicle对象
    public Vehicle Vehicle { get; set; }
}

此方案最适合 Cosmos DB 的文档模型，查询性能最优，但需要评估是否符合业务数据更新的需求。

关键注意事项 #

• 确保 Ride 和 Vehicle 的 PartitionKey 完全一致：Cosmos DB 查询是分区内执行的，跨分区关联查询不仅效率极低，也无法在服务器端完成。
• 原生 SQL 查询中必须同时过滤两边的 IsDeleted 状态，避免返回已删除数据。

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