Mongoose：如何实现嵌套子文档数组中数量的递进扣减

嘿，我来帮你搞定这个库存递进扣减的问题！你已经成功定位到目标文档和对应的products数组了，接下来的核心就是用MongoDB的聚合管道来实现从第一个item开始依次扣减的逻辑，咱们一步步拆解。

核心思路 #

要实现从数组首个元素开始递进扣减，我们需要：

1. 精准定位到目标product（通过products._id）
2. 遍历该product下的items数组，跟踪剩余需要扣减的数量
3. 对每个item计算实际扣减量：如果剩余扣减额大于当前item的库存，就把当前item库存清0，剩余扣减额减去当前库存；否则直接从当前item库存中扣除剩余量，剩余扣减额归0
4. 用聚合管道完成原子性的更新操作（避免并发问题）

完整代码实现 #

假设你要购买的数量是purchaseQty，以下是完整的findOneAndUpdate实现：

// 定义要购买的数量，这里以7为例
const purchaseQty = 7;

Shop.findOneAndUpdate(
  { _id: '111', 'products._id': 'productId' },
  [
    // 第一步：遍历products数组，处理目标product的items
    {
      $set: {
        products: {
          $map: {
            input: '$products',
            as: 'product',
            in: {
              $cond: [
                { $eq: ['$$product._id', 'productId'] },
                // 处理目标product的items数组
                {
                  ...'$$product',
                  items: {
                    $reduce: {
                      input: '$$product.items',
                      initialValue: { remaining: purchaseQty, updatedItems: [] },
                      in: {
                        $let: {
                          vars: {
                            currentItem: '$$this',
                            remaining: '$$value.remaining'
                          },
                          in: {
                            remaining: {
                              $max: [0, { $subtract: ['$$remaining', '$$currentItem.quantity'] }]
                            },
                            updatedItems: {
                              $concatArrays: [
                                '$$value.updatedItems',
                                [
                                  {
                                    _id: '$$currentItem._id',
                                    quantity: {
                                      $max: [
                                        0,
                                        { $subtract: ['$$currentItem.quantity', '$$remaining'] }
                                      ]
                                    }
                                  }
                                ]
                              ]
                            }
                          }
                        }
                      }
                    }
                  }
                },
                // 非目标product保持原样
                '$$product'
              ]
            }
          }
        }
      }
    },
    // 第二步：提取更新后的items数组（因为上面的items是包含remaining和updatedItems的对象）
    {
      $set: {
        products: {
          $map: {
            input: '$products',
            as: 'product',
            in: {
              $cond: [
                { $eq: ['$$product._id', 'productId'] },
                {
                  ...'$$product',
                  items: '$$product.items.updatedItems'
                },
                '$$product'
              ]
            }
          }
        }
      }
    }
  ],
  { new: true } // 返回更新后的文档，方便验证结果
)

关键逻辑解释 #

• $map遍历products数组：精准找到需要扣减的目标product，其他product不做改动
• $reduce处理items数组：用remaining变量跟踪剩余扣减量，updatedItems收集更新后的item数据
• $max确保库存不出现负数：不管扣减多少，item的quantity最终不会小于0
• 原子性更新：整个操作是MongoDB的原子操作，不用担心并发购买导致的库存不一致问题

注意事项 #

• 提前校验purchaseQty为正数，避免出现反向增加库存的情况
• 如果purchaseQty大于所有items的库存总和，最终所有item的quantity都会变为0，此时你可以根据业务需求抛出“库存不足”的错误
• 如果需要批量处理多个product，只需要调整匹配条件和$map内的判断逻辑即可

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