Redis令牌桶限流

最新推荐文章于 2024-05-28 13:31:35 发布

3333333_

最新推荐文章于 2024-05-28 13:31:35 发布

阅读量55

点赞数

分类专栏： Redis PHP 文章标签： redis 缓存 php

本文链接：https://blog.csdn.net/wade1010/article/details/128841468

版权

PHP 同时被 2 个专栏收录

64 篇文章 0 订阅

订阅专栏

Redis

39 篇文章 0 订阅

订阅专栏

常用限流

Nginx

Nginx+lua

kong

dinggo

redis令牌桶

<?php
class TrafficShaper{
    private $_config;
    private $_queque;
    private $_max;
    private $redis;    
    
    public function __construct($config,$queque,$max){
        $this->_config=$config;
        $this->_queque=$queque;
        $this->_max=$max;
        $this->_redis=$this->connect();
    }
    
    public function add($num=0){
        $curnum=intval($this->_redis->lSize($this->_queque));
        $max=intval($this->_max);
        $num=$max>=$curnum+$num?$num:$max-$curnum;
        if($num>0){
            $tokens=array_fill(0,$num,1);
            foreach($tokens as $token){
                $this->_redis->lpush($this->_queque,$token);
            }
            return $num;
        }
        return 0;
    }
    
    public function get(){
        return $this->_redis->rpop($this->_queque);
    }
    
    public function reset(){
       return $this->_redis->del($this->_queque);
    }
    
}