Mongoose中将_id字段设置为数字类型并实现自动递增序列号

本教程详细阐述如何在Mongoose中自定义_id字段的数据类型为数字，而非默认的ObjectId。我们将通过创建自定义SchemaType来强制_id为正整数，并进一步介绍如何结合Mongoose的生命周期钩子和独立的计数器集合，为数字_id字段实现可靠的自动递增序列号生成机制，确保数据一致性和唯一性。

1. 理解Mongoose的_id字段与默认行为

在Mongoose中，每个文档默认都会有一个名为_id的特殊字段，它作为文档的唯一标识符。Mongoose默认将_id字段的数据类型设置为ObjectId，这是一种12字节的BSON类型，由MongoDB自动生成，能够保证在分布式环境下的唯一性。

当尝试将_id直接设置为Number类型时，例如：

const mySchema = new mongoose.Schema({
  _id: Number
});

这种做法并不能让MongoDB自动生成递增的数字_id。实际上，如果我们在创建文档时不手动指定_id，Mongoose仍然会为其生成一个ObjectId。如果手动指定一个数字，例如_id: 1，则可以正常保存。但是，要实现自动递增的序列号，我们需要更复杂的机制。

腾讯智影-AI数字人

基于AI数字人能力，实现7*24小时AI数字人直播带货，低成本实现直播业务快速增增，全天智能在线直播

73

查看详情

2. 自定义_id为数字类型：NumberId SchemaType

为了确保_id字段严格为数字类型，并且可以施加额外的验证（例如必须是正整数），我们可以创建一个自定义的Mongoose SchemaType。这比简单地将类型设为Number更强大和灵活。

2.1 创建自定义SchemaType

我们将创建一个名为NumberId的自定义SchemaType，它继承自mongoose.SchemaType，并实现其cast方法来验证输入值。

const mongoose = require('mongoose');

/**
 * 自定义SchemaType：NumberId
 * 用于确保_id字段为正整数。
 */
function NumberId(key, options) {
  // 调用父类构造函数，并指定自定义类型名称
  mongoose.SchemaType.call(this, key, options, 'NumberId');
}

// 继承自 mongoose.SchemaType
NumberId.prototype = Object.create(mongoose.SchemaType.prototype);

// 实现 cast() 方法进行类型转换和验证
NumberId.prototype.cast = function (val) {
  if (typeof val !== 'number') {
    throw new Error('NumberId: ' + val + ' 不是一个数字。');
  }
  if (val % 1 !== 0) {
    throw new Error('NumberId: ' + val + ' 不是一个整数。');
  }
  if (val < 0) {
    throw new Error('NumberId: ' + val + ' 是一个负数。');
  }
  return val;
};

// 注册新的SchemaType
mongoose.Schema.Types.NumberId = NumberId;

在上面的代码中：

• 我们定义了一个NumberId函数，它作为我们自定义SchemaType的构造函数。
• 通过Object.create(mongoose.SchemaType.prototype)，NumberId继承了Mongoose SchemaType的所有基本功能。
• cast方法是核心，它负责将输入值转换为期望的类型，并在转换过程中进行验证。如果值不符合要求，它会抛出错误。
• 最后，通过mongoose.Schema.Types.NumberId = NumberId;将这个自定义类型注册到M

以上就是Mongoose中将_id字段设置为数字类型并实现自动递增序列号的详细内容，更多请关注php中文网其它相关文章！