MongoDB：限制关联模型中的记录数量

嘿，很高兴看到你用Node+MongoDB开发婚礼应用来练手，这个宾客餐桌分配的需求很贴合实际场景！关于如何限制每个餐桌关联的宾客数量，我给你整理了几个实用的方案：

1. 应用层校验（最直接易上手） #

这是开发初期最容易实现的方式——在给宾客分配餐桌前，先校验目标餐桌的剩余容量。首先我们需要完善你的Schema，给两个模型加上关联字段：

const guestSchema = new mongoose.Schema({
  firstname: { type: String, trim: true },
  surname: { type: String, trim: true },
  attending: String,
  allergies: String,
  menu: String,
  table: { type: mongoose.Schema.Types.ObjectId, ref: 'Table' } // 关联对应的餐桌
});

const tableSchema = new mongoose.Schema({
  name: String,
  capacity: { type: Number, required: true, min: 1 }, // 补充餐桌容量字段
  guests: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Guest' }] // 存关联的宾客ID，双向关联更方便管理
});

接下来可以在宾客保存的中间件里做校验，或者写专门的分配逻辑：

// 用pre-save中间件自动校验
guestSchema.pre('save', async function(next) {
  if (!this.table) return next(); // 未分配餐桌则跳过校验

  // 先获取目标餐桌信息
  const targetTable = await this.model('Table').findById(this.table);
  if (!targetTable) return next(new Error('关联的餐桌不存在'));

  // 查询该餐桌当前已分配的宾客数量
  const assignedGuestsCount = await this.model('Guest').countDocuments({ table: this.table });
  if (assignedGuestsCount >= targetTable.capacity) {
    return next(new Error(`该餐桌容量为${targetTable.capacity}，已无剩余空位`));
  }

  next();
});

如果是双向关联，也可以直接通过餐桌的guests数组长度判断：

async function assignGuestToTable(guestId, tableId) {
  const table = await Table.findById(tableId);
  if (!table) throw new Error('目标餐桌不存在');
  
  if (table.guests.length >= table.capacity) {
    throw new Error(`该餐桌已满，容量上限为${table.capacity}`);
  }

  // 同时更新宾客和餐桌的关联关系
  await Guest.findByIdAndUpdate(guestId, { table: tableId });
  table.guests.push(guestId);
  await table.save();
}

2. 数据库层校验（更严谨的兜底） #

从MongoDB 3.6开始支持文档级验证，你可以给Table集合设置验证规则，强制限制guests数组的长度不超过capacity：

const tableSchema = new mongoose.Schema({
  name: String,
  capacity: { type: Number, required: true, min: 1 },
  guests: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Guest' }]
}, {
  validateBeforeSave: true,
  // 集合级验证器
  collection: {
    validator: function(doc) {
      return doc.guests.length <= doc.capacity;
    },
    message: props => `宾客数量(${props.value.guests.length})已超过餐桌容量(${props.value.capacity})`
  }
});

不过要注意：这种验证只在直接更新Table文档时生效，如果只是单独更新Guest的关联字段，数据库不会自动触发校验，所以通常需要和应用层逻辑配合使用。

3. 原子操作（解决并发分配问题） #

如果担心多人同时操作时出现“超容”的竞态条件（比如两个请求同时校验同一餐桌，都判断有剩余空位，最后导致超员），可以用MongoDB的原子更新操作，把校验和更新合并成一个原子步骤：

async function assignGuestToTable(guestId, tableId) {
  // 原子更新餐桌：只有当宾客数小于容量时，才添加新宾客ID
  const updatedTable = await Table.findOneAndUpdate(
    { 
      _id: tableId,
      $expr: { $lt: [{ $size: '$guests' }, '$capacity'] } // 用$size获取数组长度，和capacity比较
    },
    { $push: { guests: guestId } },
    { new: true } // 返回更新后的餐桌文档
  );

  if (!updatedTable) {
    throw new Error('该餐桌已满，无法添加新宾客');
  }

  // 更新宾客的关联信息
  await Guest.findByIdAndUpdate(guestId, { table: tableId });
  return updatedTable;
}

这种方式利用MongoDB的原子性保证，避免了并发场景下的超容问题，适合对数据准确性要求较高的场景。

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