269 lines
7.3 KiB
PHP
269 lines
7.3 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\driver;
|
|
|
|
use think\cache\Driver;
|
|
|
|
/**
|
|
* 文件类型缓存类
|
|
* @author liu21st <liu21st@gmail.com>
|
|
*/
|
|
class File extends Driver
|
|
{
|
|
protected $options = [
|
|
'expire' => 0,
|
|
'cache_subdir' => true,
|
|
'prefix' => '',
|
|
'path' => CACHE_PATH,
|
|
'data_compress' => false,
|
|
];
|
|
|
|
protected $expire;
|
|
|
|
/**
|
|
* 构造函数
|
|
* @param array $options
|
|
*/
|
|
public function __construct($options = [])
|
|
{
|
|
if (!empty($options)) {
|
|
$this->options = array_merge($this->options, $options);
|
|
}
|
|
if (substr($this->options['path'], -1) != DS) {
|
|
$this->options['path'] .= DS;
|
|
}
|
|
$this->init();
|
|
}
|
|
|
|
/**
|
|
* 初始化检查
|
|
* @access private
|
|
* @return boolean
|
|
*/
|
|
private function init()
|
|
{
|
|
// 创建项目缓存目录
|
|
if (!is_dir($this->options['path'])) {
|
|
if (mkdir($this->options['path'], 0755, true)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* 取得变量的存储文件名
|
|
* @access protected
|
|
* @param string $name 缓存变量名
|
|
* @param bool $auto 是否自动创建目录
|
|
* @return string
|
|
*/
|
|
protected function getCacheKey($name, $auto = false)
|
|
{
|
|
$name = md5($name);
|
|
if ($this->options['cache_subdir']) {
|
|
// 使用子目录
|
|
$name = substr($name, 0, 2) . DS . substr($name, 2);
|
|
}
|
|
if ($this->options['prefix']) {
|
|
$name = $this->options['prefix'] . DS . $name;
|
|
}
|
|
$filename = $this->options['path'] . $name . '.php';
|
|
$dir = dirname($filename);
|
|
|
|
if ($auto && !is_dir($dir)) {
|
|
mkdir($dir, 0755, true);
|
|
}
|
|
return $filename;
|
|
}
|
|
|
|
/**
|
|
* 判断缓存是否存在
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @return bool
|
|
*/
|
|
public function has($name)
|
|
{
|
|
return $this->get($name) ? true : false;
|
|
}
|
|
|
|
/**
|
|
* 读取缓存
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param mixed $default 默认值
|
|
* @return mixed
|
|
*/
|
|
public function get($name, $default = false)
|
|
{
|
|
$filename = $this->getCacheKey($name);
|
|
if (!is_file($filename)) {
|
|
return $default;
|
|
}
|
|
$content = file_get_contents($filename);
|
|
$this->expire = null;
|
|
if (false !== $content) {
|
|
$expire = (int) substr($content, 8, 12);
|
|
if (0 != $expire && time() > filemtime($filename) + $expire) {
|
|
return $default;
|
|
}
|
|
$this->expire = $expire;
|
|
$content = substr($content, 32);
|
|
if ($this->options['data_compress'] && function_exists('gzcompress')) {
|
|
//启用数据压缩
|
|
$content = gzuncompress($content);
|
|
}
|
|
$content = unserialize($content);
|
|
return $content;
|
|
} else {
|
|
return $default;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 写入缓存
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param mixed $value 存储数据
|
|
* @param integer|\DateTime $expire 有效时间（秒）
|
|
* @return boolean
|
|
*/
|
|
public function set($name, $value, $expire = null)
|
|
{
|
|
if (is_null($expire)) {
|
|
$expire = $this->options['expire'];
|
|
}
|
|
if ($expire instanceof \DateTime) {
|
|
$expire = $expire->getTimestamp() - time();
|
|
}
|
|
$filename = $this->getCacheKey($name, true);
|
|
if ($this->tag && !is_file($filename)) {
|
|
$first = true;
|
|
}
|
|
$data = serialize($value);
|
|
if ($this->options['data_compress'] && function_exists('gzcompress')) {
|
|
//数据压缩
|
|
$data = gzcompress($data, 3);
|
|
}
|
|
$data = "<?php\n//" . sprintf('%012d', $expire) . "\n exit();?>\n" . $data;
|
|
$result = file_put_contents($filename, $data);
|
|
if ($result) {
|
|
isset($first) && $this->setTagItem($filename);
|
|
clearstatcache();
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 自增缓存（针对数值缓存）
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param int $step 步长
|
|
* @return false|int
|
|
*/
|
|
public function inc($name, $step = 1)
|
|
{
|
|
if ($this->has($name)) {
|
|
$value = $this->get($name) + $step;
|
|
$expire = $this->expire;
|
|
} else {
|
|
$value = $step;
|
|
$expire = 0;
|
|
}
|
|
|
|
return $this->set($name, $value, $expire) ? $value : false;
|
|
}
|
|
|
|
/**
|
|
* 自减缓存（针对数值缓存）
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @param int $step 步长
|
|
* @return false|int
|
|
*/
|
|
public function dec($name, $step = 1)
|
|
{
|
|
if ($this->has($name)) {
|
|
$value = $this->get($name) - $step;
|
|
$expire = $this->expire;
|
|
} else {
|
|
$value = -$step;
|
|
$expire = 0;
|
|
}
|
|
|
|
return $this->set($name, $value, $expire) ? $value : false;
|
|
}
|
|
|
|
/**
|
|
* 删除缓存
|
|
* @access public
|
|
* @param string $name 缓存变量名
|
|
* @return boolean
|
|
*/
|
|
public function rm($name)
|
|
{
|
|
$filename = $this->getCacheKey($name);
|
|
try {
|
|
return $this->unlink($filename);
|
|
} catch (\Exception $e) {
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 清除缓存
|
|
* @access public
|
|
* @param string $tag 标签名
|
|
* @return boolean
|
|
*/
|
|
public function clear($tag = null)
|
|
{
|
|
if ($tag) {
|
|
// 指定标签清除
|
|
$keys = $this->getTagItem($tag);
|
|
foreach ($keys as $key) {
|
|
$this->unlink($key);
|
|
}
|
|
$this->rm('tag_' . md5($tag));
|
|
return true;
|
|
}
|
|
$files = (array) glob($this->options['path'] . ($this->options['prefix'] ? $this->options['prefix'] . DS : '') . '*');
|
|
foreach ($files as $path) {
|
|
if (is_dir($path)) {
|
|
$matches = glob($path . '/*.php');
|
|
if (is_array($matches)) {
|
|
array_map('unlink', $matches);
|
|
}
|
|
rmdir($path);
|
|
} else {
|
|
unlink($path);
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 判断文件是否存在后，删除
|
|
* @param $path
|
|
* @return bool
|
|
* @author byron sampson <xiaobo.sun@qq.com>
|
|
* @return boolean
|
|
*/
|
|
private function unlink($path)
|
|
{
|
|
return is_file($path) && unlink($path);
|
|
}
|
|
|
|
}
|