在 WordPress 中使用 Memcached CAS 乐观锁

什么是 Memcached CAS 协议

Memcached 于 1.2.4 版本新增 CAS(Check and Set)协议类同于 Java 并发的 CAS(Compare and Swap)原子操作,处理同一 item 被多个线程更改过程的并发问题。

在 Memcached 中,每个 key 关联有一个 64-bit 长度的 long 型惟一数值,表示该 key 对应 value 的版本号。这个数值由 Memcached server 产生,从1开始,且同一 Memcached server 不会重复。在两种情况下这个版本数值会加1:

  1. 新增一个key-value对;
  2. 对某已有key对应的value值更新成功。删除item版本值不会减小。

如何在 PHP 使用 Memcached CAS

Memcached::cas() 执行一个“检查并设置”的操作,因此,它仅在当前客户端最后一次取值后,该 key 对应的值没有被其他客户端修改的情况下, 才能够将值写入。检查是通过 cas_token 参数进行的, 这个参数是 Memcache 指定给已经存在的元素的一个唯一的64位值, 怎样获取这个值请查看 Memcached::get*() 系列方法的文档。注意:这个值作为 double 类型是因为PHP的整型空间限制。

这是 Memcached 扩展比 Memcache 扩展一个非常重要的优势, 在这样一个系统级(Memcache自身提供)的冲突检测机制(乐观锁)下, 我们才能保证高并发下的数据安全。

给 WordPress 添加 Memcached CAS 协议

在 WordPress 的 object-cache.php 我们可以增加下面这两个函数:


function wp_cache_get_with_cas( $key, $group = '', &$cas_token = null ) {
	global $wp_object_cache;
	return $wp_object_cache->get_with_cas( $key, $group, $cas_token );
}

function wp_cache_cas( $cas_token, $key, $data, $group = '', $expire = 0  ) {
	global $wp_object_cache;
	return $wp_object_cache->cas( $cas_token, $key, $data, $group, $expire );
}

然后在对应的 WP_Object_Cache 中添加相应的方法:


function get_with_cas( $id, $group = 'default', &$cas_token=null){
	$key	= $this->key( $id, $group );

	if(defined('Memcached::GET_EXTENDED')) {
		$result	= $this->mc->get($key, null, Memcached::GET_EXTENDED);

		if ($this->mc->getResultCode() == Memcached::RES_NOTFOUND) {
			$value	= false;
		}else{
			$value		= $result['value'];
			$cas_token 	= $result['cas'];
		}
	}else{
		$value	= $this->mc->get($key, null, $cas_token);

		if ($this->mc->getResultCode() == Memcached::RES_NOTFOUND) {
			$value	= false;
		}
	}

	return $value;
}

function cas( $cas_token, $id, $data, $group = 'default', $expire = 0 ) {
	$key = $this->key( $id, $group );

	if ( is_object( $data ) ) {
		$data = clone $data;
	}

	$this->cache[ $key ] = $data;

	return $this->mc->cas( $cas_token, $key, $data, $expire );
}

用法:


$cas_token = '';
$item	= wp_cache_get_with_cas($key, $cache_group, $cas_token);
// 对 $item 进行一些处理
wp_cache_cas($cas_token, $key, $item, $cache_group, DAY_IN_SECONDS);

热门文章