resurtm
12 years ago
5 changed files with 230 additions and 0 deletions
@ -0,0 +1,76 @@ |
|||||||
|
<?php |
||||||
|
|
||||||
|
namespace yii\mutex; |
||||||
|
|
||||||
|
use Yii; |
||||||
|
use yii\base\Component; |
||||||
|
|
||||||
|
abstract class Mutex extends Component |
||||||
|
{ |
||||||
|
public $autoRelease = true; |
||||||
|
|
||||||
|
private $_locks = array(); |
||||||
|
|
||||||
|
|
||||||
|
public function init() |
||||||
|
{ |
||||||
|
if ($this->autoRelease) { |
||||||
|
register_shutdown_function(array($this, 'shutdownFunction')); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* NEVER CALL THIS METHOD UNDER ANY CIRCUMSTANCES |
||||||
|
*/ |
||||||
|
public function shutdownFunction() |
||||||
|
{ |
||||||
|
foreach ($this->_locks as $lock) { |
||||||
|
$this->release($lock); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public function acquireLock($name, $timeout = 0) |
||||||
|
{ |
||||||
|
if ($this->acquire($name, $timeout)) { |
||||||
|
$this->_locks[] = $name; |
||||||
|
return true; |
||||||
|
} else { |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public function releaseLock($name) |
||||||
|
{ |
||||||
|
if ($this->release($name)) { |
||||||
|
unset($this->_locks[array_search($name, $this->_locks)]); |
||||||
|
return true; |
||||||
|
} else { |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public function getIsLockAcquired($name) |
||||||
|
{ |
||||||
|
if (in_array($name, $this->_locks)) { |
||||||
|
return true; |
||||||
|
} else { |
||||||
|
return $this->getIsAcquired($name); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public function getIsLockLocal($name) |
||||||
|
{ |
||||||
|
return in_array($name, $this->_locks); |
||||||
|
} |
||||||
|
|
||||||
|
abstract protected function acquire($name, $timeout = 0); |
||||||
|
|
||||||
|
abstract protected function release($name); |
||||||
|
|
||||||
|
protected function getIsAcquired($name) |
||||||
|
{ |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
abstract public function getIsDistributed(); |
||||||
|
} |
@ -0,0 +1,29 @@ |
|||||||
|
<?php |
||||||
|
|
||||||
|
namespace yii\mutex\db; |
||||||
|
|
||||||
|
use Yii; |
||||||
|
use yii\db\Connection; |
||||||
|
use yii\base\InvalidConfigException; |
||||||
|
|
||||||
|
abstract class Mutex extends \yii\mutex\Mutex |
||||||
|
{ |
||||||
|
/** |
||||||
|
* @var string|Connection |
||||||
|
*/ |
||||||
|
public $db = 'db'; |
||||||
|
|
||||||
|
public function init() |
||||||
|
{ |
||||||
|
parent::init(); |
||||||
|
$this->db = Yii::$app->getComponent($this->db); |
||||||
|
if (!$this->db instanceof Connection) { |
||||||
|
throw new InvalidConfigException(''); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public function getIsDistributed() |
||||||
|
{ |
||||||
|
return true; |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,28 @@ |
|||||||
|
<?php |
||||||
|
|
||||||
|
namespace yii\mutex\db\mssql; |
||||||
|
|
||||||
|
use Yii; |
||||||
|
use yii\base\InvalidConfigException; |
||||||
|
|
||||||
|
class Mutex extends \yii\mutex\db\Mutex |
||||||
|
{ |
||||||
|
public function init() |
||||||
|
{ |
||||||
|
parent::init(); |
||||||
|
$driverName = $this->db->driverName; |
||||||
|
if ($driverName !== 'sqlsrv' && $driverName !== 'dblib' && $driverName !== 'mssql') { |
||||||
|
throw new InvalidConfigException(''); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
protected function acquire($name, $timeout = 0) |
||||||
|
{ |
||||||
|
// http://msdn.microsoft.com/en-us/library/ms189823.aspx |
||||||
|
} |
||||||
|
|
||||||
|
protected function release($name) |
||||||
|
{ |
||||||
|
// http://msdn.microsoft.com/en-us/library/ms178602.aspx |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,38 @@ |
|||||||
|
<?php |
||||||
|
|
||||||
|
namespace yii\mutex\db\mysql; |
||||||
|
|
||||||
|
use Yii; |
||||||
|
use yii\base\InvalidConfigException; |
||||||
|
|
||||||
|
class Mutex extends \yii\mutex\db\Mutex |
||||||
|
{ |
||||||
|
public function init() |
||||||
|
{ |
||||||
|
parent::init(); |
||||||
|
if ($this->db->driverName !== 'mysql') { |
||||||
|
throw new InvalidConfigException(''); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
protected function acquire($name, $timeout = 0) |
||||||
|
{ |
||||||
|
return (boolean)$this->db |
||||||
|
->createCommand('SELECT GET_LOCK(:name, :timeout)', array(':name' => $name, ':timeout' => $timeout)) |
||||||
|
->queryScalar(); |
||||||
|
} |
||||||
|
|
||||||
|
protected function release($name) |
||||||
|
{ |
||||||
|
return (boolean)$this->db |
||||||
|
->createCommand('SELECT RELEASE_LOCK(:name)', array(':name' => $name)) |
||||||
|
->queryScalar(); |
||||||
|
} |
||||||
|
|
||||||
|
protected function getIsAcquired($name) |
||||||
|
{ |
||||||
|
return (boolean)$this->db |
||||||
|
->createCommand('SELECT IS_FREE_LOCK(:name)', array(':name' => $name)) |
||||||
|
->queryScalar(); |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,59 @@ |
|||||||
|
<?php |
||||||
|
|
||||||
|
namespace yii\mutex\unix; |
||||||
|
|
||||||
|
use Yii; |
||||||
|
use yii\base\InvalidConfigException; |
||||||
|
|
||||||
|
class Mutex extends \yii\mutex\Mutex |
||||||
|
{ |
||||||
|
private $_files = array(); |
||||||
|
|
||||||
|
|
||||||
|
public function init() |
||||||
|
{ |
||||||
|
// if (stripos(php_uname('s'), 'win') === 0) { |
||||||
|
// throw new InvalidConfigException(''); |
||||||
|
// } |
||||||
|
} |
||||||
|
|
||||||
|
protected function acquire($name, $timeout = 0) |
||||||
|
{ |
||||||
|
$file = fopen(Yii::$app->getRuntimePath() . '/mutex.' . md5($name) . '.lock', 'w+'); |
||||||
|
if ($file === false) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
$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; |
||||||
|
} |
||||||
|
|
||||||
|
protected function release($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; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
protected function getIsAcquired($name) |
||||||
|
{ |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
public function getIsDistributed() |
||||||
|
{ |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue