nodejs实现ip拦截,并将ip保存到数据库黑名单表(咸鱼版)

一   第一步实现请求的限制,不让一个账号一直访问,一定时间内限定次数

1.安装一下代码包

npm install express-rate-limit

2.它有两种用法,一种全局,一种只对一个接口生效,我这里只用一个生效的api

const createAccountLimiter = rateLimit({
    windowMs: 60 * 60 * 1000,      // 1个小时内
    max: 5,    // 限制最多5次
    message:'已加入黑名单',    //提示消息
    standardHeaders: true,    // Return rate limit info in the `RateLimit-*` headers
    legacyHeaders: true,     // Disable the `X-RateLimit-*` headers
})

3.接下来就要使用了

app.get('/api/login', createAccountLimiter, (req, res) => {

            //这里是你的数据库操作

}

这样就完成了,中间件帮你拦截成功!!!

二  现在需要想办法把ip地址保存到是数据库相当于建立一个黑名单表

这个时候,不得不动他的包的源码了

 在node_moduls下面找到express-rate-limit文件夹,这个是我们刚刚安装的包

打开index.cjs文件

大概在120多行有一个函数

 

就是这个async handler,我们可以拿到request.ip,这个地方就是中间件帮你拦截时候它拿到的包。

接下来我们操作都要在这个函数里面写

      const strings = "insert  into ip_lock values (?,?,?)"
      //查询用户是否已经在黑名单
  
          db.query(strings,[ipadress,'true',ipid],function(err,rows,fields){
              if(err) throw err;
          })
        
      })

这样我们就把这个重复调用我们接口的黑ip存到了数据库,其他的细节你们可以自己搭配使用。

另外提一句,获取的ip,reques.ip需要正则表达式提取一下,不然他的格式会是#f之类的前缀

let ipadress = request.ip.match(/\d+\.\d+\.\d+\.\d+/)[0]

猜你喜欢

转载自blog.csdn.net/qq_43644046/article/details/126076528