更新数组中的对象或使用唯一字段插入数组

Question

我有一个文档，其中包含一个包含对象元素的数组属性。

例如

{
    "_id": "61c01098bc29520008efc00f",
    "name": "codedump",
    "employments": [
         {
              "type": "PERMANENT",
              "salary": 1000000000
         },
         {
              "type": "TEMP",
              "salary": 1000000000
         },
         {
              "type": "PART_TIME",
              "salary": 1000000000
         },
    ]
}

并且在使用type 更新就业工资时，如果该类型已经在就业中，那么我只需要更新salary。否则我需要将新项目推送到数组中。我正在尝试找到一种方法在单个查询中完成这项工作。

1. 如果该类型已经存在于数组中（用于更新）：

/// Update data:
             {
                  "type": "PERMANENT",
                  "salary": 10
             }
// For this case I want the example document to become:
{
    "_id": "61c01098bc29520008efc00f",
    "name": "codedump",
    "employments": [
         {
              "type": "PERMANENT",
              "salary": 10 // From 1000000000 to 10         
         },
         {
              "type": "TEMP",
              "salary": 1000000000
         },
         {
              "type": "PART_TIME",
              "salary": 1000000000
         },
    ]
}

1. 如果类型不存在：

/// Update data:
             {
                  "type": "INVESTMENT",
                  "salary": 10000
             }
// For this case I want the example document to become:
{
    "_id": "61c01098bc29520008efc00f",
    "name": "codedump",
    "employments": [
         {
              "type": "PERMANENT",
              "salary": 1000000000      
         },
         {
              "type": "TEMP",
              "salary": 1000000000
         },
         {
              "type": "PART_TIME",
              "salary": 1000000000
         },
         {
              "type": "INVESTMENT",
              "salary": 10000
         }
    ]
}

Answer 1

我不知道你想要的东西是否可以在一次操作中实现......

话虽如此，我会怎么做，使用$ 位置运算符来更新我想要的数组元素。 Doc & playground.

db.collection.update({
  "employments.type": "PERMANENT",
  
},
{
  "$set": {
    "employments.$.salary": 10
  }
},
{
  "multi": false,
  "upsert": false
})

此操作仅在元素存在于数组中时才有效。否则，操作将失败。然后您可以在代码中捕获异常并使用$push 执行插入。 Playground

db.collection.update({},
{
  "$push": {
    "employments": [
      {
        "type": "INVESTMENT",
        "salary": 10000
      }
    ]
  }
},
{
  "multi": false,
  "upsert": false
})

Answer 2

技术上可以通过 1 次调用，从 Mongo v4.2 开始，您可以使用 pipelined updates，这实质上允许您在更新正文中使用聚合运算符，因此您几乎可以实现任何目标。

我会说，但是它绝对不是很有效，这真的取决于您想要最大化/最小化系统的哪个部分（网络/数据库负载/等）。

你可以这样做：

const input =  {
  "type": "INVESTMENT",
  "salary": 10000
}

db.collection.update({},
[
  {
    $set: {
      employments: {
        $cond: [
          {
            $in: [
              input.type,
              "$employments.type"
            ]
          },
          {
            $map: {
              input: "$employments",
              in: {
                $cond: [
                  {
                    $eq: [
                      "$$this.type",
                      input.type,
                    ]
                  },
                  {
                    type: "$$this.type",
                    salary: input.salary
                  },
                  "$$this"
                ]
              }
            }
          },
          {
            $concatArrays: [
              "$employments",
              [
                input
              ]
            ]
          }
        ]
      }
    }
  }
])

Mongo Playground