php和memcached中的简单速率限制器

时间:2011-05-05 18:40:49

标签: php memcached

我正在尝试根据here的评论制作一个简单的速率限制器:

function set_session_rate_limit($memcache, $name, $user_session, $time)
{
   $memcache->add($name . $user_session, 0, $time);

   return $memcache->increment($name . $user_session);
}


set_session_rate_limit($memcache, 'login_fail_', $user_session, 300);

function get_session_rate_limit($memcache, $name, $user_session)
{
   return $memcache->get($name . $user_session);
}


var_dump(get_session_rate_limit($memcache, 'login_fail_', $user_session));

为什么上面的代码返回bool(false)?

2 个答案:

答案 0 :(得分:1)

Read add()方法语法:
bool Memcache::add ( string $key , mixed $var [, int $flag [, int $expire ]] )
你有 $ key = $ name。 $ USER_SESSION
$ var = 0
$ flag = $ time
所以,在$ time之前写null(flag)。
$memcache->add($name . $user_session, 0, null, $time);
我认为方法set在这里会更有用。

答案 1 :(得分:1)

这可能不合适,但我相信速率限制器应该尽可能地裸露。下面我尝试了一个简单的基于会话的限制器。

<?php
    session_start();
    const cap = 3;//Max http requests a host can make
    const period = 5;//the period in which it limits,60 means 1 minuts
    $stamp_init = date("Y-m-d H:i:s");
    if( !isset( $_SESSION['FIRST_REQUEST_TIME'] ) ){
            $_SESSION['FIRST_REQUEST_TIME'] = $stamp_init;
    }
    $first_request_time = $_SESSION['FIRST_REQUEST_TIME'];
    $stamp_expire = date( "Y-m-d H:i:s", strtotime( $first_request_time )+( period ) );
    if( !isset( $_SESSION['REQ_COUNT'] ) ){
            $_SESSION['REQ_COUNT'] = 0;
    }
    $req_count = $_SESSION['REQ_COUNT'];
    $req_count++;
    if( $stamp_init > $stamp_expire ){//Expired
            $req_count = 1;
            $first_request_time = $stamp_init;
    }
    $_SESSION['REQ_COUNT'] = $req_count;
    $_SESSION['FIRST_REQUEST_TIME'] = $first_request_time;
    header('X-RateLimit-Limit: '.cap);
    header('X-RateLimit-Remaining: ' . ( cap-$req_count ) );
    if( $req_count > cap){//Too many requests
            http_response_code( 429 );
            exit();
    }