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

本文介绍了如何使用express-rate-limit中间件限制API的访问次数,防止恶意请求。在Node.js应用中,通过设置窗口时间和最大访问次数,可以有效地控制接口的调用频率。当超过限制时,将返回自定义的错误信息。此外,还探讨了如何修改中间件源码,捕获请求的IP地址并将其存入数据库,创建一个黑名单系统。在实际操作中,需要对请求的IP进行正则提取,确保格式正确。
摘要由CSDN通过智能技术生成

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

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]

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值