在没有cas的Memcached中实现原子计数器

时间:2012-10-30 04:32:52

标签: php algorithm transactions memcached

我们有一个网页想要限制uo到100个人可以同时访问,所以我们使用memcached来实现一个全局计数器,例如。

我们正在使用http://www.php.net/manual/en/class.memcache.php所以没有cas,当前代码就像是

$count = $memcache_obj->get('count');
if ($count < 100) {
   $memcache_obj->set('count', $count+1);
   echo "Welcome";
} else {
   echo "No luck";
}

正如您所看到的,上面的代码中存在竞争条件,但是如果我们不打算替换支持memcached的{​​{1}}扩展名,那么它只能使用PHP代码来支持它吗? / p>

4 个答案:

答案 0 :(得分:5)

如果您关注竞争条件,并且计数值完全是任意的,您可以在任何业务逻辑之前直接使用Memcache::increment

increment方法将在增量发生后返回当前值;其中,你可以比较结果。如果尚未设置密钥,增量也将返回 false ;允许您的应用程序根据需要处理它。

$current = $memcache_obj->increment('count');

if($current === false) {

  // NOT_FOUND, so let's create it
  $memcache_obj->set('count',1); // <-- still risk of race-condition
  echo "Your the first!";

} else if ($current < 100) {

  echo "Hazah! Your under the limit.";

} else {

  echo "Ah Snap! No Luck";
  // If your worried about the value growing _too_ big, just drop the value down.
  // $memcache_obj->decrement('count');

}

答案 1 :(得分:5)

作为对“emcconville”的回答。即使没有CAS,这也是非阻塞的。

如果您关注竞争条件,并且计数值完全是任意的,您可以在任何业务逻辑之前直接使用Memcache::increment

increment方法将在增量发生后返回当前值;其中,你可以比较结果。如果尚未设置密钥,增量也将返回 false ;允许您的应用程序根据需要处理它。

$current = $memcache_obj->increment('count');

if($current === false) {

  // NOT_FOUND, so let's create it
  // Will return false if has just been created by someone else.
  $memcache_obj->add('count',0); // <-- no risk of race-condition

  // At this point 'count' key is created by us or someone else (other server/process).
  // "increment" will update 0 or whatever it is at the moment by 1.
  $current = $memcache_obj->increment('count')
  echo "You are the $current!";

} 
if ($current < 100) {

  echo "Hazah! You are under the limit. Congrats!";

} else {

  echo "Ah Snap! No Luck - you reached the limit.";
  // If your worried about the value growing _too_ big, just drop the value down.
  // $memcache_obj->decrement('count');

}

答案 2 :(得分:1)

function memcache_atomic_increment($counter_name, $delta = 1) {

    $mc = new Memcache;
    $mc->connect('localhost', 11211) or die("Could not connect");

    while (($mc->increment($counter_name, $delta) === false) &&
           ($mc->add($counter_name, ($delta<0)?0:$delta, 0, 0) === false)) {
        // loop until one of them succeeds
    }

    return intval($mc->get($counter_name));
}

答案 3 :(得分:-1)

Memcache::add中的评论包含example locking function,你试过了吗?