redis 限制接口訪問頻率

代碼：

<?php

/**
 *
 */
class myRedis {
    private static $redis = null;

    /**
     * @return null|Redis
     */
    public static function instance() {
        if(self::$redis === null) {
            $redis = new Redis();
            $redis->connect('127.0.0.1', 6379);
            self::$redis = $redis;
        }
        return self::$redis;
    }

    /**
     * @param $api
     * @param string $default
     * @return string
     */
    private static function getApiLimit($api, $default = '3000') {
        $info = [
            'user.info' => '500',
            'user.login' => '1000'
        ];
        return isset($info[$api]) ? $info[$api] : $default;
    }

    /**
     * @return bool
     */
    public static function check() {

        $redis = self::instance();
        $api = "md5_check";
        $time = self::getApiLimit($api);
        $key = "api_limit_" . $api;

        $num = $redis->incr($key);

        if($num == 1) {
            $redis->pExpire($key, $time);
        } else {
            // echo "false";
            return false;
        }
        // echo "true";
        return true;
    }
}

只要在需要限制訪問頻率的接口處加上 myRedis::check() 即可限制該接口訪問平率為 2s 一次

 

不難發現此代碼的邏輯非常簡單：

設置 key 的有效時間為 2s，當 key 過期后執行 $redis->incr($key); 的結果為 1，所以，每次 $redis->incr($key); 結果為 1，則說明距離上一次訪問達到了 2s，重新設置 key 的有效時間並返回 true 即可，其余情況則返回 false。