PHP-限流-TokenBucket

最新推荐文章于 2024-08-26 18:13:56 发布

ErfengDu

最新推荐文章于 2024-08-26 18:13:56 发布

阅读量1.2k

点赞数 2

分类专栏：随手来一段代码

本文链接：https://blog.csdn.net/du_denglan/article/details/122818317

版权

php redis 缓存

随手来一段代码专栏收录该内容

4 篇文章 0 订阅

订阅专栏

public function checkLimit($cacheKey = '', $initNum = '', $expire = '', $retry = true)
{
    $nowTime = time();
    Redis::watch($cacheKey);
    $redisData = Redis::get($cacheKey);
    $limitData = $redisData ? json_decode($redisData, true) : ['num' => $initNum, 'time' => $nowTime];

    // （单位时间访问频率 / 单位时间）*（当前时间 - 上次访问时间） = 上次请求至今可增加的访问次数
    $addNum = intval(($initNum / $expire) * ($nowTime - $limitData['time']));
    $newNum = min($initNum, (($limitData['num'] - 1) + $addNum));
    Log::debug($cacheKey . '-剩余领牌次数:' . $newNum);
    if ($newNum <= 0) {
        return ['status' => false, 'msg' => '当前时刻令牌用完啦！'];
    }
    $limitData = json_encode(['num' => $newNum, 'time' => $nowTime]);
    Redis::multi();
    Redis::set($cacheKey, $limitData);
    if (!Redis::exec()) {
        if (!$retry) {
            return ['status' => false, 'msg' => '访问频次过多！'];
        } else {
            //执行3次丢弃
            for ($i = 0; $i < 3; $i++) {
                $aRet = self::_limit($cacheKey, $initNum, $expire, false);
                if (Arr::get($aRet, 'status', false)) {
                    break;
                }
                if (Arr::get($aRet, 'msg') == '当前时刻令牌用完啦') {
                    return $aRet;
                }
            }
        }
    }
    return ['status' => true, 'msg' => 'ok'];
}