MongoDB能否在单个更新语句中推送含动态计算字段的数组元素？

你的需求是在单个更新操作中完成三个动作：匹配指定账户、验证余额充足、更新账户余额并推送包含计算后余额的交易记录。之前的写法无效是因为普通更新模式下，$push的字段值不能使用聚合表达式引用现有字段，只能传入常量。

MongoDB 4.2及以上版本支持聚合管道式更新，可以完美解决这个问题。以下是可行的代码示例：

db.collection.updateOne(
  // 查询条件：匹配账户ID，且余额足够完成取款
  { 
    id: "123", 
    accountBalance: { $gte: 100 } 
  },
  // 使用聚合管道执行更新
  [
    // 第一步：更新账户余额（减去取款金额）
    {
      $set: {
        accountBalance: { $subtract: ["$accountBalance", 100] }
      }
    },
    // 第二步：向交易数组推送新记录，直接引用更新后的余额
    {
      $push: {
        transactions: {
          transactionId: "4e920b00-f46f-431f-ba06-9c65bbf1837c",
          transactionType: "STAKE",
          transactionAmount: -100,
          requestingClient: "test person",
          currentBalance: "$accountBalance"
        }
      }
    }
  ]
)

关键说明： #

1. 聚合管道模式：将更新操作包裹在数组中，就可以使用聚合表达式（如$subtract、直接引用$字段名）来处理字段值。
2. 字段一致性：先通过$set更新accountBalance，再在$push中直接引用这个更新后的值，确保交易记录的currentBalance与账户余额完全一致。
3. 并发安全：单个updateOne操作是原子性的，不会出现并发修改导致的不一致问题。

如果你的MongoDB版本低于4.2，只能通过先查询文档、计算后再更新的方式实现，但这种方式存在并发风险，建议优先升级版本使用管道更新方案。

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