使用LINQ关联多个MongoDB集合的实现及性能疑问

场景说明 #

现有三个MongoDB集合，结构如下（非MongoDB最佳实践，但需基于此开发）：

// Pets集合
[{
    "_id": ObjectId,
    "Name": string,
    "BodyParts": [{ "GroupId": ObjectId }]
}]

// BodyParts集合
[{
    "_id": ObjectId,
    "PartName": string,
    "Colors": [{ "ColorId": ObjectId }]
}]

// Colors集合
[{
    "_id": ObjectId,
    "ColorName": string
}]

需求：实现GetPet(string petId)方法，传入Pet ID后返回包含完整层级的Pet对象，对应C#类结构：

class Pet
{
    public string Id { get; set; }
    public string Name { get; set; }
    public List<BodyPart> BodyParts { get; set; }
}

class BodyPart
{
    public string Id { get; set; }
    public string PartName { get; set; }
    public List<Color> Colors { get; set; }
}    

class Color
{
    public string Id { get; set; }
    public string ColorName { get; set; }
}

public Pet GetPet(string petId)
{
    // 待实现的LINQ查询逻辑
}

LINQ实现方案（MongoDB 6 + 2.17.1驱动） #

通过MongoDB LINQ的Lookup方法实现多集合关联，底层会将LINQ查询翻译为MongoDB聚合管道，仅发起单次数据库调用。

using MongoDB.Bson;
using MongoDB.Driver;
using System.Linq;

public class PetService
{
    private readonly IMongoCollection<PetDocument> _petsCollection;
    private readonly IMongoCollection<BodyPartDocument> _bodyPartsCollection;
    private readonly IMongoCollection<ColorDocument> _colorsCollection;

    // 构造函数注入集合实例
    public PetService(IMongoDatabase database)
    {
        _petsCollection = database.GetCollection<PetDocument>("Pets");
        _bodyPartsCollection = database.GetCollection<BodyPartDocument>("BodyParts");
        _colorsCollection = database.GetCollection<ColorDocument>("Colors");
    }

    public Pet GetPet(string petId)
    {
        var targetPetId = ObjectId.Parse(petId);

        var result = _petsCollection.AsQueryable()
            // 过滤目标Pet
            .Where(p => p.Id == targetPetId)
            // 第一次关联：Pet关联BodyParts
            .Lookup(
                _bodyPartsCollection.AsQueryable(),
                pet => pet.BodyParts.Select(bp => bp.GroupId),
                bodyPart => bodyPart.Id,
                (pet, matchedBodyParts) => new { Pet = pet, BodyParts = matchedBodyParts }
            )
            // 第二次关联：为每个BodyPart关联对应的Colors
            .SelectMany(
                temp => temp.BodyParts.Select(bp => new 
                { 
                    temp.Pet, 
                    BodyPart = bp,
                    Colors = _colorsCollection.AsQueryable()
                        .Where(c => bp.Colors.Select(co => co.ColorId).Contains(c.Id))
                        .ToList()
                }),
                (_, bpWithColors) => bpWithColors
            )
            // 重组数据为目标Pet结构
            .GroupBy(item => item.Pet)
            .Select(g => new Pet
            {
                Id = g.Key.Id.ToString(),
                Name = g.Key.Name,
                BodyParts = g.Select(item => new BodyPart
                {
                    Id = item.BodyPart.Id.ToString(),
                    PartName = item.BodyPart.PartName,
                    Colors = item.Colors.Select(c => new Color
                    {
                        Id = c.Id.ToString(),
                        ColorName = c.ColorName
                    }).ToList()
                }).ToList()
            })
            .FirstOrDefault();

        return result;
    }

    // 对应MongoDB集合的实体类（映射原始文档结构）
    private class PetDocument
    {
        public ObjectId Id { get; set; }
        public string Name { get; set; }
        public List<BodyPartRef> BodyParts { get; set; }
    }

    private class BodyPartRef
    {
        public ObjectId GroupId { get; set; }
    }

    private class BodyPartDocument
    {
        public ObjectId Id { get; set; }
        public string PartName { get; set; }
        public List<ColorRef> Colors { get; set; }
    }

    private class ColorRef
    {
        public ObjectId ColorId { get; set; }
    }

    private class ColorDocument
    {
        public ObjectId Id { get; set; }
        public string ColorName { get; set; }
    }
}

关键说明 #

1. 底层调用次数：上述LINQ查询会被MongoDB驱动翻译为包含多个$lookup阶段的聚合管道，所有关联逻辑在数据库端执行，客户端仅发起单次数据库请求，比三次单独查询更高效。
2. 性能优化：建议为以下字段创建索引：
   • Pets._id（默认已有）
   • BodyParts._id（默认已有）
   • Colors._id（默认已有）
   • BodyParts.Colors.ColorId
   • Pets.BodyParts.GroupId
3. 类型映射：通过定义内部Document类映射MongoDB原始文档结构，再转换为业务层的Pet/BodyPart/Color类，避免直接修改业务类的序列化规则。

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