mirror of
https://github.com/lkddi/nexusphp.git
synced 2026-04-03 14:10:57 +08:00
65 lines
1.4 KiB
PHP
65 lines
1.4 KiB
PHP
<?php
|
|
|
|
namespace Nexus\Database;
|
|
|
|
use App\Exceptions\LockFailException;
|
|
use Illuminate\Cache\LuaScripts;
|
|
use Illuminate\Cache\RedisLock;
|
|
|
|
class NexusLock extends RedisLock
|
|
{
|
|
|
|
/**
|
|
* @var \Redis
|
|
*/
|
|
protected $redis;
|
|
/**
|
|
* NexusLock constructor.
|
|
* @param string $name
|
|
* @param int $seconds
|
|
* @param null $owner
|
|
*/
|
|
public function __construct($name, $seconds, $owner = null)
|
|
{
|
|
parent::__construct(NexusDB::redis(), $name, $seconds, $owner);
|
|
}
|
|
|
|
/**
|
|
* Attempt to acquire the lock.
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function acquire()
|
|
{
|
|
if ($this->seconds > 0) {
|
|
return $this->redis->set($this->name, $this->owner, ['nx', 'ex' => $this->seconds]) == true;
|
|
} else {
|
|
return $this->redis->setnx($this->name, $this->owner) == true;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Release the lock.
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function release()
|
|
{
|
|
return (bool) $this->redis->eval(LuaScripts::releaseLock(), [$this->name, $this->owner], 1);
|
|
}
|
|
|
|
/**
|
|
* @throws LockFailException
|
|
*/
|
|
public static function lockOrFail($name, $seconds, $owner = null): NexusLock
|
|
{
|
|
$lock = new self($name, $seconds, $owner);
|
|
if (!$lock->acquire()) {
|
|
do_log("$name failed to acquire lock", 'error');
|
|
throw new LockFailException($name, $lock->owner());
|
|
}
|
|
return $lock;
|
|
}
|
|
|
|
}
|