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.
105 lines
2.8 KiB
105 lines
2.8 KiB
12 years ago
|
<?php
|
||
12 years ago
|
/**
|
||
|
* @link http://www.yiiframework.com/
|
||
|
* @copyright Copyright (c) 2008 Yii Software LLC
|
||
|
* @license http://www.yiiframework.com/license/
|
||
|
*/
|
||
12 years ago
|
|
||
12 years ago
|
namespace yii\mutex;
|
||
12 years ago
|
|
||
|
use Yii;
|
||
|
use yii\base\InvalidConfigException;
|
||
11 years ago
|
use yii\helpers\FileHelper;
|
||
12 years ago
|
|
||
12 years ago
|
/**
|
||
|
* @author resurtm <resurtm@gmail.com>
|
||
|
* @since 2.0
|
||
|
*/
|
||
12 years ago
|
class FileMutex extends Mutex
|
||
12 years ago
|
{
|
||
12 years ago
|
/**
|
||
12 years ago
|
* @var string the directory to store mutex files. You may use path alias here.
|
||
11 years ago
|
* Defaults to the "mutex" subdirectory under the application runtime path.
|
||
12 years ago
|
*/
|
||
|
public $mutexPath = '@runtime/mutex';
|
||
|
/**
|
||
11 years ago
|
* @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;
|
||
|
/**
|
||
12 years ago
|
* @var resource[] stores all opened lock files. Keys are lock names and values are file handles.
|
||
|
*/
|
||
11 years ago
|
private $_files = [];
|
||
12 years ago
|
|
||
|
|
||
12 years ago
|
/**
|
||
|
* Initializes mutex component implementation dedicated for UNIX, GNU/Linux, Mac OS X, and other UNIX-like
|
||
|
* operating systems.
|
||
|
* @throws InvalidConfigException
|
||
|
*/
|
||
12 years ago
|
public function init()
|
||
|
{
|
||
12 years ago
|
if (stripos(php_uname('s'), 'win') === 0) {
|
||
12 years ago
|
throw new InvalidConfigException('FileMutex does not have MS Windows operating system support.');
|
||
|
}
|
||
|
$this->mutexPath = Yii::getAlias($this->mutexPath);
|
||
|
if (!is_dir($this->mutexPath)) {
|
||
11 years ago
|
FileHelper::createDirectory($this->mutexPath, $this->dirMode, true);
|
||
12 years ago
|
}
|
||
12 years ago
|
}
|
||
|
|
||
12 years ago
|
/**
|
||
11 years ago
|
* Acquires lock by given name.
|
||
12 years ago
|
* @param string $name of the lock to be acquired.
|
||
|
* @param integer $timeout to wait for lock to become released.
|
||
|
* @return boolean acquiring result.
|
||
|
*/
|
||
12 years ago
|
protected function acquireLock($name, $timeout = 0)
|
||
12 years ago
|
{
|
||
11 years ago
|
$fileName = $this->mutexPath . '/' . md5($name) . '.lock';
|
||
|
$file = fopen($fileName, 'w+');
|
||
12 years ago
|
if ($file === false) {
|
||
|
return false;
|
||
|
}
|
||
11 years ago
|
if ($this->fileMode !== null) {
|
||
|
@chmod($fileName, $this->fileMode);
|
||
|
}
|
||
12 years ago
|
$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;
|
||
|
}
|
||
|
|
||
12 years ago
|
/**
|
||
11 years ago
|
* Releases lock by given name.
|
||
12 years ago
|
* @param string $name of the lock to be released.
|
||
|
* @return boolean release result.
|
||
|
*/
|
||
12 years ago
|
protected function releaseLock($name)
|
||
12 years ago
|
{
|
||
|
if (!isset($this->_files[$name]) || !flock($this->_files[$name], LOCK_UN)) {
|
||
|
return false;
|
||
|
} else {
|
||
|
fclose($this->_files[$name]);
|
||
|
unset($this->_files[$name]);
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
}
|