在使用 Node.js 开发应用程序时，Mongoose 是一个非常常见且重要的工具，它提供了一个优雅的方式来处理与 MongoDB 数据库的交互。Mongoose 中的中间件是一种非常强大且常用的功能，在数据操作时，它可以帮助你完成很多重要的任务。本文将详细介绍在 Mongoose 中如何使用中间件，包括什么是中间件，中间件的分类和使用方法，以及如何编写和测试自己的中间件。
什么是中间件？
在 Mongoose 中，中间件是针对操作进行执行的一些函数，包括 pre 和 post，它们可以在执行数据库操作之前（pre）或之后（post）执行。
Mongoose 中的中间件功能非常强大，能够实现很多有用的任务，比如执行验证、方便数据处理、统一处理异常等。
中间件分类
在 Mongoose 中，中间件可以分为以下三类：
1. Model 中间件
使用 schema.pre 和 schema.post 注册的中间件是 Model 中间件，可用于执行操作之前和之后的操作。
2. Document 中间件
使用 schema.pre 和 schema.post 注册的中间件是 Document 中间件，可用于为单个文档执行操作之前和之后定义操作。
3. Query 中间件
使用 queryMiddleware() 注册的中间件是 Query 中间件，可用于拦截查询和更新操作，并在执行之前执行操作。
中间件的使用方法
1. pre 和 post
在 Mongoose 中，pre 和 post 分别表示在操作之前和之后执行。比如，在一个 save 操作执行之前我们可以对数据进行验证，如果验证不通过则拒绝执行保存操作。
-- -------------------- ---- -------
------------------ -------------- -
------------------------------- -
-------------------------- --- ----- ----- -- -
------- ------ ----------
------------- - -----
-------
---
-
---- -
-------
-
---在此例子中，我们在保存文档之前使用 bcrypt 进行加密处理。
2. 异步控制
中间件还支持异步控制：
schema.pre('save', async function() {
await doStuff();
});
schema.post('save', async function() {
await doOtherStuff();
});在这个例子中，我们可以使用 await 关键字来等待异步函数执行完成，并确保在执行其他操作之前完全结束。
3. 错误处理
我们还可以通过 next 参数来处理错误，这样我们就可以在执行中间件时抛出任何错误：
-- -------------------- ---- -------
------------------ -------------- -
-- ----------- -- --------------- -
----- --- - --- --------------- -- -----------
----------
-
---- -
-------
-
---4. 范围
中间件还支持执行范围的定义，这样我们就可以仅在指定的条件下执行中间件。比如：
-- -------------------- ---- -------
-------------------- - ------ ---- -- -------------- -
--------------------- --- -----------
-------
---
------------------------------ -------------- -
----------------------------------
- ---------- -------------------- --
- ------ - --------- - ---------- -------------------- - - --
- ------ ---- --
----
--
---5. Headers
还可以通过定义 Model 和 Schema 插件来使用 Mongoose 对头的功能：
-- -------------------- ---- -------
------------------------------ -
----------------------- - --------------------- - -
---------- ------------- ---- -------- -
------ - --------
------ --------
------ --------
------ -------------
-
--
---这个插件定义了一个 transform 函数，它将 _id 字段转换为 id，并从其他对象中删除不需要的字段。
编写和测试自己的中间件
编写自己的中间件非常容易，只需要简单地使用 pre 和 post 方法来定义您的中间件。测试您的中间件同样也很简单，您可以使用 Mocha、Chai、Sinon 或 Jest 等流行的测试工具。
以下是一个完整的示例，该示例使用 Mocha 和 Chai 来测试 pre 和 post 中间件是否有效，以确保保存操作被正确拦截：
-- -------------------- ---- -------
----- ------ - -----------------------
----- -------- - --------------------
----- - ------ - - ---------
----- ---------- - --- --------
------ -------
--------- -------
---
---------------------- -------------- -
-- -- ---- ---------- ----
---------------- ---- -------
-------
---
----------------------- ------------- ----- -
-- -- --------- ---- ----- --- ---- --------- ---------
----------------- ---- -------
-------
---
----- ---- - ---------------------- ------------
-------------- ----- ------------ -- -- -
---------- --- --- --- ---- ---------- ------- ------ -- -
----- ---- - --- ------
------ -------------------
--------- ---------
---
--------------- -- -
-- ----- ------ ----------
-------
---
---
---在这个例子中，我们定义了一个 userSchema，并将 pre 和 post 钩子附加到 save 方法上。然后，我们使用 Mocha 和 Chai 来执行测试，并检查中间件是否正确地执行。
结论
在本文中，我们详细介绍了 Mongoose 中的中间件的使用方法，包括中间件的分类和使用方法，以及如何编写和测试自己的中间件。使用 Mongoose 中的中间件，可以极大地简化我们的开发工作，提高应用程序的安全性和效率。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/6731e3420bc820c5823b0229