php用户访问频率,检查/设置用户访问频率

/**

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

* @param $lime_slot 时间片 单位秒

* @param $count 整数

* @param $user_marked 用户唯一标示，默认为客户端IP

* @return array('status'=>1,'info'=>'')

* @author leeyi */

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

$redis->lpush($cache_key, NOW_TIME);

$ret['status'] = 1;

} else {

$datetime = $redis->lindex($cache_key, -1); // -1 标示列表最后一个元素

if( (NOW_TIME-$datetime)

$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;

}