Mongoose防止保存两张单据和子单据

Mongoose 是一个用于在 Node.js 环境中操作 MongoDB 数据库的对象模型库。它提供了一种直接的、基于模式的解决方案来对 MongoDB 进行建模，并且包含内置类型转换、验证、查询构建、业务逻辑钩子等功能。

基础概念

在 Mongoose 中，可以通过定义模型（Model）和模式（Schema）来防止保存重复的单据和子单据。这通常涉及到设置唯一索引（unique index）和使用中间件（middleware）或钩子（hooks）来实现业务逻辑。

类型

唯一索引：在数据库层面防止插入重复值。
预保存钩子（pre-save hook）：在文档保存到数据库之前执行的函数。
静态方法：在模型上定义的方法，用于执行特定的业务逻辑。

应用场景

防止订单重复创建：确保同一用户在同一时间不能创建两个相同的订单。
防止子单据重复关联：确保子单据不会被重复关联到同一个主单据。

示例代码

假设我们有一个订单模型和一个子单据模型，我们想要防止保存重复的订单和子单据关联。

const mongoose = require('mongoose');
const { Schema } = mongoose;

// 定义订单模式
const orderSchema = new Schema({
  orderId: { type: String, required: true, unique: true }, // 唯一索引防止重复订单
  customerName: String,
  // 其他字段...
});

// 定义子单据模式
const subOrderSchema = new Schema({
  subOrderId: { type: String, required: true },
  orderId: { type: Schema.Types.ObjectId, ref: 'Order', required: true },
  // 其他字段...
});

// 在订单模型上添加预保存钩子
orderSchema.pre('save', async function(next) {
  const Order = mongoose.model('Order');
  // 检查是否存在相同的订单
  const existingOrder = await Order.findOne({ orderId: this.orderId });
  if (existingOrder) {
    return next(new Error('订单已存在'));
  }
  next();
});

// 创建模型
const Order = mongoose.model('Order', orderSchema);
const SubOrder = mongoose.model('SubOrder', subOrderSchema);

module.exports = { Order, SubOrder };