364 lines
8.6 KiB
PHP
364 lines
8.6 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2006~2018 http://thinkphp.cn All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
// +----------------------------------------------------------------------
|
|
// | Author: liu21st <liu21st@gmail.com>
|
|
// +----------------------------------------------------------------------
|
|
|
|
namespace think\cache;
|
|
|
|
use think\Container;
|
|
|
|
/**
|
|
* 缓存基础类
|
|
*/
|
|
abstract class Driver
|
|
{
|
|
/**
|
|
* 驱动句柄
|
|
* @var object
|
|
*/
|
|
protected $handler = null;
|
|
|
|
/**
|
|
* 缓存读取次数
|
|
* @var integer
|
|
*/
|
|
protected $readTimes = 0;
|
|
|
|
/**
|
|
* 缓存写入次数
|
|
* @var integer
|
|
*/
|
|
protected $writeTimes = 0;
|
|
|
|
/**
|
|
* 缓存参数
|
|
* @var array
|
|
*/
|
|
protected $options = [];
|
|
|
|
/**
|
|
* 缓存标签
|
|
* @var string
|
|
*/
|
|
protected $tag;
|
|
|
|
/**
|
|
* 序列化方法
|
|
* @var array
|
|
*/
|
|
protected static $serialize = ['serialize', 'unserialize', 'think_serialize:', 16];
|
|
|
|
/**
|
|
* 判断缓存是否存在
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @return bool
|
|
*/
|
|
abstract public function has($name);
|
|
|
|
/**
|
|
* 读取缓存
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param mixed $default 默认值
|
|
* @return mixed
|
|
*/
|
|
abstract public function get($name, $default = false);
|
|
|
|
/**
|
|
* 写入缓存
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param mixed $value 存储数据
|
|
* @param int $expire 有效时间 0为永久
|
|
* @return boolean
|
|
*/
|
|
abstract public function set($name, $value, $expire = null);
|
|
|
|
/**
|
|
* 自增缓存（针对数值缓存）
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param int $step 步长
|
|
* @return false|int
|
|
*/
|
|
abstract public function inc($name, $step = 1);
|
|
|
|
/**
|
|
* 自减缓存（针对数值缓存）
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param int $step 步长
|
|
* @return false|int
|
|
*/
|
|
abstract public function dec($name, $step = 1);
|
|
|
|
/**
|
|
* 删除缓存
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @return boolean
|
|
*/
|
|
abstract public function rm($name);
|
|
|
|
/**
|
|
* 清除缓存
|
|
* @access public
|
|
* @param string $tag 标签名
|
|
* @return boolean
|
|
*/
|
|
abstract public function clear($tag = null);
|
|
|
|
/**
|
|
* 获取有效期
|
|
* @access protected
|
|
* @param integer|\DateTime $expire 有效期
|
|
* @return integer
|
|
*/
|
|
protected function getExpireTime($expire)
|
|
{
|
|
if ($expire instanceof \DateTime) {
|
|
$expire = $expire->getTimestamp() - time();
|
|
}
|
|
|
|
return $expire;
|
|
}
|
|
|
|
/**
|
|
* 获取实际的缓存标识
|
|
* @access protected
|
|
* @param string $name 缓存名
|
|
* @return string
|
|
*/
|
|
protected function getCacheKey($name)
|
|
{
|
|
return $this->options['prefix'] . $name;
|
|
}
|
|
|
|
/**
|
|
* 读取缓存并删除
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @return mixed
|
|
*/
|
|
public function pull($name)
|
|
{
|
|
$result = $this->get($name, false);
|
|
|
|
if ($result) {
|
|
$this->rm($name);
|
|
return $result;
|
|
} else {
|
|
return;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 如果不存在则写入缓存
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param mixed $value 存储数据
|
|
* @param int $expire 有效时间 0为永久
|
|
* @return mixed
|
|
*/
|
|
public function remember($name, $value, $expire = null)
|
|
{
|
|
if (!$this->has($name)) {
|
|
$time = time();
|
|
while ($time + 5 > time() && $this->has($name . '_lock')) {
|
|
// 存在锁定则等待
|
|
usleep(200000);
|
|
}
|
|
|
|
try {
|
|
// 锁定
|
|
$this->set($name . '_lock', true);
|
|
|
|
if ($value instanceof \Closure) {
|
|
// 获取缓存数据
|
|
$value = Container::getInstance()->invokeFunction($value);
|
|
}
|
|
|
|
// 缓存数据
|
|
$this->set($name, $value, $expire);
|
|
|
|
// 解锁
|
|
$this->rm($name . '_lock');
|
|
} catch (\Exception $e) {
|
|
$this->rm($name . '_lock');
|
|
throw $e;
|
|
} catch (\throwable $e) {
|
|
$this->rm($name . '_lock');
|
|
throw $e;
|
|
}
|
|
} else {
|
|
$value = $this->get($name);
|
|
}
|
|
|
|
return $value;
|
|
}
|
|
|
|
/**
|
|
* 缓存标签
|
|
* @access public
|
|
* @param string $name 标签名
|
|
* @param string|array $keys 缓存标识
|
|
* @param bool $overlay 是否覆盖
|
|
* @return $this
|
|
*/
|
|
public function tag($name, $keys = null, $overlay = false)
|
|
{
|
|
if (is_null($name)) {
|
|
|
|
} elseif (is_null($keys)) {
|
|
$this->tag = $name;
|
|
} else {
|
|
$key = $this->getTagkey($name);
|
|
|
|
if (is_string($keys)) {
|
|
$keys = explode(',', $keys);
|
|
}
|
|
|
|
$keys = array_map([$this, 'getCacheKey'], $keys);
|
|
|
|
if ($overlay) {
|
|
$value = $keys;
|
|
} else {
|
|
$value = array_unique(array_merge($this->getTagItem($name), $keys));
|
|
}
|
|
|
|
$this->set($key, implode(',', $value), 0);
|
|
}
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 更新标签
|
|
* @access protected
|
|
* @param string $name 缓存标识
|
|
* @return void
|
|
*/
|
|
protected function setTagItem($name)
|
|
{
|
|
if ($this->tag) {
|
|
$key = $this->getTagkey($this->tag);
|
|
$prev = $this->tag;
|
|
$this->tag = null;
|
|
|
|
if ($this->has($key)) {
|
|
$value = explode(',', $this->get($key));
|
|
$value[] = $name;
|
|
|
|
if (count($value) > 1000) {
|
|
array_shift($value);
|
|
}
|
|
|
|
$value = implode(',', array_unique($value));
|
|
} else {
|
|
$value = $name;
|
|
}
|
|
|
|
$this->set($key, $value, 0);
|
|
$this->tag = $prev;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取标签包含的缓存标识
|
|
* @access protected
|
|
* @param string $tag 缓存标签
|
|
* @return array
|
|
*/
|
|
protected function getTagItem($tag)
|
|
{
|
|
$key = $this->getTagkey($tag);
|
|
$value = $this->get($key);
|
|
|
|
if ($value) {
|
|
return array_filter(explode(',', $value));
|
|
} else {
|
|
return [];
|
|
}
|
|
}
|
|
|
|
protected function getTagKey($tag)
|
|
{
|
|
return 'tag_' . md5($tag);
|
|
}
|
|
|
|
/**
|
|
* 序列化数据
|
|
* @access protected
|
|
* @param mixed $data
|
|
* @return string
|
|
*/
|
|
protected function serialize($data)
|
|
{
|
|
if (is_scalar($data) || !$this->options['serialize']) {
|
|
return $data;
|
|
}
|
|
|
|
$serialize = self::$serialize[0];
|
|
|
|
return self::$serialize[2] . $serialize($data);
|
|
}
|
|
|
|
/**
|
|
* 反序列化数据
|
|
* @access protected
|
|
* @param string $data
|
|
* @return mixed
|
|
*/
|
|
protected function unserialize($data)
|
|
{
|
|
if ($this->options['serialize'] && 0 === strpos($data, self::$serialize[2])) {
|
|
$unserialize = self::$serialize[1];
|
|
|
|
return $unserialize(substr($data, self::$serialize[3]));
|
|
} else {
|
|
return $data;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 注册序列化机制
|
|
* @access public
|
|
* @param callable $serialize 序列化方法
|
|
* @param callable $unserialize 反序列化方法
|
|
* @param string $prefix 序列化前缀标识
|
|
* @return $this
|
|
*/
|
|
public static function registerSerialize($serialize, $unserialize, $prefix = 'think_serialize:')
|
|
{
|
|
self::$serialize = [$serialize, $unserialize, $prefix, strlen($prefix)];
|
|
}
|
|
|
|
/**
|
|
* 返回句柄对象，可执行其它高级方法
|
|
*
|
|
* @access public
|
|
* @return object
|
|
*/
|
|
public function handler()
|
|
{
|
|
return $this->handler;
|
|
}
|
|
|
|
public function getReadTimes()
|
|
{
|
|
return $this->readTimes;
|
|
}
|
|
|
|
public function getWriteTimes()
|
|
{
|
|
return $this->writeTimes;
|
|
}
|
|
}
|