arrays - 在 Mongodb 子文档数组中，有没有办法在每个子文档中添加一个新字段

Question

假设我有一个像

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

我需要在每个子文档“状态”中添加一个新键：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工作。

但是您仍然想更新所有文档，那么您应该使用 someprogramming code或 some script。下面的代码使用cursor 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 bulk update的更好方法下面的代码显示了如何使用 mongo bulk 进行更新：

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();