使用他们的文档聚合填充 id 数组

Question

我正在为 mongodb 中的一些聚合函数而苦苦挣扎。 我想在作者的文档中获取书籍文档，其中只有书籍 ID 作为字符串 ID 数组，如下所示：

作者文档

{
  "_id" : "10",
  "full_name" : "Joi Dark",
  "books" : ["100", "200", "351"],
}

及其他文件（书籍）：

{
    "_id" : "100",
    "title" : "Node.js In Action",
    "ISBN" : "121215151515154",
    "date" : "2015-10-10"
}

所以结果我想要这个：

{
  "_id" : "10",
  "full_name" : "Joi Dark",
  "books" : [
       {
           "_id" : "100",
           "title" : "Node.js In Action",
           "ISBN" : "121215151515154",
           "date" : "2015-10-10"
       },
       {
           "_id" : "200",
           "title" : "Book 2",
           "ISBN" : "1212151454515154",
           "date" : "2015-10-20"
       },
       {
           "_id" : "351",
           "title" : "Book 3",
           "ISBN" : "1212151454515154",
           "date" : "2015-11-20"
       }
  ],
}

Answer 1

使用$lookup，它根据localField 与foreignField 的匹配，从from 中的指定集合中检索数据：

db.authors.aggregate([
  { "$lookup": {
    "from": "$books",
    "foreignField": "_id",
    "localField": "books",
    "as": "books"
  }}
])

as 是在文档中写入包含相关文档的“数组”的位置。如果您指定一个现有属性（例如在此处完成），则该属性将被输出中的新数组内容覆盖。

如果您在 MongoDB 3.4 之前有 MongoDB，那么您可能需要先将 $unwind 的数组 "books" 作为 localField：

db.authors.aggregate([
  { "$unwind": "$books" },
  { "$lookup": {
    "from": "$books",
    "foreignField": "_id",
    "localField": "books",
    "as": "books"
  }}
])

这会为原始文档中的每个数组成员创建一个新文档，因此再次使用$unwind 和$group 来创建原始表单：

db.authors.aggregate([
  { "$unwind": "$books" },
  { "$lookup": {
    "from": "$books",
    "foreignField": "_id",
    "localField": "books",
    "as": "books"
  }},
  { "$unwind": "$books" },
  { "$group": {
    "_id": "$_id",
    "full_name": { "$first" "$full_name" },
    "books": { "$push": "$books" }
  }}
])

如果实际上您的_id 值在ObjectId 类型的外部集合中，但您在localField 中有值是“字符串”版本，那么您需要转换数据以便类型匹配。没有其他办法。

通过shell运行这样的东西来转换：

var ops = [];
db.authors.find().forEach(doc => {
  doc.books = doc.books.map( book => new ObjectId(book.valueOf()) );

  ops.push({
    "updateOne": {
      "filter": { "_id": doc._id },
      "update": {
        "$set": { "books": doc.books }
      }
    }
  });

  if ( ops.length >= 500 ) {
    db.authors.bulkWrite(ops);
    ops = [];
  }
});

if ( ops.length > 0 ) {
  db.authors.bulkWrite(ops);
  ops = [];
}

这会将"books" 数组中的所有值转换为真正的ObjectId 值，这些值可以在$lookup 操作中实际匹配。