NodeJS 频繁请求服务器限速工具

HumorChen99

已于 2022-10-14 16:25:31 修改

阅读量1.8k

点赞数

CC 4.0 BY-SA版权

分类专栏： NodeJS 文章标签： js nodejs 限速

于 2020-11-21 12:01:15 首次发布

本文链接：https://blog.youkuaiyun.com/HumorChen99/article/details/109894085

NodeJS 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种简单的限速器实现方式，通过记录请求执行时间并计算每秒查询数（QPS）来控制请求频率，确保系统的稳定运行。文章详细展示了限速器的核心代码逻辑及其使用方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

示范

const rate_limit=require('./RateLimitUtil')
//配置
rate_limit.config(5)
//执行
setInterval(function () {
    if(rate_limit.isExecutable()){
        console.log('现在的QPS是'+rate_limit.getQPS()+',允许执行')
        //......你的业务逻辑
    }else{
        console.log('现在的QPS是'+rate_limit.getQPS()+'拒绝执行')
    }
},100)

限速器代码

let conf={
    qps:1000
}
let exeLog=new Array()
let fresh=function(){
    let now=new Date().getTime()
    now=now-now%1000
    let top_limit=now+1000
    let newExeLog=new Array()
    for(let i=0;i<exeLog.length;i++)
    {
        let item=exeLog[i]
        if(!(item>top_limit||item<now))
        {
            newExeLog.push(item)
        }
    }
    exeLog=newExeLog
}
/*
是否可执行
true 没有超速
false 即将超速不允许执行
 */
let isExecutable=function () {
    fresh()
    if (getQPS()<conf.qps){
        exeLog.push(new Date().getTime())
        return true
    }
    return false
}
/*
获取现在的QPS
 */
let getQPS=function(){
    return exeLog.length
}
/*
重置
 */
let reset=function(){
    exeLog=new Array()
}
/*
配置qps
 */
let config=function (qps) {
    conf.qps=qps
}
module.exports={
    config:config,
    isExecutable:isExecutable,
    getQPS:getQPS,
    reset:reset
}