MongoDB 数组 - 原子更新或推送元素

我在 MongoDB 中有以下文档。

{
    "_id" : ObjectId("521aff65e4b06121b688fabc"),
    "user" : "abc",
    "servers" : [
        {
            "name" : "server1",
            "cpu" : 4,
            "memory" : 4
        },
        {
            "name" : "server2",
            "cpu" : 6,
            "memory" : 6
        },
        {
            "name" : "server3",
            "cpu" : 8,
            "memory" : 8
        }
    ]
}

根据某些事件，我必须更新现有服务器的 cpu 和内存字段，或者如果阵列中不存在新服务器，则将其添加到阵列中。目前，我分两步执行此操作。首先检查服务器是否已经存在于数组中。如果是，则更新 cpu 和内存字段。否则，将新的子文档推送到数组中。由于应用程序的多线程性质，有时同一服务器会多次添加到数组中。是否有任何原子操作符可以执行以下两个操作（类似于$setOnInsert操作员）：

1. 如果数组中存在元素，则更新其字段。
2. 如果数组中不存在元素，则推送新元素。

注：操作员$addToSet在上述情况下不起作用，因为 cpu 或内存的值可能不同。

我想你可以使用查找并修改()这样做是因为它提供了原子更新。 但你的文档结构不合适。 如果您可以将文档更改为此（即 serverID 成为数组的键）：

{
    "_id" : ObjectId("521aff65e4b06121b688fabc"),
    "user" : "abc",
    "servers" : {
       "server1" : {
            "cpu" : 4,
            "memory" : 4
        },
        "server2" : {
            "cpu" : 6,
            "memory" : 6
        },
        "server3" : {
            "cpu" : 8,
            "memory" : 8
        },
    }
}

然后你可以使用一个原子命令查找并修改()无需使用两个单独的更新即可更新find() and update():

db.collection.findAndModify
(
 {query:
    {"_id" : ObjectId("521aff65e4b06121b688fabc")}, 
  update: 
    {$set: {"servers.server4": {"cpu":5, "memory":5 }}}, 
  new: true}
)

使用此时，如果servers.server4不存在，则将其插入，否则更新。