MongoDB嵌套数组查询：按score过滤并定位匹配元素

文档结构 #

存在大量如下结构的MongoDB文档：

{   
    '_id': ObjectID('62f8199dc1e8c0f11820cb91'),
    'name': 'foo',
    'score': 4500,
    'searchable': [
        {
            'title': 'blah',
            'date': 'some_date',
            'search_text': "Blah blah blah ...."
        },
        {
            'title': 'bleep',
            'date': 'some_date',
            'search_text': "Lorem Lorem Lorem ...."
        },
        {
            'title': 'bloop',
            'date': 'some_date',
            'search_text': "Ipsum Ipsum Ipsum ...."
        }
    ]
}

需求 #

• 在searchable数组的search_text字段中检索特定字符串
• 按score的数值范围（最小值到最大值）过滤文档
• 返回匹配的searchable元素所属的文档ID，以及该元素在数组中的索引位置

示例预期结果：

• 检索字符串为“Lorem”、score范围3000-5000时，返回：documentID: '62f8199dc1e8c0f11820cb91', searchable: [1]
• 若有多个匹配元素，返回格式类似：documentID: '62f8199dc1e8c0f11820cb91', searchable: [1, 2, 5]

问题 #

尝试用聚合框架结合$unwind实现，但始终无法得到符合预期的结果，需要可行方案。

通过聚合管道多阶段组合实现，核心是保留数组索引、过滤匹配项后聚合结果：

1. $match：先过滤符合score范围的文档 #

先缩小数据集，提升后续操作效率：

{
    $match: {
        score: { $gte: 3000, $lte: 5000 }
    }
}

2. $addFields：为数组元素添加索引 #

用$map和$indexOfArray给每个searchable元素绑定对应的数组索引：

{
    $addFields: {
        indexedSearchable: {
            $map: {
                input: "$searchable",
                as: "item",
                in: {
                    item: "$$item",
                    index: { $indexOfArray: ["$searchable", "$$item"] }
                }
            }
        }
    }
}

3. $unwind：展开带索引的数组 #

将indexedSearchable数组拆分为单个文档，方便后续过滤：

{
    $unwind: "$indexedSearchable"
}

4. $match：过滤search_text匹配的元素 #

筛选search_text包含目标字符串的项（$regex实现模糊匹配，精确匹配可改用$eq）：

{
    $match: {
        "indexedSearchable.item.search_text": { $regex: "Lorem", $options: "i" } // i表示忽略大小写，可按需调整
    }
}

5. $group：按文档ID聚合匹配的索引 #

将同一文档下的所有匹配索引收集到数组中：

{
    $group: {
        _id: "$_id",
        searchableIndices: { $addToSet: "$indexedSearchable.index" }
    }
}

6. $project：格式化输出结果 #

调整字段名以符合预期格式：

{
    $project: {
        _id: 0,
        documentID: "$_id",
        searchable: "$searchableIndices"
    }
}

完整聚合管道代码 #

db.collection.aggregate([
    {
        $match: {
            score: { $gte: 3000, $lte: 5000 }
        }
    },
    {
        $addFields: {
            indexedSearchable: {
                $map: {
                    input: "$searchable",
                    as: "item",
                    in: {
                        item: "$$item",
                        index: { $indexOfArray: ["$searchable", "$$item"] }
                    }
                }
            }
        }
    },
    {
        $unwind: "$indexedSearchable"
    },
    {
        $match: {
            "indexedSearchable.item.search_text": { $regex: "Lorem", $options: "i" }
        }
    },
    {
        $group: {
            _id: "$_id",
            searchableIndices: { $addToSet: "$indexedSearchable.index" }
        }
    },
    {
        $project: {
            _id: 0,
            documentID: "$_id",
            searchable: "$searchableIndices"
        }
    }
])

结果验证 #

针对示例文档执行上述管道后，返回：

{ "documentID" : ObjectId("62f8199dc1e8c0f11820cb91"), "searchable" : [ 1 ] }

若文档存在多个匹配项，searchable数组会包含所有对应的索引值。

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