MongoDB：通过聚合函数映射ID数组为对象并保留顺序

问题根源 #

$lookup默认返回的匹配结果是按Tracks集合的存储顺序排列的，和Playlists中tracks数组的原有顺序无关，直接使用会打乱歌曲排列。

解决方案 #

通过原ID数组的索引映射来重新排序关联后的歌曲列表，以下提供两种兼容不同MongoDB版本的实现方式：

方法一：兼容所有MongoDB版本（使用$map+$filter）

通过遍历原tracks数组，逐个从关联结果中匹配对应歌曲，完全保留原顺序：

from pymongo import MongoClient

client = MongoClient("mongodb://localhost:27017/")
db = client["your_db_name"]
playlists_coll = db["Playlists"]

pipeline = [
    # 关联Tracks集合，获取所有匹配的歌曲到临时字段
    {
        "$lookup": {
            "from": "Tracks",
            "localField": "tracks",
            "foreignField": "_id",
            "as": "matched_tracks"
        }
    },
    # 根据原tracks数组顺序重构歌曲列表
    {
        "$addFields": {
            "tracks": {
                "$map": {
                    "input": "$tracks",
                    "as": "track_id",
                    "in": {
                        "$arrayElemAt": [
                            {
                                "$filter": {
                                    "input": "$matched_tracks",
                                    "as": "matched_track",
                                    "cond": {"$eq": ["$$matched_track._id", "$$track_id"]}
                                }
                            },
                            0
                        ]
                    }
                }
            }
        }
    },
    # 移除临时字段（可选）
    {
        "$project": {"matched_tracks": 0}
    }
]

# 执行聚合并获取结果
result = list(playlists_coll.aggregate(pipeline))

方法二：MongoDB 5.2+版本可用（使用$sortArray+$indexOfArray）

利用$sortArray根据歌曲ID在原数组中的索引排序，代码更简洁：

from pymongo import MongoClient

client = MongoClient("mongodb://localhost:27017/")
db = client["your_db_name"]
playlists_coll = db["Playlists"]

pipeline = [
    {
        "$lookup": {
            "from": "Tracks",
            "localField": "tracks",
            "foreignField": "_id",
            "as": "matched_tracks"
        }
    },
    {
        "$addFields": {
            "tracks": {
                "$sortArray": {
                    "input": "$matched_tracks",
                    "sortBy": {"$indexOfArray": ["$tracks", "$_id"]}
                }
            }
        }
    },
    {"$project": {"matched_tracks": 0}}
]

result = list(playlists_coll.aggregate(pipeline))

注意事项 #

• 确保Tracks集合的_id字段与Playlists.tracks中的ID类型完全一致（如均为ObjectId或字符串），避免匹配失败。
• 若需要过滤掉Tracks中不存在的歌曲，可在$map中添加$ifNull或$ne判断处理空值。

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