### 令牌桶 令牌桶算法(Token Bucket)和 Leaky Bucket 效果一樣但方向相反的算法,更加容易理解.隨著時間流逝,系統會按恒定1/QPS時間間隔(如果QPS=100,則間隔是10ms)往桶里加入Token(想象和漏洞漏水相反,有個水龍頭在不斷的加水),如果桶已經滿了就不再加了.新請求來臨時,會各自拿走一個Token,如果沒有Token可拿了就阻塞或者拒絕服務. 令牌桶的另外一個好處是可以方便的改變速度. 一旦需要提高速率,則按需提高放入桶中的令牌的速率. 一般會定時(比如100毫秒)往桶中增加一定數量的令牌, 有些變種算法則實時的計算應該增加的令牌的數量. 具體實現可參考php 基于redis使用令牌桶算法實現流量控制 ~~~ <?php class TrafficShaper { private $_config; // redis設定 private $_redis; // redis對象 private $_queue; // 令牌桶 private $_max; // 最大令牌數 /** * 初始化 * @param Array $config redis連接設定 */ public function __construct($config, $queue, $max) { $this->_config = $config; $this->_queue = $queue; $this->_max = $max; $this->_redis = $this->connect(); } /** * 加入令牌 * @param Int $num 加入的令牌數量 * @return Int 加入的數量 */ public function add($num = 0) { // 當前剩余令牌數 $curnum = intval($this->_redis->lSize($this->_queue)); // 最大令牌數 $maxnum = intval($this->_max); // 計算最大可加入的令牌數量，不能超過最大令牌數 $num = $maxnum >= $curnum + $num ? $num : $maxnum - $curnum; // 加入令牌 if ($num > 0) { $token = array_fill(0, $num, 1); $this->_redis->lPush($this->_queue, ...$token); return $num; } return 0; } /** * 獲取令牌 * @return Boolean */ public function get() { return $this->_redis->rPop($this->_queue) ? true : false; } /** * 重設令牌桶，填滿令牌 */ public function reset() { $this->_redis->delete($this->_queue); $this->add($this->_max); } private function connect() { try { $redis = new Redis(); $redis->connect($this->_config['host'], $this->_config['port'], $this->_config['timeout'], $this->_config['reserved'], $this->_config['retry_interval']); if (empty($this->_config['auth'])) { $redis->auth($this->_config['auth']); } $redis->select($this->_config['index']); } catch (\RedisException $e) { throw new Exception($e->getMessage()); return false; } return $redis; } } $config = array( 'host' => 'localhost', 'port' => 6379, 'index' => 0, 'auth' => '', 'timeout' => 1, 'reserved' => NULL, 'retry_interval' => 100, ); // 令牌桶容器 $queue = 'mycontainer'; // 最大令牌數 $max = 5; // 創建TrafficShaper對象 $oTrafficShaper = new TrafficShaper($config, $queue, $max); // 重設令牌桶，填滿令牌 $oTrafficShaper->reset(); // 循環獲取令牌，令牌桶內只有5個令牌，因此最后3次獲取失敗 for ($i = 0; $i < 8; $i++) { var_dump($oTrafficShaper->get()); } // 加入10個令牌，最大令牌為5，因此只能加入5個 $add_num = $oTrafficShaper->add(10); var_dump($add_num); // 循環獲取令牌，令牌桶內只有5個令牌，因此最后1次獲取失敗 for ($i = 0; $i < 6; $i++) { var_dump($oTrafficShaper->get()); } ?> ~~~