Mongoose中的Discriminator机制详解-优快云博客

Mongoose中的Discriminator机制详解

mongoose Automattic/mongoose: Mongoose 是一个流行的Node.js对象数据映射（ODM）库，专为MongoDB设计，能够简化在Node.js中使用MongoDB数据库的操作，提供了丰富的查询构建、模型定义、数据验证等特性。项目地址: https://gitcode.com/gh_mirrors/mo/mongoose

什么是Discriminator

Discriminator是Mongoose提供的一种模式继承机制，它允许开发者在同一个MongoDB集合上创建具有重叠模式的不同模型。这种机制非常适合需要存储多种相似但略有差异的文档类型的场景。

Discriminator的核心概念

基本用法

使用model.discriminator()函数可以创建Discriminator模型。该函数接收三个参数：

模型名称
区分器模式（Discriminator Schema）
可选的键（默认为模型名称）

const eventSchema = new Schema({ timestamp: Date });
const Event = mongoose.model('Event', eventSchema);

const ClickedLinkEvent = Event.discriminator('ClickedLink', 
  new Schema({ url: String })
);

存储机制

所有Discriminator模型创建的文档都会存储在基础模型的集合中。例如，在上面的例子中，ClickedLinkEvent和基础Event模型的文档都会存储在同一个集合中。

区分键（Discriminator Key）

Mongoose通过一个特殊的字段（默认为__t）来区分不同类型的文档。这个字段会自动添加到每个文档中，用于标识文档属于哪个Discriminator模型。

Discriminator的高级用法

更新区分键

默认情况下，Mongoose不允许直接更新区分键。如果需要修改，可以使用以下方法：

await Event.findOneAndUpdate(
  { _id: docId },
  { $set: { __t: 'NewType' } },
  { overwriteDiscriminatorKey: true }
);

嵌套Discriminator

Discriminator不仅可以用于顶级文档，还可以用于嵌套文档：

数组中的Discriminator

const itemSchema = new Schema({ name: String });
const Item = mongoose.model('Item', itemSchema);

const productSchema = new Schema({
  items: [itemSchema]
});

const Product = mongoose.model('Product', productSchema);

const DiscountedItem = Item.discriminator('Discounted', 
  new Schema({ discount: Number })
);

单嵌套Discriminator

const ownerSchema = new Schema({ name: String });
const Owner = mongoose.model('Owner', ownerSchema);

const propertySchema = new Schema({
  owner: ownerSchema
});

const Property = mongoose.model('Property', propertySchema);

const CompanyOwner = Owner.discriminator('Company', 
  new Schema({ registrationNumber: String })
);