124 lines
2.9 KiB
PHP
124 lines
2.9 KiB
PHP
<?php
|
|
|
|
namespace traits;
|
|
|
|
use bw\Redis;
|
|
|
|
|
|
/**
|
|
* 缓存
|
|
*/
|
|
trait CacheTrait
|
|
{
|
|
|
|
|
|
private $cache_lock_error_msg = "获取锁失败！";
|
|
|
|
private $cache_lock_thorws_excption = true;
|
|
|
|
public function setCacheLockThorwsExcption($cache_lock_thorws_excption){
|
|
$this->cache_lock_thorws_excption = $cache_lock_thorws_excption;
|
|
return $this;
|
|
}
|
|
|
|
|
|
public function getCacheLockThorwsExcption(){
|
|
return $this->cache_lock_thorws_excption;
|
|
}
|
|
|
|
|
|
|
|
|
|
public function hasRedis($is_interrupt = false) {
|
|
$error_msg = '';
|
|
try {
|
|
$redis = $this->getRedis();
|
|
|
|
// 检测连接是否正常
|
|
$redis->ping();
|
|
} catch (\BadFunctionCallException $e) {
|
|
// 缺少扩展
|
|
$error_msg = $e->getMessage() ? $e->getMessage() : "缺少 redis 扩展";
|
|
} catch (\RedisException $e) {
|
|
// 连接拒绝
|
|
\think\Log::write('redis connection redisException fail: ' . $e->getMessage());
|
|
$error_msg = $e->getMessage() ? $e->getMessage() : "redis 连接失败";
|
|
} catch (\Exception $e) {
|
|
// 异常
|
|
\think\Log::write('redis connection fail: ' . $e->getMessage());
|
|
$error_msg = $e->getMessage() ? $e->getMessage() : "redis 连接异常";
|
|
}
|
|
|
|
if ($error_msg) {
|
|
if ($is_interrupt || $this->cache_lock_thorws_excption) {
|
|
throw new \Exception($error_msg);
|
|
} else {
|
|
$this->setCacheLockErrorMsg($error_msg);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
public function getRedis() {
|
|
if (!isset($GLOBALS['SPREDIS'])) {
|
|
$GLOBALS['SPREDIS'] = (new Redis())->getRedis();
|
|
}
|
|
|
|
return $GLOBALS['SPREDIS'];
|
|
}
|
|
|
|
|
|
|
|
public function getLock($key,$suffix = "-lock-suffix", $timeout = 120){
|
|
$this->hasRedis(true);
|
|
$redis = $this->getRedis();
|
|
$hashKey = $key.$suffix;
|
|
// $redis->del($hashKey);
|
|
//判断锁是否存在，如果存在，返回错误
|
|
if ($redis->EXISTS($hashKey)){
|
|
if($this->cache_lock_thorws_excption){
|
|
throw new \Exception($this->cache_lock_error_msg);
|
|
}else{
|
|
return false;
|
|
}
|
|
|
|
}
|
|
//如果不存在创建锁并返回
|
|
$redis->set($hashKey, 1,$timeout);
|
|
return true;
|
|
}
|
|
|
|
|
|
public function setCacheLockErrorMsg($cache_lock_error_msg){
|
|
$this->cache_lock_error_msg = $cache_lock_error_msg;
|
|
return $this;
|
|
}
|
|
|
|
|
|
public function getCacheLockErrorMsg(){
|
|
return $this->cache_lock_error_msg;
|
|
}
|
|
|
|
|
|
public function freeLock($key,$suffix = "-lock-suffix"){
|
|
$this->hasRedis(true);
|
|
$redis = $this->getRedis();
|
|
if(!$key || !$suffix) return true;
|
|
$hashKey = $key.$suffix;
|
|
if (!$redis->EXISTS($hashKey)) return true;
|
|
$redis->del($hashKey);
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|