您现在的位置是：首页 > php

php实现的滑动时间窗口的限流算法

马力 2022-08-16 09:16:47 【php】 4056人已围观

简介php实现的滑动时间窗口的限流算法

/**
	 * 滑动窗口限流
	 * Class SlidingWindow
	 * @package App\Common
	 */
	class SlidingWindow
	{
		protected $timeStamp;                       //当前的时间戳

protected $_is_open_minimum_check = false;  //是否开启最小精度检查 默认不开启

protected $_minimum_time_range_size = 1;    // 最小限制范围精度单位秒钟
		protected $_minimum_time_range_qps  = 50;   // 最小范围精度对应的时间内允许的请求数量

protected $_maximum_time_range_size = 60;   // 最大限制范围单位秒钟
		protected $_maximum_time_range_qps = 3000;  // 最大限制范围对应的时间内允许的请求数量

protected $key = 'sliding_window';

/** @var $redis \Redis */
		public $redis;

public function __construct($redis)
		{
			$this->timeStamp = time();
			$this->redis = $redis;
		}

//设置是否开启最小精度的检查 false为关闭 true为开启
		public function setOpenMininumCheck($open)
		{
			$this->_is_open_minimum_check = $open;
			return $this;
		}

//设置最小精度时间 单位秒
		public function setMininumTime($time)
		{
			$this->_minimum_time_range_size = $time;
			return $this;
		}

//设置最小精度的限制数量
		public function setMinnumQps($qps)
		{
			$this->_minimum_time_range_qps = $qps;
			return $this;
		}

//设置最大精度时间 单位秒
		public function setMaxnumTime($time)
		{
			$this->_maximum_time_range_size = $time;
			return $this;
		}

//设置最大精度的限制数量
		public function setMaxnumQps($qps)
		{
			$this->_maximum_time_range_qps = $qps;
			return $this;
		}
		//设置自定义key
		public function setKey($key = '')
		{
			if(!empty($key)){
				$this->key = $key;
			}
			return $this;
		}

public function luaGrant($value)
		{
			//极限并发情况下的,使用lua脚本会被redis当成一个整体的原子操作
			$luaScript = <<<EOT
							local key = KEYS[1]
							redis.call('expire',key,tonumber(ARGV[2]))
							redis.call('zremrangebyscore',key,0,tonumber(ARGV[4]))
							local max_count = redis.call('zcount',key,tonumber(ARGV[4]),tonumber(ARGV[5]))
							if(max_count < tonumber(ARGV[8])) then
								if (tonumber(ARGV[1]))then
								   local min_count = redis.call('zcount',key,tonumber(ARGV[3]),tonumber(ARGV[5]))
								   if(min_count >= tonumber(ARGV[7])) then
										 return 0
								   else
									 redis.call('zadd',key,tonumber(ARGV[5]),tonumber(ARGV[6]))
									 return 1
								   end
								else
									redis.call('zadd',key,tonumber(ARGV[5]),tonumber(ARGV[6]))
									return 1
								end
							else
							   return 0
							end
	EOT;
			$min_time = $this->timeStamp - $this->_minimum_time_range_size + 1;
			$max_time = $this->timeStamp - $this->_maximum_time_range_size;
			$now_time = $this->timeStamp;
			$res =  $this->redis->eval($luaScript, [
				$this->key,
				$this->_is_open_minimum_check,
				$this->_maximum_time_range_size,
				$min_time,
				$max_time,
				$now_time,
				$value,
				$this->_minimum_time_range_qps,
				$this->_maximum_time_range_qps
			], 1);
	//        return json_encode(["status" => (bool)$res,"time" => date('Y-m-d H:i:s',$this->timeStamp)]);
			return (bool)$res;
		}
	}

使用示例

//uid为 具体的每个用户
	$res = (new SlidingWindow($redis))
		->setKey()
		->setOpenMininumCheck(true)
		->setMininumTime(5)
		->setMinnumQps(500)
		->setMaxnumTime(60)
		->setMaxnumQps(3000)
		->luaGrant($uid);
	if(!$res){
		//限流了
		echo json_encode(["status" => false, "text" => "当前访问人数过多，请刷新重试", "is_limit" => true],JSON_UNESCAPED_UNICODE);
		exit;
	}

使用swoole的压测脚本示例

/**
	 * 压测
	 * Class StressTest
	 */
	class StressTest
	{
		//每秒并发数
		public $second_num = 100;
		//时间秒
		public $second = 60;

public $timer = 0;
		//协程通道
		public $channel;

public function output()
		{
			go(function (){
				$success = 0;
				$error = 0;
				$cw = 0;
				$datas = [];
				while (true) {
					$data = $this->channel->pop();
					if(isset($data['status']) && $data['status'] === false){
						$datas['error'][$data['time']] ++;
						$error++;
					}else if(isset($data['status']) && $data['status'] === true){
						$datas['success'][$data['time']] ++;
						$success++;
					}else{
						$cw++;
					}
					echo "成功的数量为：$success".PHP_EOL;
					echo "失败的数量为：$error".PHP_EOL;
					echo "错误的数量为：$cw".PHP_EOL;
					echo "详细数据:".json_encode($datas).PHP_EOL;
				}
			});
		}

public function __construct($host,$path,$port = 80)
		{
			\Swoole\Runtime::enableCoroutine(SWOOLE_HOOK_ALL);
			$this->channel = new \Swoole\Coroutine\Channel(10);
			self::log(LOG_INFO,"压测开始");
			sleep(1);
			\Swoole\Timer::tick(1000,function ($timer_id) use ($host,$path,$port) {
				$this->timer ++;
				if($this->timer <= $this->second){
					for($i = 0; $i < $this->second_num; $i++) {
						go(function () use ($host,$path,$port) {
							$client = new \Swoole\Coroutine\Http\Client($host,$port);
							$client->set(['timeout' => -1]);
							$client->get($path);
							$res = $client->getBody();
							$client->close();
							$ress = json_decode($res,true);
							$this->channel->push($ress);
							self::log(LOG_INFO,$res);
						});
					}
				}else{
					\Swoole\Timer::clear($timer_id);
				}
			});
			$this->output();
			\Swoole\Event::wait();
		}

/**
		 * 日志输出
		 * @param $level
		 * @param $format
		 */
		public static function log($level, $format) {
			$args = func_get_args();
			$message = [];

$time = microtime(true);
			$message[] = date('Y-m-d H:i:s', $time);
			$message[] = ' ';

if (isset($args[2])) {
				$message[] = vsprintf($format, array_slice($args, 2));
			} else {
				$message[] = $format;
			}

if ($level <= 3) {
				$prefix = "\e[31m";
			}
			else if ($level <= 4) {
				$prefix = "\x1b[33m";
			}
			else if ($level <= 5) {
				$prefix = "\x1b[35m";
			}
			else if ($level <= 6) {
				$prefix = "\x1b[32m";
			}
			else {
				$prefix = "\x1b[37m";
			}
			$message[] = "\x1b[0m";
			echo $prefix.implode('', $message) , PHP_EOL;
		}
	}
	//替换你本地的具体地址
	new StressTest("127.0.0.1","/live_recent.php",8082);

很赞哦！ (0)

上一篇：laravel的上线配置操作

下一篇：lumberjack 实现按时间分割日志