Mongoose 实现 MongoDB 自增 ID-JavaScript中文网-JavaScript教程资源分享门户

在 MongoDB 中，每个文档都有一个唯一的 _id 字段，它可以是任何类型的值，例如字符串、数字、对象等。默认情况下，MongoDB 会自动为每个文档生成一个 ObjectId 类型的 _id 字段，这个字段可以保证在分布式系统中的唯一性。

然而，在某些场景下，我们可能需要使用数字类型的自增 ID，例如在实现分页功能时，需要对数据进行排序。本文将介绍如何使用 Mongoose 实现 MongoDB 中的自增 ID。

安装 Mongoose

首先，我们需要安装 Mongoose，可以使用 npm 进行安装：

npm install mongoose

创建自增 ID Schema

在 Mongoose 中，我们可以通过定义 Schema 来描述数据模型，以下是一个简单的例子：

const mongoose = require('mongoose');

const userSchema = new mongoose.Schema({
  name: String,
  age: Number,
  email: String,
});

上面的代码定义了一个 User Schema，它包含了三个字段：name、age 和 email。现在我们需要为 User Schema 添加一个自增 ID 字段。

Mongoose 提供了一个插件 mongoose-auto-increment，它可以自动为指定的 Schema 添加自增 ID 字段。可以通过 npm 进行安装：

npm install mongoose-auto-increment

安装完成后，我们需要引入该插件，并调用 initialize() 方法进行初始化：

const mongoose = require('mongoose');
const autoIncrement = require('mongoose-auto-increment');

const connection = mongoose.createConnection('mongodb://localhost/my_database');

autoIncrement.initialize(connection);

现在，我们可以使用 autoIncrement.plugin() 方法为指定的 Schema 添加自增 ID 字段：

// javascriptcn.com 代码示例
const mongoose = require('mongoose');
const autoIncrement = require('mongoose-auto-increment');

const connection = mongoose.createConnection('mongodb://localhost/my_database');

autoIncrement.initialize(connection);

const userSchema = new mongoose.Schema({
  name: String,
  age: Number,
  email: String,
});

userSchema.plugin(autoIncrement.plugin, {
  model: 'User',
  field: 'id',
  startAt: 1,
  incrementBy: 1,
});

const User = connection.model('User', userSchema);

module.exports = User;

上面的代码中，我们调用了 userSchema.plugin() 方法，并传入了 autoIncrement.plugin 方法以及一些参数：

model: 指定模型名称，这个名称将用于在 MongoDB 中创建一个新的集合。
field: 指定自增 ID 字段的名称，这个字段将被添加到 Schema 中。
startAt: 指定自增 ID 的起始值，默认为 1。
incrementBy: 指定每次自增的步长，默认为 1。

使用自增 ID

现在，我们可以使用自增 ID 来创建新的文档了：

// javascriptcn.com 代码示例
const User = require('./models/user');

const user = new User({
  name: 'John',
  age: 25,
  email: 'john@example.com',
});

user.save((err, doc) => {
  if (err) {
    console.error(err);
    return;
  }
  console.log(doc);
});

上面的代码中，我们创建了一个新的 User 实例，并调用了 save() 方法将其保存到数据库中。Mongoose 会自动为我们生成一个自增 ID。

总结

本文介绍了如何使用 Mongoose 实现 MongoDB 中的自增 ID。使用自增 ID 可以方便地对数据进行排序，并实现分页功能。在实际开发中，我们可以根据具体需求来选择使用自增 ID 还是 ObjectId，以达到最优的性能和效果。

来源：JavaScript中文网，转载请注明来源本文地址：https://www.javascriptcn.com/post/6586ad6bd2f5e1655d111b3b