You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							104 lines
						
					
					
						
							2.8 KiB
						
					
					
				
			
		
		
	
	
							104 lines
						
					
					
						
							2.8 KiB
						
					
					
				<?php | 
						|
/** | 
						|
 * @link http://www.yiiframework.com/ | 
						|
 * @copyright Copyright (c) 2008 Yii Software LLC | 
						|
 * @license http://www.yiiframework.com/license/ | 
						|
 */ | 
						|
 | 
						|
namespace yii\mutex; | 
						|
 | 
						|
use Yii; | 
						|
use yii\base\InvalidConfigException; | 
						|
use yii\helpers\FileHelper; | 
						|
 | 
						|
/** | 
						|
 * @author resurtm <resurtm@gmail.com> | 
						|
 * @since 2.0 | 
						|
 */ | 
						|
class FileMutex extends Mutex | 
						|
{ | 
						|
	/** | 
						|
	 * @var string the directory to store mutex files. You may use path alias here. | 
						|
	 * Defaults to the "mutex" subdirectory under the application runtime path. | 
						|
	 */ | 
						|
	public $mutexPath = '@runtime/mutex'; | 
						|
	/** | 
						|
	 * @var integer the permission to be set for newly created mutex files. | 
						|
	 * This value will be used by PHP chmod() function. No umask will be applied. | 
						|
	 * If not set, the permission will be determined by the current environment. | 
						|
	 */ | 
						|
	public $fileMode; | 
						|
	/** | 
						|
	 * @var integer the permission to be set for newly created directories. | 
						|
	 * This value will be used by PHP chmod() function. No umask will be applied. | 
						|
	 * Defaults to 0775, meaning the directory is read-writable by owner and group, | 
						|
	 * but read-only for other users. | 
						|
	 */ | 
						|
	public $dirMode = 0775; | 
						|
	/** | 
						|
	 * @var resource[] stores all opened lock files. Keys are lock names and values are file handles. | 
						|
	 */ | 
						|
	private $_files = []; | 
						|
 | 
						|
 | 
						|
	/** | 
						|
	 * Initializes mutex component implementation dedicated for UNIX, GNU/Linux, Mac OS X, and other UNIX-like | 
						|
	 * operating systems. | 
						|
	 * @throws InvalidConfigException | 
						|
	 */ | 
						|
	public function init() | 
						|
	{ | 
						|
		if (stripos(php_uname('s'), 'win') === 0) { | 
						|
			throw new InvalidConfigException('FileMutex does not have MS Windows operating system support.'); | 
						|
		} | 
						|
		$this->mutexPath = Yii::getAlias($this->mutexPath); | 
						|
		if (!is_dir($this->mutexPath)) { | 
						|
			FileHelper::createDirectory($this->mutexPath, $this->dirMode, true); | 
						|
		} | 
						|
	} | 
						|
 | 
						|
	/** | 
						|
	 * Acquires lock by given name. | 
						|
	 * @param string $name of the lock to be acquired. | 
						|
	 * @param integer $timeout to wait for lock to become released. | 
						|
	 * @return boolean acquiring result. | 
						|
	 */ | 
						|
	protected function acquireLock($name, $timeout = 0) | 
						|
	{ | 
						|
		$fileName = $this->mutexPath . '/' . md5($name) . '.lock'; | 
						|
		$file = fopen($fileName, 'w+'); | 
						|
		if ($file === false) { | 
						|
			return false; | 
						|
		} | 
						|
		if ($this->fileMode !== null) { | 
						|
			@chmod($fileName, $this->fileMode); | 
						|
		} | 
						|
		$waitTime = 0; | 
						|
		while (!flock($file, LOCK_EX | LOCK_NB)) { | 
						|
			$waitTime++; | 
						|
			if ($waitTime > $timeout) { | 
						|
				fclose($file); | 
						|
				return false; | 
						|
			} | 
						|
			sleep(1); | 
						|
		} | 
						|
		$this->_files[$name] = $file; | 
						|
		return true; | 
						|
	} | 
						|
 | 
						|
	/** | 
						|
	 * Releases lock by given name. | 
						|
	 * @param string $name of the lock to be released. | 
						|
	 * @return boolean release result. | 
						|
	 */ | 
						|
	protected function releaseLock($name) | 
						|
	{ | 
						|
		if (!isset($this->_files[$name]) || !flock($this->_files[$name], LOCK_UN)) { | 
						|
			return false; | 
						|
		} else { | 
						|
			fclose($this->_files[$name]); | 
						|
			unset($this->_files[$name]); | 
						|
			return true; | 
						|
		} | 
						|
	} | 
						|
}
 | 
						|
 |