PHP中使用Memache作为进程锁的操作类分享

<?php

// 使用Memache 作为进程锁 

class lock_processlock{

	// key 的前缀
	protected $sLockKeyPre;
	// 重试间隔
 protected $iLockRetryInterval;
	//重试次数
 protected $iLockRetryCount;
	//锁的过期时间
 protected $iLockCacheTimeout;
 // 锁过期后的回调函数
 protected $onLockTimeoutFunc;
	// memache 的实例
 protected $oMemcache;
	// 存储memcache失败后重试次数
 protected $iMemcacheRetryCount;

	 public function __construct ($onLockTimeoutFunc=NULL) {
  $aLockConfig = get_config('', 'lock');
  $this->sLockKeyPre = self::LOCK_KEY_PRE;
  $this->iLockRetryInterval = self::LOCK_RETRY_INTERVAL;
  $this->iLockRetryCount =self::LOCK_RETRY_COUNT;
  $this->iLockCacheTimeout = self::LOCK_CACHE_TIMEOUT;
  $this->iMemcacheRetryCount = self::LOCK_CACHE_TIMEOUT;
		if(!$onLockTimeoutFunc){
			// 如果加锁不成功则调用回调函数,如果没有回调函数,使用本类中所带的
			$onLockTimeoutFunc ='onLockTimeout' ;
		}
  $this->onLockTimeoutFunc = $onLockTimeoutFunc;
 }

	/**
	连接memcache 服务器
	*/
	public function connect() {
		if (! isset ( $this->oMemcache )) {
			$this->oMemcache = new Memcache ();
			$this->oMemcache->connect ( '127.0.0.1', 11211 );
		}
		return $this->oMemcache;
	}

	/*
	向MeMcache中添加 key
	*/
	public addMemcache($sKey, $sValue, $iTimeout){

		for($i= 0 ; $i<$this->iMemcacheRetryCount){
			$bRes = $this->oMemcache->add($sKey, $sValue, $iTimeout);
			if($bRes){
				return true ;
			}
				// 如果加锁不成功,sleep 之后,从新加锁
			usleep($this->iLockRetryInterval*1000);

		}
		return false ; 

	}

	/*
	加锁
	*/
	public function lock($sLockID){

		$oMemcache = $this->connect();
	 $sKey = $this->sLockKeyPre . $sLockID;

		// 加锁如果不成功可以多试几次 

		for($i = 0 ; $i <$this->iLockRetryCount ; $i++){

			// 这里设置value 的值可以随便设置
			if($this->addMemcache($sKey,'1',$this->iLockCacheTimeout)){
				return true ;
			}

			// 如果加锁不成功,sleep 之后,从新加锁
			usleep($this->iLockRetryInterval*1000);

		}

		// 若还不成功,则加锁失败,调用回调函数,.也就是失败后需要处理的操作
		if(is_callable($this->onLockTimeoutFunc)){
			// 调用函数
			call_user_func($this->onLockTimeoutFunc);
		}

	}

	/*
	解锁操作
	*/
	public function unlock($sLockID){

		$oMemcache = $this->connect();
	 $sKey = $this->sLockKeyPre . $sLockID;
		// 删除key
		return $this->oMemcache->delete($sKey);

	}

	/**
	如果加锁不成功,则执行如下操作
	*/
	 public function onLockTimeout(){

		 echo ("加锁超时");
	 }

}

// 应用实例 

 $oLock = new lock_processlock();
 $lockResource = "test";
 // 加锁
 $oLock->lock($lockResource);
 // 解锁
 $oLock->unlock($lockResource);