Comment implémenter l'algorithme du bucket de jetons en utilisant php

Implémentation php+redis du code de l'algorithme du bucket de jetons :

<?phpnamespace Api\Lib;/**
 * 限流控制
 */class RateLimit{
    private $minNum = 60; //单个用户每分访问数
    private $dayNum = 10000; //单个用户每天总的访问量

    public function minLimit($uid)
    {
        $minNumKey = $uid . &#39;_minNum&#39;;
        $dayNumKey = $uid . &#39;_dayNum&#39;;
        $resMin    = $this->getRedis($minNumKey, $this->minNum, 60);
        $resDay    = $this->getRedis($minNumKey, $this->minNum, 86400);
        if (!$resMin[&#39;status&#39;] || !$resDay[&#39;status&#39;]) {
            exit($resMin[&#39;msg&#39;] . $resDay[&#39;msg&#39;]);
        }
    }

    public function getRedis($key, $initNum, $expire)
    {
        $nowtime  = time();
        $result   = [&#39;status&#39; => true, &#39;msg&#39; => &#39;&#39;];
        $redisObj = $this->di->get(&#39;redis&#39;);
        $redis->watch($key);
        $limitVal = $redis->get($key);
        if ($limitVal) {
            $limitVal = json_decode($limitVal, true);
            $newNum   = min($initNum, ($limitVal[&#39;num&#39;] - 1) + (($initNum / $expire) * ($nowtime - $limitVal[&#39;time&#39;])));
            if ($newNum > 0) {
                $redisVal = json_encode([&#39;num&#39; => $newNum, &#39;time&#39; => time()]);
            } else {
                return [&#39;status&#39; => false, &#39;msg&#39; => &#39;当前时刻令牌消耗完！&#39;];
            }
        } else {
            $redisVal = json_encode([&#39;num&#39; => $initNum, &#39;time&#39; => time()]);
        }
        $redis->multi();
        $redis->set($key, $redisVal);
        $rob_result = $redis->exec();
        if (!$rob_result) {
            $result = [&#39;status&#39; => false, &#39;msg&#39; => &#39;访问频次过多！&#39;];
        }
        return $result;
    }}

Points de code :

1. Définissez d'abord les règles

. Les différentes règles uniques incluent le nombre de visites d'utilisateurs par minute ($minNum), le nombre total de visites par jour par un seul utilisateur ($dayNum), le nombre total de visites sur l'interface, etc.

2. Taux de calcul

Cet exemple de code utilise les secondes comme unité de temps minimale, taux = nombre de visites/heure ($initNum / $expire)

3. time Méthode de calcul du nombre de tokens ajoutés après une visite

Obtenez l'heure de la dernière visite, qui est l'heure à laquelle le token a été déposé pour la dernière fois Calculez le décalage horaire entre l'heure actuelle et la dernière visite multipliée. par le taux pour obtenir le nombre qui doit être ajouté cette fois. Le nombre de jetons après le réapprovisionnement des jetons ne peut pas être supérieur à la valeur minimale du nombre réapprovisionné et à la valeur initialisée. Le numéro prévaudra.

4. Déroulement du programme

Initialisez le nombre de jetons ($minNum) lors de la première visite, stockez-le dans Redis et enregistrez l'horodatage actuel afin de calculer les jetons à compléter. la prochaine fois Nombre de cartes.

Récupérez le nombre de jetons restant lors de la deuxième visite, et ajoutez le nombre de jetons qui doivent être complétés cette fois-ci. Si le nombre de jetons après supplémentation est > 0, la visite en cours est valide et accessible, sinon. Le token n'est plus accessible après utilisation. La raison pour laquelle le jeton est d'abord réapprovisionné, puis jugé si le jeton est >0, est qu'il existe également le concept de taux, c'est-à-dire que si les jetons restants la dernière fois étaient 0 mais que le jeton qui devrait être réapprovisionné cette fois est >1. , alors il est toujours accessible cette fois.

5. Traitement simultané

Utilisez le mécanisme de verrouillage optimiste de Redis.

Pour plus de connaissances connexes, veuillez faire attention au Site Web PHP chinois ! !

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!