关于 Mongoose 的 Middleware 详解-JavaScript中文网-JavaScript教程资源分享门户

Mongoose 是一个在 Node.js 环境下操作 MongoDB 数据库的工具，它提供了丰富的 API，可以帮助我们更方便地操作数据库。而 Middleware（中间件）则是 Mongoose 中一个非常重要的概念，它可以让我们在进行数据库操作前或操作后进行一些自定义逻辑的处理，这在实际开发中非常有用。

Middleware 概述

Middleware 可以理解为一个拦截器，它可以在进行数据库操作前或操作后对数据进行一些处理，比如对数据进行验证、对数据进行格式化、在保存数据前对数据进行加密等等。

Mongoose 中的 Middleware 可以分为四种类型：文档中间件、模型中间件、聚合中间件和查询中间件。不同类型的 Middleware 触发的时间点不同，下面将详细介绍每种类型的 Middleware。

文档中间件

文档中间件是针对单个文档的操作的，它可以分为以下几种：

init：在文档实例化时触发。
validate：在文档验证时触发，比如在保存文档前进行数据验证。
save：在文档保存时触发。
remove：在文档删除时触发。

下面是一个示例代码，演示如何在文档保存前对数据进行加密：

// javascriptcn.com 代码示例
const mongoose = require('mongoose');
const bcrypt = require('bcrypt');

const UserSchema = new mongoose.Schema({
  username: String,
  password: String
});

UserSchema.pre('save', function(next) {
  const user = this;

  if (!user.isModified('password')) {
    return next();
  }

  bcrypt.genSalt(10, (err, salt) => {
    if (err) {
      return next(err);
    }

    bcrypt.hash(user.password, salt, (err, hash) => {
      if (err) {
        return next(err);
      }

      user.password = hash;
      next();
    });
  });
});

const User = mongoose.model('User', UserSchema);

上面的代码中，我们在 UserSchema 上注册了 pre('save') Middleware，这个 Middleware 会在保存文档前对密码进行加密。

模型中间件

模型中间件是针对整个模型操作的，它可以分为以下几种：

init：在模型实例化时触发。
validate：在模型验证时触发。
save：在模型保存时触发。
remove：在模型删除时触发。

下面是一个示例代码，演示如何在模型保存前对数据进行验证：

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

const UserSchema = new mongoose.Schema({
  username: String,
  password: String
});

UserSchema.pre('save', function(next) {
  const user = this;

  if (!user.username) {
    return next(new Error('Username is required'));
  }

  next();
});

const User = mongoose.model('User', UserSchema);

上面的代码中，我们在 UserSchema 上注册了 pre('save') Middleware，这个 Middleware 会在保存模型前对用户名进行验证，如果用户名为空，则会返回一个错误。

聚合中间件

聚合中间件是针对聚合操作的，它可以分为以下几种：

init：在聚合实例化时触发。
validate：在聚合验证时触发。
save：在聚合保存时触发。
remove：在聚合删除时触发。

下面是一个示例代码，演示如何在聚合查询前对数据进行处理：

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

const UserSchema = new mongoose.Schema({
  username: String,
  password: String
});

UserSchema.pre('aggregate', function(next) {
  const pipeline = this.pipeline();

  pipeline.unshift({ $match: { deleted: false } });

  next();
});

const User = mongoose.model('User', UserSchema);

上面的代码中，我们在 UserSchema 上注册了 pre('aggregate') Middleware，这个 Middleware 会在进行聚合查询前添加一个 $match 操作，确保查询结果中不会包含已删除的数据。

查询中间件

查询中间件是针对单个查询操作的，它可以分为以下几种：

count：在查询数量时触发。
deleteOne：在删除单个文档时触发。
deleteMany：在删除多个文档时触发。
find：在查询文档时触发。
findOne：在查询单个文档时触发。
findOneAndDelete：在查询并删除单个文档时触发。
findOneAndRemove：在查询并删除单个文档时触发。
findOneAndUpdate：在查询并更新单个文档时触发。
updateOne：在更新单个文档时触发。
updateMany：在更新多个文档时触发。

下面是一个示例代码，演示如何在查询单个文档时对数据进行处理：

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

const UserSchema = new mongoose.Schema({
  username: String,
  password: String
});

UserSchema.pre('findOne', function(next) {
  this.where({ deleted: false });
  next();
});

const User = mongoose.model('User', UserSchema);

上面的代码中，我们在 UserSchema 上注册了 pre('findOne') Middleware，这个 Middleware 会在查询单个文档时添加一个查询条件，确保查询结果中不会包含已删除的数据。

总结

Middleware 是 Mongoose 中一个非常重要的概念，它可以让我们在进行数据库操作前或操作后进行一些自定义逻辑的处理。本文详细介绍了 Mongoose 中四种类型的 Middleware，并给出了相应的示例代码。

在实际开发中，我们可以根据具体的业务需求来编写自己的 Middleware，以实现更加灵活和高效的数据处理。

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