检查/设置用户访问频率

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

/**

 * 检查/设置用户访问频率 指定 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;


}