MongoDB添加数据时自动更新数组中xAverage平均值的方法

嘿，这个问题我之前也碰到过——要原子性地给嵌套数组加元素同时更新平均值，确实不能分开用aggregate和update，不然会有并发问题（比如两个请求同时加数据，平均值计算就错了）。MongoDB 4.2+支持的聚合管道式更新是完美解决这个需求的方案，我给你两种实用的实现方式，看你需求选：

方案1：通用型（支持$addToSet，自动处理重复数据） #

如果你坚持用$addToSet（避免重复数据点），可以用这种方法，它会先判断新数据点是否已经存在，再决定要不要更新平均值：

collection.updateOne(
  { id: req.params.id },
  [
    {
      $set: {
        datasets: {
          $map: {
            input: "$datasets",
            as: "t",
            in: {
              $cond: {
                // 匹配到目标version的datasets元素
                if: { $eq: ["$$t.version", req.params.version] },
                then: {
                  $mergeObjects: [
                    "$$t",
                    {
                      // 添加新数据点（自动去重）
                      data: { $addToSet: ["$$t.data", req.body] },
                      // 计算新的xAverage：如果数据已存在则保持原值，否则计算新平均值
                      xAverage: {
                        $cond: {
                          if: { $in: [req.body, "$$t.data"] },
                          then: "$$t.xAverage",
                          else: {
                            $avg: { $concatArrays: ["$$t.data", [req.body]] }
                          }
                        }
                      }
                    }
                  ]
                },
                // 不匹配的元素保持原样
                else: "$$t"
              }
            }
          }
        }
      }
    }
  ],
  function(err, result) {
    if (err) {
      res.send(err);
    } else {
      res.send(result);
    }
  }
);

方案2：高性能型（适合$push，数据无重复场景） #

如果你的场景里用$push（允许重复数据，或者能保证新数据点一定不重复），可以用数学公式直接计算新平均值，不用遍历整个data数组，性能会好很多，尤其是data数组很大的时候：

新平均值 = (当前平均值 × 当前数据长度 + 新数据点的x值) ÷ (当前数据长度 + 1)

代码实现：

collection.updateOne(
  { id: req.params.id },
  [
    {
      $set: {
        datasets: {
          $map: {
            input: "$datasets",
            as: "t",
            in: {
              $cond: {
                if: { $eq: ["$$t.version", req.params.version] },
                then: {
                  $mergeObjects: [
                    "$$t",
                    {
                      data: { $push: ["$$t.data", req.body] },
                      xAverage: {
                        $divide: [
                          // 分子：旧平均值×旧长度 + 新x值
                          { $add: [
                            { $multiply: ["$$t.xAverage", { $size: "$$t.data" }] },
                            req.body.x
                          ] },
                          // 分母：旧长度+1
                          { $add: [ { $size: "$$t.data" }, 1 ] }
                        ]
                      }
                    }
                  ]
                },
                else: "$$t"
              }
            }
          }
        }
      }
    }
  ],
  function(err, result) {
    if (err) {
      res.send(err);
    } else {
      res.send(result);
    }
  }
);

关键说明 #

• 这两种方法都是原子操作，MongoDB会把更新整个文档的过程作为一个原子步骤，不会出现并发冲突。
• 必须确保你的MongoDB版本是4.2或以上，因为聚合管道式更新是从这个版本才开始支持的。
• 如果用方案1的$addToSet，要注意$in判断是基于整个数据点对象的（比如{x:0,y:1}和{x:0,y:1}才会被认为重复），如果只需要判断x值重复，要把$in的条件改成{ $in: [req.body.x, { $map: { input: "$$t.data", as: "d", in: "$$d.x" } }] }。

内容的提问来源于stack exchange，提问作者guygrinberger