MongoDB删除支付项后，如何利用已删除值重新计算totalPaidAmount

问题背景 #

现有MongoDB文档结构如下：

{
  _id: ObjectId('asdu7329n'),
  payments: [
    { _id: ObjectId('28sdf310'), paidAmount: 20 },
    { _id: ObjectId('2fsd410'), paidAmount: 15 },
    { _id: ObjectId('2fs32gd70'), paidAmount: 35 },
  ],
  totalPaidAmount: 70
}

需求是删除指定支付项后，基于现有totalPaidAmount字段用$subtract更新计算新值，避免对大数组（如1000+条）重新求和影响性能。目前已实现用$filter删除支付项的代码，但无法获取被删除支付项的paidAmount值来更新totalPaidAmount：

const query = { _id: ObjectId(saleId) };
const update = [
  { $set: { payments: { $filter: {
    input: '$payments',
    cond: {
      $ne: [ "$$this._id", ObjectId(/* paymentId to delete */) ]
    }
  }}}}
]

await salesSchema.findOneAndUpdate(query, update);

期望实现的逻辑类似：

{
  $set: {
    totalPaidAmount: {
      $subtract: [ '$totalPaidAmount', /* 被删除支付项的paidAmount */ ] 
    }
  }
}

解决方案 #

可以在更新管道中直接从原payments数组筛选出被删除项的金额，无需额外查询或数组求和。以下是两种实现方式：

方案一：分步处理（可读性更高） #

const query = { _id: ObjectId(saleId) };
// 定义要删除的支付项ID
const paymentIdToDelete = ObjectId(/* 替换为实际支付项ID */);

const update = [
  // 第一步：提取要删除的支付项金额到临时字段
  {
    $set: {
      deletedPaymentAmount: {
        $getField: {
          field: "paidAmount",
          input: {
            $arrayElemAt: [
              {
                $filter: {
                  input: "$payments",
                  cond: { $eq: ["$$this._id", paymentIdToDelete] }
                }
              },
              0
            ]
          }
        }
      }
    }
  },
  // 第二步：过滤掉目标支付项
  {
    $set: {
      payments: {
        $filter: {
          input: "$payments",
          cond: { $ne: ["$$this._id", paymentIdToDelete] }
        }
      }
    }
  },
  // 第三步：更新总额并移除临时字段
  {
    $set: {
      totalPaidAmount: { $subtract: ["$totalPaidAmount", "$deletedPaymentAmount"] },
      deletedPaymentAmount: "$$REMOVE" // 删除临时字段，可选
    }
  }
];

await salesSchema.findOneAndUpdate(query, update);

方案二：合并为单个阶段（更简洁） #

const query = { _id: ObjectId(saleId) };
const paymentIdToDelete = ObjectId(/* 替换为实际支付项ID */);

const update = [
  {
    $set: {
      // 过滤支付项数组
      payments: {
        $filter: {
          input: "$payments",
          cond: { $ne: ["$$this._id", paymentIdToDelete] }
        }
      },
      // 直接计算更新后的总额
      totalPaidAmount: {
        $subtract: [
          "$totalPaidAmount",
          {
            $getField: {
              field: "paidAmount",
              input: {
                $arrayElemAt: [
                  {
                    $filter: {
                      input: "$payments",
                      cond: { $eq: ["$$this._id", paymentIdToDelete] }
                    }
                  },
                  0
                ]
              }
            }
          }
        ]
      }
    }
  }
];

await salesSchema.findOneAndUpdate(query, update);

逻辑说明 #

1. 匹配目标支付项：用$filter从原数组中筛选出ID匹配的支付项，再通过$arrayElemAt取出唯一结果（支付项ID唯一）。
2. 提取金额：用$getField获取匹配项的paidAmount值。
3. 更新总额：将原totalPaidAmount减去提取出的金额，得到新的总额。

这种方式无需遍历整个数组求和，在大数组场景下性能更优。

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