当前位置 : 主页 > 网络编程 > PHP >

检查/设置用户访问频率 指定 user_marked在 time_slot 秒内最多访问 count次;

来源:互联网 收集:自由互联 发布时间:2021-06-30
检查/设置用户访问频率指定user_marked在time_slot秒内最多访问count次; 1. [代码] [PHP]代码 /** * 检查/设置用户访问频率 指定 user_marked在 time_slot 秒内最多访问 count次; * * @param $lime_slot 时间
检查/设置用户访问频率 指定 user_marked在 time_slot 秒内最多访问 count次;

1. [代码][PHP]代码    

/**
 * 检查/设置用户访问频率 指定 user_marked在 time_slot 秒内最多访问 count次;
 *
 * @param $lime_slot    时间片 单位秒
 * @param $count        整数
 * @param $user_marked  用户唯一标示,默认为客户端IP
 *
 * @return array('status'=>1,'info'=>'') 
 * @author leeyi <leeyisoft@qq.com>
 */
function check_rate_limiting($time_slot, $count, $user_marked = '') {
    $user_marked = empty($user_marked) ? get_client_ip() : $user_marked;
    $cache_key   = 'rate.limiting:'.$user_marked;
    $redis       = new \Redis;
    $options = array (
        'host'    => C('REDIS_HOST') ? C('REDIS_HOST') : '127.0.0.1',
        'port'    => C('REDIS_PORT') ? C('REDIS_PORT') : 6379,
        'timeout' => C('DATA_CACHE_TIMEOUT') ? C('DATA_CACHE_TIMEOUT') : false,
    );
    $res = $redis->connect($options['host'], $options['port'], $options['timeout']);
    $ret = array('status'=>0, 'info'=>'');
    if (false===$res) {
        $ret['info'] = '链接Redis失败';
        return $ret;
    }
    $redis->expire($cache_key, $time_slot); // 设置过期时间
    $list_len = $redis->llen($cache_key);
    if ($list_len<$count) {
        $redis->lpush($cache_key, NOW_TIME);
        $ret['status'] = 1;
    } else {
        $datetime = $redis->lindex($cache_key, -1); // -1 标示列表最后一个元素
        if ((NOW_TIME-$datetime)<$time_slot) {
            $ret['info'] = '访问频率超过了限制,请稍后重试。';
            // $redis->ltrim($cache_key, -1, 0); //清空列表
        } else {
            $redis->lpush($cache_key, NOW_TIME);
            // 
            $redis->ltrim($cache_key, 0, $count-1);
            $ret['status'] = 1;
        }
    }
    1==$ret['status'] && $ret['info'] = $list_len+1;
    return $ret;
}
网友评论