如何将MongoDB查询结果转换为userId为键、文档为值的字典

MongoDB 原生普通查询默认返回文档数组，你可以根据使用场景选择「应用层遍历转换」或者「数据库聚合直接生成结构」两种方案实现需求。

方案1：查询后在应用侧组装（最常用，性能开销低） #

拿到查询返回的文档数组后，遍历一次即可组装成目标字典，不同环境代码示例如下：

• MongoDB Shell 环境

const userDict = {}
// 把yourCollection替换成你实际的集合名，find()内可加你的查询过滤条件
db.yourCollection.find().forEach(doc => {
  userDict[doc.userId] = doc
})

// 调用示例：userDict[1234] 即可拿到 { userId: 1234, name: 'Mike' } 对应文档

• Node.js 驱动

async function getUserDict(query = {}) {
  const docs = await db.collection('yourCollection').find(query).toArray()
  return docs.reduce((dict, doc) => {
    dict[doc.userId] = doc
    return dict
  }, {})
}

• Python pymongo 驱动

def get_user_dict(query={}):
    docs = list(db.your_collection.find(query))
    return {doc["userId"]: doc for doc in docs}

注意：如果集合数据量超过万级，不要一次性全量拉取所有文档，建议用游标分批遍历组装，避免内存占用过高。

方案2：聚合管道在数据库侧直接生成字典结构 #

如果不想在应用层做转换，可以用聚合管道直接在数据库端组装好目标结构返回，示例如下：

db.yourCollection.aggregate([
  // 把每个文档转换为k-v结构，k为字符串类型的userId（BSON对象键要求为字符串）
  { $project: { k: { $toString: "$userId" }, v: "$$ROOT" } },
  // 把所有k-v条目聚合为单个数组
  { $group: { _id: null, userEntries: { $push: { k: "$k", v: "$v" } } } },
  // 把k-v数组转换为目标字典结构，移除多余字段
  { $replaceRoot: { newRoot: { $arrayToObject: "$userEntries" } } }
])

该聚合执行后直接返回你需要的字典格式结果，注意返回的键是字符串类型的userId，如果需要数字类型的键，更推荐用方案1在应用层组装。

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