Db.class.php
5.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
<?php
// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2006-2014 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;
defined('THINK_PATH') or exit();
/**
* 数据库方式缓存驱动 CREATE TABLE think_cache ( cachekey varchar(255) NOT NULL, expire int(11) NOT NULL, data blob, datacrc int(32), UNIQUE KEY `cachekey` (`cachekey`) );
*/
class Db extends Cache
{
/**
* 架构函数
*
* @param array $options
* 缓存参数
* @access public
*/
public function __construct($options = array())
{
if (empty($options)) {
$options = array(
'table' => C('DATA_CACHE_TABLE')
);
}
$this->options = $options;
$this->options['prefix'] = isset($options['prefix']) ? $options['prefix'] : C('DATA_CACHE_PREFIX');
$this->options['length'] = isset($options['length']) ? $options['length'] : 0;
$this->options['expire'] = isset($options['expire']) ? $options['expire'] : C('DATA_CACHE_TIME');
$this->handler = \Think\Db::getInstance();
}
/**
* 读取缓存
*
* @access public
* @param string $name
* 缓存变量名
* @return mixed
*/
public function get($name)
{
$name = $this->options['prefix'] . addslashes($name);
N('cache_read', 1);
$result = $this->handler->query('SELECT `data`,`datacrc` FROM `' . $this->options['table'] . '` WHERE `cachekey`=\'' . $name . '\' AND (`expire` =0 OR `expire`>' . time() . ') LIMIT 0,1');
if (false !== $result) {
$result = $result[0];
if (C('DATA_CACHE_CHECK')) { // 开启数据校验
if ($result['datacrc'] != md5($result['data'])) { // 校验错误
return false;
}
}
$content = $result['data'];
if (C('DATA_CACHE_COMPRESS') && function_exists('gzcompress')) {
// 启用数据压缩
$content = gzuncompress($content);
}
$content = unserialize($content);
return $content;
} else {
return false;
}
}
/**
* 写入缓存
*
* @access public
* @param string $name
* 缓存变量名
* @param mixed $value
* 存储数据
* @param integer $expire
* 有效时间（秒）
* @return boolean
*/
public function set($name, $value, $expire = null)
{
$data = serialize($value);
$name = $this->options['prefix'] . addslashes($name);
N('cache_write', 1);
if (C('DATA_CACHE_COMPRESS') && function_exists('gzcompress')) {
// 数据压缩
$data = gzcompress($data, 3);
}
if (C('DATA_CACHE_CHECK')) { // 开启数据校验
$crc = md5($data);
} else {
$crc = '';
}
if (is_null($expire)) {
$expire = $this->options['expire'];
}
$expire = ($expire == 0) ? 0 : (time() + $expire); // 缓存有效期为0表示永久缓存
$result = $this->handler->query('select `cachekey` from `' . $this->options['table'] . '` where `cachekey`=\'' . $name . '\' limit 0,1');
if (! empty($result)) {
// 更新记录
$result = $this->handler->execute('UPDATE ' . $this->options['table'] . ' SET data=\'' . $data . '\' ,datacrc=\'' . $crc . '\',expire=' . $expire . ' WHERE `cachekey`=\'' . $name . '\'');
} else {
// 新增记录
$result = $this->handler->execute('INSERT INTO ' . $this->options['table'] . ' (`cachekey`,`data`,`datacrc`,`expire`) VALUES (\'' . $name . '\',\'' . $data . '\',\'' . $crc . '\',' . $expire . ')');
}
if ($result) {
if ($this->options['length'] > 0) {
// 记录缓存队列
$this->queue($name);
}
return true;
} else {
return false;
}
}
/**
* 删除缓存
*
* @access public
* @param string $name
* 缓存变量名
* @return boolean
*/
public function rm($name)
{
$name = $this->options['prefix'] . addslashes($name);
return $this->handler->execute('DELETE FROM `' . $this->options['table'] . '` WHERE `cachekey`=\'' . $name . '\'');
}
/**
* 清除缓存
*
* @access public
* @return boolean
*/
public function clear()
{
return $this->handler->execute('TRUNCATE TABLE `' . $this->options['table'] . '`');
}
}