栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

PHP-限流-TokenBucket

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

PHP-限流-TokenBucket

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'];
}
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/732589.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号