Db.class.php
5.89 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
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
<?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\Session\Driver;
/**
* 数据库方式Session驱动
* CREATE TABLE think_session (
* session_id varchar(255) NOT NULL,
* session_expire int(11) NOT NULL,
* session_data blob,
* UNIQUE KEY `session_id` (`session_id`)
* );
*/
class Db
{
/**
* Session有效时间
*/
protected $lifeTime = '';
/**
* session保存的数据库名
*/
protected $sessionTable = '';
/**
* 数据库句柄
*/
protected $hander = array();
/**
* 打开Session
*
* @access public
* @param string $savePath
* @param mixed $sessName
*/
public function open($savePath, $sessName)
{
$this->lifeTime = C('SESSION_EXPIRE') ? C('SESSION_EXPIRE') : ini_get('session.gc_maxlifetime');
$this->sessionTable = C('SESSION_TABLE') ? C('SESSION_TABLE') : C("DB_PREFIX") . "session";
// 分布式数据库
$host = explode(',', C('DB_HOST'));
$port = explode(',', C('DB_PORT'));
$name = explode(',', C('DB_NAME'));
$user = explode(',', C('DB_USER'));
$pwd = explode(',', C('DB_PWD'));
if (1 == C('DB_DEPLOY_TYPE')) {
// 读写分离
if (C('DB_RW_SEPARATE')) {
$w = floor(mt_rand(0, C('DB_MASTER_NUM') - 1));
if (is_numeric(C('DB_SLAVE_NO'))) { // 指定服务器读
$r = C('DB_SLAVE_NO');
} else {
$r = floor(mt_rand(C('DB_MASTER_NUM'), count($host) - 1));
}
// 主数据库链接
$hander = mysql_connect($host[$w] . (isset($port[$w]) ? ':' . $port[$w] : ':' . $port[0]), isset($user[$w]) ? $user[$w] : $user[0], isset($pwd[$w]) ? $pwd[$w] : $pwd[0]);
$dbSel = mysql_select_db(isset($name[$w]) ? $name[$w] : $name[0], $hander);
if (! $hander || ! $dbSel) {
return false;
}
$this->hander[0] = $hander;
// 从数据库链接
$hander = mysql_connect($host[$r] . (isset($port[$r]) ? ':' . $port[$r] : ':' . $port[0]), isset($user[$r]) ? $user[$r] : $user[0], isset($pwd[$r]) ? $pwd[$r] : $pwd[0]);
$dbSel = mysql_select_db(isset($name[$r]) ? $name[$r] : $name[0], $hander);
if (! $hander || ! $dbSel) {
return false;
}
$this->hander[1] = $hander;
return true;
}
}
// 从数据库链接
$r = floor(mt_rand(0, count($host) - 1));
$hander = mysql_connect($host[$r] . (isset($port[$r]) ? ':' . $port[$r] : ':' . $port[0]), isset($user[$r]) ? $user[$r] : $user[0], isset($pwd[$r]) ? $pwd[$r] : $pwd[0]);
$dbSel = mysql_select_db(isset($name[$r]) ? $name[$r] : $name[0], $hander);
if (! $hander || ! $dbSel) {
return false;
}
$this->hander = $hander;
return true;
}
/**
* 关闭Session
*
* @access public
*/
public function close()
{
if (is_array($this->hander)) {
$this->gc($this->lifeTime);
return (mysql_close($this->hander[0]) && mysql_close($this->hander[1]));
}
$this->gc($this->lifeTime);
return mysql_close($this->hander);
}
/**
* 读取Session
*
* @access public
* @param string $sessID
*/
public function read($sessID)
{
$hander = is_array($this->hander) ? $this->hander[1] : $this->hander;
$res = mysql_query("SELECT session_data AS data FROM " . $this->sessionTable . " WHERE session_id = '$sessID' AND session_expire >" . time(), $hander);
if ($res) {
$row = mysql_fetch_assoc($res);
return $row['data'];
}
return "";
}
/**
* 写入Session
*
* @access public
* @param string $sessID
* @param String $sessData
*/
public function write($sessID, $sessData)
{
$hander = is_array($this->hander) ? $this->hander[0] : $this->hander;
$expire = time() + $this->lifeTime;
mysql_query("REPLACE INTO " . $this->sessionTable . " ( session_id, session_expire, session_data) VALUES( '$sessID', '$expire', '$sessData')", $hander);
if (mysql_affected_rows($hander)) {
return true;
}
return false;
}
/**
* 删除Session
*
* @access public
* @param string $sessID
*/
public function destroy($sessID)
{
$hander = is_array($this->hander) ? $this->hander[0] : $this->hander;
mysql_query("DELETE FROM " . $this->sessionTable . " WHERE session_id = '$sessID'", $hander);
if (mysql_affected_rows($hander)) {
return true;
}
return false;
}
/**
* Session 垃圾回收
*
* @access public
* @param string $sessMaxLifeTime
*/
public function gc($sessMaxLifeTime)
{
$hander = is_array($this->hander) ? $this->hander[0] : $this->hander;
mysql_query("DELETE FROM " . $this->sessionTable . " WHERE session_expire < " . time(), $hander);
return mysql_affected_rows($hander);
}
}