在Mongodb子文档数组中,有任何方法可以在每个子库中添加新字段

Question

假设我有一个类似的文档

{
    "_id" : 5,
    "rows": [
        { "id" : "aab", "value":100},
        { "id" : "aac", "value":400},
        { "id" : "abc", "value":200},
        { "id" : "xyz", "value":300}
    ]
}

我需要在每个子文档"status"中添加一个新键:1,结果应该是这样的

{
    "_id" : 5,
    "rows": [
        { "id" : "aab", "value":100, "status":1},
        { "id" : "aac", "value":400, "status":1},
        { "id" : "abc", "value":200, "status":1},
        { "id" : "xyz", "value":300, "status":1}
    ]
}

我如何通过单个更新查询来完成此操作？

Answer 1

Mongo位置运算符$elemMatch有问题；

$ 运算符可以更新与 $elemMatch() 运算符指定的多个查询条件匹配的第一个数组元素。

因此，在使用 mongo 查询的情况下，您应该仅更新特定的匹配条件。如果您设置rows.aac为匹配，那么您将添加status:1到数组中row.aac，检查查询如下：

db.collectionName.update({
  "_id": 5,
  "rows": {
    "$elemMatch": {
      "id": "abc"
    }
  }
}, {
  $set: {
    "rows.$.status": 1
  }
}, true, false) // here you insert new field so upsert true

mongo 更新显示了如何upsert工作multi。

但您仍然想更新所有文档，那么您应该使用一些programming code或一些script. 下面的代码使用游标 forEach更新所有数据：

db.collectionName.find().forEach(function(data) {
  for (var ii = 0; ii < data.rows.length; ii++) {
    db.collectionName.update({
      "_id": data._id,
      "rows.id": data.rows[ii].id
    }, {
      "$set": {
        "rows.$.status": 1
      }
    }, true, false);
  }
})

如果您的文档大小更大，那么使用mongo 批量更新的更好方法如下代码显示了如何使用 mongo 批量更新：

var bulk = db.collectionName.initializeOrderedBulkOp();
var counter = 0;
db.collectionName.find().forEach(function(data) {
  for (var ii = 0; ii < data.rows.length; ii++) {

    var updatedDocument = {
      "$set": {}
    };

    var setStatus = "rows." + ii + ".status";
    updatedDocument["$set"][setStatus] = 101;
    // queue the update
    bulk.find({
      "_id": data._id
    }).update(updatedDocument);
    counter++;
    //  re-initialize every 1000 update statements
    if (counter % 1000 == 0) {
      bulk.execute();
      bulk = db.collectionName.initializeOrderedBulkOp();
    }
  }

});
// Add the rest in the queue
if (counter % 1000 != 0)
  bulk.execute();