如何在Mongoose中实现MongoDB文档字段的数值累加更新？

我正在开发一个记账网站，包含Entries和Users两个集合，对应的Schema如下：

用户Schema #

const userSchema = new mongoose.Schema({
  name: {
    type: String,
    require: true, // 注意：这里应该是`required`，拼写错误会导致验证失效
    unique: true,
    trim: true,
    max: 20,
    min: 3,
  },
  fbalance: {
    type: Number,
    default: 0,
  },
  cbalance: {
    type: Number,
    default: 0,
  },
});

销售记录Schema #

const entrySchema = new mongoose.Schema({
  receivername: {
    type: String,
    require: true, // 同样建议改为`required`
    unique: true,
    trim: true,
    max: 20,
    min: 3,
  },
  price: {
    type: Number,
    require: true, // 建议改为`required`
    default: 0,
  },
  product: {
    type: String,
    required:true
  },
});

现在需要实现：发布销售记录时，将记录中的price值累加到对应用户的cbalance字段中，但当前操作只是用price替换cbalance，无法实现累加效果。

解决方案：使用MongoDB的$inc操作符 #

Mongoose支持直接使用MongoDB的更新操作符，其中$inc专门用于对数值字段进行递增/递减操作，正好满足累加需求。

具体实现代码

假设你已经通过Schema创建了User和Entry模型，下面是创建销售记录并累加用户余额的逻辑：

const Entry = mongoose.model('Entry', entrySchema);
const User = mongoose.model('User', userSchema);

async function addSalesEntry(entryData) {
  try {
    // 1. 先保存新的销售记录
    const newEntry = await Entry.create(entryData);

    // 2. 找到对应用户，累加cbalance
    const updateResult = await User.updateOne(
      { name: entryData.receivername }, // 查询条件：匹配用户的name字段
      { $inc: { cbalance: entryData.price } } // $inc会把cbalance的值加上entryData.price
    );

    // 可以检查是否找到并更新了用户
    if (updateResult.matchedCount === 0) {
      console.warn(`未找到名为${entryData.receivername}的用户`);
    }

    return { entry: newEntry, updateResult };
  } catch (err) {
    console.error('操作失败:', err);
    throw err;
  }
}

关键说明 #

• $inc的作用：它会将目标字段的当前值与指定的数值相加，如果字段不存在，会先将字段初始化为0再执行累加操作，完美适配你的余额累加场景。
• 注意Schema中的拼写错误：把所有require: true改为required: true，否则Mongoose的字段必填验证不会生效，可能导致空值存入数据库。
• 原子性：updateOne操作是原子性的，不用担心并发情况下的余额计算错误。

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