redis实现限速器的几种方式

清疚 2023-07-21 06:07 39阅读 0赞

redis 实现限速器的几种方式。

**GET + INCR + EXPIRE**

先获取 key 的当前值，如果没有超出限制再执行 INCR 增1，如果 key 不存在，使用 redis 的事务初始化 key 和过期时间。

**伪代码：**

count = redis.GET(key)
    
    if redis return nil {
    
      redis.MULTI
    
        redis.INCR(key)
    
        redis.EXPIRE(key, expire_time)
    
      redis.EXEC
    
      count = 1
    
    }
    
    if count > limit {
    
      return 超出限制
    
    } else {
    
      redis.INCR(key)
    
    }

高并发下的问题：

如果同时10个并发程序执行 GET 返回了 nil, 那么这10个并发程序都会执行 redis 的事务将 key 增一，但每个程序的 count 值都为1，如果 limit 设置的值小于10，那么真正执行的程序就超过限制了。如果执行完事务后再查一次 redis 赋值给 count，那么每个程序可能都会返回10，从而没有程序能够继续执行。

key 已经存在的情况下，先 GET 后 INCR 的逻辑也可能会出现实际执行的程序数多于 limit 的情况。

**INCR + EXPIRE**

先 INCR, 如果值为1说明是 key 刚设置的，此时再执行 EXPIRE

伪代码：

count = redis.INCR(key)
    
    if count == 1 {
    
      redis.EXPIRE(key, expire_time)
    
    }
    
    if count > limit {
    
      return 超出限制
    
    }

慎用

如果 INCR 之后程序挂掉了，没有执行 EXPIRE, 那么这个 key 就没有过期时间了，具体的影响视需求而定。

lua脚本

local current
    
    current = redis.call("incr",KEYS[1])
    
    if tonumber(current) == 1 then
    
        redis.call("expire",KEYS[1],1)
    
    end

更多学习内容请访问：

[腾讯T3-T4标准精品PHP架构师教程目录大全，只要你看完保证薪资上升一个台阶（持续更新）][T3-T4_PHP]

**以上内容希望帮助到大家，**很多PHPer在进阶的时候总会遇到一些问题和瓶颈，业务代码写多了没有方向感，不知道该从那里入手去提升，对此我整理了一些资料，包括但不限于：分布式架构、高可扩展、高性能、高并发、服务器性能调优、TP6，laravel，YII2，Redis，Swoole、Swoft、Kafka、Mysql优化、shell脚本、Docker、微服务、Nginx等多个知识点高级进阶干货需要的可以免费分享给大家**，需要的可以加入**我的官方群[点击此处][Link 1]。

[T3-T4_PHP]: https://zhuanlan.zhihu.com/p/102278033
[Link 1]: https://link.zhihu.com/?target=https%3A//jq.qq.com/%3F_wv%3D1027%26k%3D5kN6y8c