PHP基于Redis使用令牌桶算法实现限流

本文介绍PHP基于Redis使用令牌桶算法实现限流！

相信大家都知道为什么要做限流，为了保证服务不中断，防止down机，就需要拒绝多余的请求。

例如：
服务器只能抗1000qps，但这时来了很大的流量，导致服务器瞬间压力飙升，使系统超负荷工作。

我们在业务层面进行限流，减轻服务器压力。

令牌桶算法

令牌桶中有初始令牌数，每来一个请求从桶中获取一个令牌，并且在一定时间间隔中可以生成令牌，多余的令牌被丢弃。由此实现限流。

1. 设置一个令牌桶，桶内存放令牌，初始令牌桶内的令牌是满的（桶内令牌的数量根据服务器情况设定）。
2. 每来一个请求从桶中获取一个令牌，当未拿到令牌，则不允许再访问。
3. 使用定时任务，定时投递令牌，最多使桶内令牌满额。

可以使用Redis的队列作为令牌桶使用，使用lPush(入队)，rPop(出队)，实现令牌投递和消费。

TokenBucket.php（令牌桶算法）

/**
 * PHP基于Redis使用令牌桶算法实现限流
 */

class TokenBucket
{
   
    /**
     * 最大令牌数
     * @var int
     */
    private $max;

    /**
     * 令牌桶
     * @var string
     */
    private $queue;

    /**
     * redis 配置
     */
    private $config;

    /**
     * redis 实例
     * @var object
     */
    private $redis;

    /**
     * 初始化
     */
    public function __construct($max, $queue = 'myBucket', $config = [])
    {
   
        $this->max = $max;
        $this->queue = $queue;
        $this->config = $config ? $config : ['host' => '127.0.0.1', 'port' => 6379];
        $this->redis = $this->connect();
    }