Mongoose 是一个 Node.js 中非常流行的 MongoDB 数据库 ORM 库，它提供了一系列的功能和 API 来帮助我们更加方便地操作 MongoDB 数据库。其中，Middleware 是 Mongoose 中一个非常有用的概念，它可以让我们在执行数据库操作前或之后执行一些自定义的逻辑，从而实现一些高级的功能。在本文中，我们将学习如何在 Mongoose 中使用 Middleware 编写插件，以及一些实用的技巧和注意事项。
Middleware 概述
Middleware 是 Mongoose 中一个非常重要的概念，它可以让我们在执行数据库操作前或之后执行一些自定义的逻辑。例如，我们可以在保存一个文档前对其进行一些额外的验证或处理，或者在删除一个文档后对其关联的其他文档进行一些操作等等。Mongoose 中的 Middleware 分为两种类型：
- 前置 Middleware：在执行数据库操作前执行的逻辑，例如验证、处理等等。
- 后置 Middleware：在执行数据库操作后执行的逻辑，例如清理、通知等等。
Mongoose 中的 Middleware 可以针对不同的操作类型进行注册，例如 save、remove、update 等等，也可以针对不同的模型进行注册。在注册 Middleware 时，我们需要指定 Middleware 的执行顺序和条件，以确保其能够正确地被执行。
编写插件
在 Mongoose 中，我们可以将一些常用的逻辑封装为插件，以便在多个模型中复用。一个 Mongoose 插件实际上就是一个包含了 Middleware 的对象，我们可以将其注册到一个或多个模型中。例如，下面是一个简单的插件，用于在保存一个文档前自动填充其创建时间和修改时间：
const timestampPlugin = function(schema, options) {
schema.add({
createdAt: { type: Date, default: Date.now },
updatedAt: { type: Date, default: Date.now }
});
schema.pre('save', function(next) {
this.updatedAt = Date.now();
if (!this.createdAt) {
this.createdAt = this.updatedAt;
}
next();
});
};
module.exports = timestampPlugin;这个插件包含了一个前置 Middleware，用于在保存一个文档前自动填充其创建时间和修改时间。我们可以将其注册到一个或多个模型中，例如：
const mongoose = require('mongoose');
const timestampPlugin = require('./timestampPlugin');
const UserSchema = new mongoose.Schema({
name: String,
email: String
});
UserSchema.plugin(timestampPlugin);
const User = mongoose.model('User', UserSchema);
const user = new User({ name: 'Alice', email: 'alice@example.com' });
user.save(function(err) {
if (err) {
console.error(err);
} else {
console.log(user.createdAt); // 输出创建时间
console.log(user.updatedAt); // 输出修改时间
}
});在上面的代码中，我们先定义了一个 User 模型，并将 timestampPlugin 插件注册到该模型中。然后，我们创建了一个 User 对象并保存到数据库中，保存成功后可以通过 user.createdAt 和 user.updatedAt 属性获取到创建时间和修改时间。
Middleware 注册顺序和条件
在注册 Middleware 时，我们需要注意它们的执行顺序和条件，以确保其能够正确地被执行。Mongoose 中的 Middleware 执行顺序分为两种：
- 串行执行：每个 Middleware 按照注册的顺序依次执行，前一个 Middleware 执行完后再执行下一个 Middleware。
- 并行执行：所有 Middleware 同时执行，不保证执行顺序。
一般来说，我们需要根据 Middleware 的逻辑和条件来决定它们的执行顺序和方式。例如，对于一个保存操作，我们可能需要先进行一些验证，再进行一些处理，最后才保存到数据库中。这时，我们可以使用串行执行的方式，并按照验证、处理、保存的顺序注册 Middleware。
在 Mongoose 中，我们可以使用 pre 和 post 方法来注册 Middleware。pre 方法用于注册前置 Middleware，post 方法用于注册后置 Middleware。例如，下面是一个简单的例子，用于在保存一个文档前进行一些验证和处理：
const userSchema = new mongoose.Schema({
name: String,
email: String
});
userSchema.pre('save', function(next) {
// 验证和处理逻辑
if (!this.name) {
return next(new Error('Name is required'));
}
if (!this.email) {
this.email = 'default@example.com';
}
next();
});在上面的代码中，我们使用 pre 方法注册了一个前置 Middleware，用于在保存一个文档前进行一些验证和处理。如果验证失败，则调用 next 方法传递一个错误对象，中止保存操作。如果验证成功，则调用 next 方法继续执行下一个 Middleware。
除了根据 Middleware 的逻辑和条件来决定它们的执行顺序和方式外，我们还需要注意一些常见的问题，例如：
- 如果一个 Middleware 中调用了 next 方法但没有传递错误对象，则会继续执行下一个 Middleware 或保存操作。
- 如果一个 Middleware 中调用了 next 方法并传递了一个错误对象，则会中止后续的 Middleware 和保存操作，并将错误对象传递给回调函数。
- 如果在一个前置 Middleware 中中止了操作，则不会执行后续的 Middleware 和保存操作。
- 如果在一个后置 Middleware 中中止了操作，则已经完成的操作不会被回滚。
总结
在本文中，我们学习了在 Mongoose 中使用 Middleware 编写插件的方法和技巧。我们了解了 Middleware 的概念和分类，以及如何编写和注册 Middleware 插件。我们还探讨了 Middleware 的执行顺序和条件，以及注意事项和常见问题。希望本文对你在使用 Mongoose 中编写高级功能和插件有所帮助。
来源：JavaScript中文网 ，转载请注明来源 本文地址：https://www.javascriptcn.com/post/658e33adeb4cecbf2d40711d