pecl install redis

use thinkacadeCache;

class TokenBucketRedisLimiter
{
    private $maxTokens;  // 桶的容量
    private $tokensPerSecond;  // 令牌生成速率
    private $lastRefillTime;  // 上次生成令牌時(shí)間
    private $tokens;  // 當(dāng)前桶中令牌數(shù)
    private $redisKey;  // Redis中存儲(chǔ)桶的鍵名
    private $redis;  // Redis連接對(duì)象

    public function __construct($redisKey, $maxTokens, $tokensPerSecond)
    {
        $this->redis = Cache::handler();  // 獲取Redis連接對(duì)象
        $this->redisKey = $redisKey;  // 存儲(chǔ)的鍵名
        $this->maxTokens = $maxTokens;  // 桶的容量
        $this->tokensPerSecond = $tokensPerSecond;  // 令牌生成速率
        $this->lastRefillTime = microtime(true);  // 上次生成令牌時(shí)間
        $this->tokens = 0;  // 當(dāng)前桶中令牌數(shù)
    }

    public function consume()
    {
        $this->refillTokens();

        if ($this->tokens <= 0) {
            return false;  // 沒(méi)有令牌，請(qǐng)求被拒絕
        }

        $this->tokens--;
        $this->redis->set($this->redisKey, $this->tokens);  // 更新Redis中存儲(chǔ)的令牌數(shù)

        return true;  // 請(qǐng)求通過(guò)，獲得了一個(gè)令牌
    }

    private function refillTokens()
    {
        $now = microtime(true);
        $timeDelta = $now - $this->lastRefillTime;  // 上次生成令牌到現(xiàn)在的時(shí)間

        $newTokens = $timeDelta * $this->tokensPerSecond;  // 生成新的令牌數(shù)
        $this->tokens = min($this->tokens + $newTokens, $this->maxTokens);  // 更新令牌數(shù)

        $this->lastRefillTime = $now;  // 更新上次生成令牌時(shí)間

        // 將桶的容量存儲(chǔ)到Redis中
        $this->redis->set($this->redisKey . ':maxTokens', $this->maxTokens);
    }
}