简介
在现代 Web 应用程序中，流量管理是一个必须掌握的技能。现在很多应用的后端处理已经足够快，然而一些不谨慎设计的请求可能会不经意间使系统的负载量暴增。为了能更好地处理这些请求，我们需要使用限流算法来保障系统的稳定性。而这就是 npm 包 @destinationstransfers/ratelimiter 的出现原因 —— 它提供了轻松实现限制流量的属性。
前置条件
本包所需要的 Node.js 的版本为 >=12.0.0，请确保您已经安装了 Node.js 环境。 如果您没有安装 Node.js，请到官方网站 https://nodejs.org/ 上面下载安装。
安装
在您的终端中输入以下命令即可安装：
npm install @destinationstransfers/ratelimiter
使用
初始化
在使用 @destinationstransfers/ratelimiter 前，您需要初始化一个新的实例：
const RateLimiter = require('@destinationstransfers/ratelimiter')
const limiter = new RateLimiter({
id: 'myApp',
maxReq: 10,
interval: 1000
})这里的 id 属性是用来区分应用程序的，maxReq 代表着最大请求次数，而 interval 表示限流量执行的时间周期。在上述例子中，我们初始化了一个名为 myApp 的限流器，此限流器最大请求次数为 10，限流的时间周期为 1000ms 。
请求
现在，我们可以开始接收请求。在执行请求前，您可以使用 limiter.check() 方法检查当前任务是否可执行。如果可执行，方法返回的是 true，否则返回的是 false：
-- -------------------- ---- -------
----- ------- - ------------------
----- --- - ---------
------------- ---- ----- -- -
-- ------------------ -
-----------------------------------
- ---- -
------
-
--
---------------- -- -- -
---------------------
--在上面的例子中，我们在中间件中使用了 limiter.check() 方法。当请求超出了限流规定的最大值时，将返回 false，接着就会向客户端返回一个带有 503 状态码的响应。
释放
在每个请求之后，我们应该使用 limiter.release() 方法释放请求：
app.use((req, res, next) => {
if (!limiter.check()) {
res.status(503).send('服务繁忙，请稍后再试！')
} else {
limiter.release()
next()
}
})全局未捕获异常处理
为了避免程序因到达请求次数而崩溃，我们需要在程序的未捕获异常中加上处理代码：
process.on('uncaughtException', err => {
if (err.code === 'ECONNREFUSED' || err.code === 'ENETDOWN' || err.code === 'ETIMEDOUT' || err.code === 'ENOTFOUND' || err.code === 'ERESTART' || err.code === 'ESOCKETTIMEDOUT' || err.code === 'ETRUNC') {
return
}
limiter.release()
})示例代码
以下是一个使用 @destinationstransfers/ratelimiter 包的完整的 Express.js 代码范例：
-- -------------------- ---- -------
----- ------- - ------------------
----- ----------- - ---------------------------------------------
----- ------- - --- -------------
--- --------
------- ---
--------- ----
--
----- --- - ---------
------------- ---- ----- -- -
-- ------------------ -
-----------------------------------
- ---- -
-----------------
------
-
--
------------ ----- ---- -- -
--------------
--
--------------- ----- ---- -- -
----------------------
--
------------------------------- --- -- -
-- --------- --- -------------- -- -------- --- ---------- -- -------- --- ----------- -- -------- --- ----------- -- -------- --- ---------- -- -------- --- ----------------- -- -------- --- --------- -
------
-
-----------------
--
---------------- -- -- -
---------------------
--总结
通过使用 @destinationstransfers/ratelimiter 我们可以轻松地完成限流功能的实现。我们只需要在初始化前设定请求的最大次数和限制周期后，便可以使用 check() 和 release() 方法分别检查和释放请求。了解并掌握限量处理技能，可以帮助我们更好地实现对服务器流量的控制，最终保证系统的稳定运行。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/600567be81e8991b448e4023