【系统架构】缓存Memcache 使用原子性操作add，实现并发锁

原文地址

memcache中Memcache::add()方法在缓存服务器之前不存在key时， 以key作为key存储一个变量var到缓存服务器。我们使用add来向服务器添加一个键值对应，如果成功则添加，否则说明存在另一个并发作业在进行操作。通过add的原子性来判断是否要执行热点代码。具体代码需结合上一篇的php使用memcache。使用该方法控制并发需要考虑到缓存的有效期、缓存基于内存的特点。

实现一个包含锁，解锁，锁状态检查的类cacheLock：

class cacheLock{
    const KEY_PREFIX = '_lock';
    private $mc;

    public function __construct(){
        $this->mc = new dlufMemcache('127.0.0.1',11211);
    }

    /**
     * 进行锁操作
     * @param [type]  $lock_id
     * @param integer $expire
     */
    public function Lock($lock_id,$expire=5){
        $mkey = self::KEY_PREFIX.$lock_id;
        for($i = 0; $i < 10; $i++){
            $flag = false;
            try{
                $flag = $this->mc->add($mkey,'1',$expire);
            }catch(Exception $e){
                $flag = false;
                //log
            }
            if($flag){
                return true;
            }else{
                //wait for 0.3 seconds
                usleep(300000);
            }
        }
        return false;
    }

    /**
     * 判断锁状态
     * @param  [type]  $lock_id
     * @return boolean
     */
    public function isLock($lock_id){
        $mkey = self::KEY_PREFIX.$lock_id;
        $ret = $this->mc->get($mkey);
        if(empty($ret) || $ret === false){
            return false;
        }
        return true;
    }

    /**
     * 解锁
     * @param  [type] $lock_id
     * @return [type]
     */
    public function unLock($lock_id){
        $mkey = self::KEY_PREFIX.$lock_id;
        $ret = $this->mc->delete($mkey);
        return $ret;
    }

 }

然后使用cacheLock进行并发热点代码控制

function useLock(){
     $lockobj = new cacheLock();
     $lock = $lockobj->Lock('cachelock');
     if(!$lock){
        echo "cachelock is lock";
        return;
     }

     /*  热点代码 写在这里
      */

     $lockobj->unLock('cachelock');
}