更新MongoDB中的子文档数组

时间:2016-08-24 14:15:55

标签: arrays mongodb mongoose updates

我有一组学生,他们有一个名字和一系列电子邮件地址。学生文档看起来像这样:

{
  "_id": {"$oid": "56d06bb6d9f75035956fa7ba"},
  "name": "John Doe",
  "emails": [
    {
      "label": "private",
      "value": "private@johndoe.com"
    },
    {
      "label": "work",
      "value": "work@johndoe.com"
    }
  ]
}

电子邮件子文档中的标签设置为每个文档唯一,因此不能有两个具有相同标签的条目。

我的问题是,在更新学生文档时,我希望实现以下目标:

  • 添加带有新标签的电子邮件应该只是添加一个带有给定标签和值的新子文档到数组
  • 如果添加包含已存在标签的电子邮件,则应将现有值设置为更新数据

例如,使用以下数据进行更新时:

{
  "_id": {"$oid": "56d06bb6d9f75035956fa7ba"},
  "emails": [
    {
      "label": "private",
      "value": "me@johndoe.com"
    },
    {
      "label": "school",
      "value": "school@johndoe.com"
    }
  ]
}

我希望电子邮件数组的结果是:

"emails": [
    {
      "label": "private",
      "value": "me@johndoe.com"
    },
    {
      "label": "work",
      "value": "work@johndoe.com"
    },
    {
      "label": "school",
      "value": "school@johndoe.com"
    }
  ]

我如何在MongoDB中实现这一点(可选择使用mongoose)?这是完全可能还是我必须自己在应用程序代码中检查数组?

3 个答案:

答案 0 :(得分:1)

您可以尝试此更新,但仅对小型数据集有效:

mongo shell:

var data = {
    "_id": ObjectId("56d06bb6d9f75035956fa7ba"),
    "emails": [
        {
          "label": "private",
          "value": "me@johndoe.com"
        },
        {
          "label": "school",
          "value": "school@johndoe.com"
        }
    ]
};

data.emails.forEach(function(email) {
    var emails = db.students.findOne({_id: data._id}).emails,
        query = { "_id": data._id },
        update = {};

    emails.forEach(function(e) {
        if (e.label === email.label) {
            query["emails.label"] = email.label;
            update["$set"] = { "emails.$.value": email.value };
        } else {
           update["$addToSet"] = { "emails": email }; 
        }
        db.students.update(query, update)
    });
});

答案 1 :(得分:1)

建议:重构您的数据以使用"标签"作为实际的字段名称。

通过一种简单的方式,MongoDB可以保证给定电子邮件标签的唯一值 - 通过在电子邮件子文档中将标签本身设置为单独的字段。您的数据需要存在于此结构中:

{
  "_id": ObjectId("56d06bb6d9f75035956fa7ba"),
  "name": "John Doe",
  "emails": {
      "private": "private@johndoe.com",
      "work" : "work@johndoe.com"
  }
}

现在,当您想要更新学生的电子邮件时,您可以执行以下更新:

db.students.update(
    {"_id": ObjectId("56d06bb6d9f75035956fa7ba")},
    {$set: {
        "emails.private" : "me@johndoe.com",
        "emails.school" : "school@johndoe.com"
    }}
);

这会将数据更改为:

{
  "_id": ObjectId("56d06bb6d9f75035956fa7ba"),
  "name": "John Doe",
  "emails": {
      "private": "me@johndoe.com",
      "work" : "work@johndoe.com",
      "school" : "school@johndoe.com"
  }
}

不可否认,这种方法存在一个缺点:您需要更改输入数据的结构,从子文档数组中的电子邮件到单个字段的单个子文档的电子邮件。但优点是您的数据需求可以通过JSON对象的工作方式自动满足。

答案 2 :(得分:0)

在调查发布的不同选项后,我决定采用自己的方法,使用lodash的unionBy()函数在代码中手动执行更新。使用express和mongoose的findById()基本上是这样的:

Student.findById(req.params.id, function(err, student) {

    if(req.body.name) student.name = req.body.name;
    if(req.body.emails && req.body.emails.length > 0) {
        student.emails = _.unionBy(req.body.emails, student.emails, 'label');
    }

    student.save(function(err, result) {
        if(err) return next(err);
        res.status(200).json(result);
    });

});

通过这种方式,我可以获得所有字段的部分更新的完全灵活性。当然,您也可以使用findByIdAndUpdate()或其他选项。

替代方法:

然而,改变模式的方式就像Vince Bowdren所建议的那样,将label作为电子邮件子文档中的单个字段,也是一个可行的选择。最后,这取决于您的个人偏好以及您是否需要对数据进行严格验证。

如果您像我一样使用猫鼬,则必须定义一个单独的模式,如下所示:

var EmailSchema = new mongoose.Schema({
  work: { type: String, validate: validateEmail },
  private: { type: String, validate: validateEmail }
}, {
  strict: false,
  _id: false
});

在架构中,您可以为已经支持的标签定义属性并添加验证。通过设置strict: false选项,您还可以允许用户发布带有自定义标签的电子邮件。但请注意,这些不会被验证。您必须在应用程序中手动应用验证,类似于我在上面的方法中进行合并的方式。