hyperf-chat/app/Cache/Repository/LockRedis.php

73 lines
1.5 KiB
PHP

<?php
namespace App\Cache\Repository;
use App\Cache\Contracts\LockRedisInterface;
/**
* Redis Lock
*
* @package App\Cache\Repository
*/
class LockRedis extends AbstractRedis implements LockRedisInterface
{
protected $prefix = 'rds-lock';
protected $lockValue = 1;
/**
* 获取是毫秒时间戳
*
* @return int
*/
private function time()
{
return intval(microtime(true) * 1000);
}
/**
* 获取 Redis 锁
*
* @param string $key 锁标识
* @param int $lockTime 过期时间/秒
* @param int $timeout 获取超时/毫秒
* @return bool
*/
public function lock(string $key, $lockTime = 1, $timeout = 0)
{
$lockName = $this->getCacheKey($key);
$start = $this->time();
do {
$lock = $this->redis()->set($lockName, $this->lockValue, ['nx', 'ex' => $lockTime]);
if ($lock || $timeout === 0) {
break;
}
// 默认 0.1 秒一次取锁
usleep(100000);
} while ($this->time() < $start + $timeout);
return $lock;
}
/**
* 释放 Redis 锁
*
* @param string $key
* @return mixed
*/
public function delete(string $key)
{
$script = <<<LAU
if redis.call("GET", KEYS[1]) == ARGV[1] then
return redis.call("DEL", KEYS[1])
else
return 0
end
LAU;
return $this->redis()->eval($script, [$this->getCacheKey($key), $this->lockValue], 1);
}
}