Mongo/Mongoose 中的复杂约束

Question

我正在开发一个使用 Mongo/Mongoose/NodeJS 的项目。

我对数据库中的数据有一个相对复杂的约束要求，我不确定是否可以实现。

我的架构如下：

const schema = new Schema(
  {
    id: ObjectId,
    date: {
      type: Date,
      required: true,
      unique: true,
    },
    eventType: {
      type: String,
      enum: ['GROUP_STAGE', 'ROUND_OF_16', 'QUARTER_FINAL', 'SEMI_FINAL', 'FINAL'],
      required: true,
    },
  },
);

需要的逻辑是：

一个 eventType 在每个日历年只能存在一次，除非 eventType 是“GROUP_STAGE”，它可以在一个日历年内存在多次。

我对 Mongo/Mongoose 非常缺乏经验，但我感觉这种限制可能是不可能的。

如果这是唯一的方法，我可以在插入/更新之前将逻辑放在 NodeJS 层中，但这感觉有点笨拙和脆弱。

如果此要求的答案是“无法完成”，那么我仍然会感谢您的确认。

编辑 - 按照已接受答案中的建议解决问题

const schema = new Schema(
  {
    id: ObjectId,
    date: {
      type: Date,
      required: true,
      unique: true,
    },
    eventType: {
      type: String,
      enum: ['GROUP_STAGE', 'ROUND_OF_16', 'QUARTER_FINAL', 'SEMI_FINAL', 'FINAL'],
      required: true,
    },
  },
);

const getFindDuplicateEventTypeInASingleYearQuery = (values) => {
  const { id, eventType, year } = values;
  const query = {
    eventType,
    date: {
      $gte: new Date(`${year}-01-01T00:00:00.000Z`),
      $lt: new Date(`${year}-12-31T23:59:59.999Z`),
    },
  };
  // If an id is supplied, exclude it from the query
  // This is to prevent false flags when updating an object
  if (id) {
    query._id = { $ne: new mongoose.Types.ObjectId(id) };
  }
  return query;
};

const checkForInvalidDuplicateEventTypeInYear = async (queryValues, model) => {
  if (queryValues.eventType !== 'GROUP_STAGE') {
    const query = getFindDuplicateEventTypesInASingleYearQuery(queryValues);
    const count = await model.countDocuments(query);
    if (count > 0) {
      return new Error(`There is already a fixture with the eventType ${queryValues.eventType} in ${queryValues.year}`);
    }
  }
  return null;
};

async function handlePreSave(next) {
  next(
    await checkForInvalidDuplicateEventTypeInYear(
      {
        eventType: this.eventType,
        year: this.date.getFullYear(),
      },
      this.constructor,
    ),
  );
}

async function handlePreFindOneAndUpdate(next) {
  const { eventType, date } = this.getUpdate();
  next(
    await checkForInvalidDuplicateEventTypeInYear(
      {
        id: this.getQuery()._id,
        eventType,
        year: date.split('-')[0],
      },
      this.model,
    ),
  );
}

schema.pre('save', handlePreSave);
schema.pre('findOneAndUpdate', handlePreFindOneAndUpdate);

module.exports = mongoose.model('Fixture', schema);

Answer 1

应用此约束的预保存挂钩会很有帮助。请查看reference

您可以传递约束结果的数据，并根据是否保存文档进行操作。

要在保存前和保存后传递数据，您可以查看this

还有更多关于中间件是如何工作的。 Reference