Mercurial > hg > rc2

comparison program/lib/Roundcube/rcube_db_mysql.php @ 0:4681f974d28b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
vanilla 1.3.3 distro, I hope
author Charlie Root
date Thu, 04 Jan 2018 15:52:31 -0500
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:4681f974d28b
1 <?php
2
3 /**
4 +-----------------------------------------------------------------------+
5 | This file is part of the Roundcube Webmail client |
6 | Copyright (C) 2005-2012, The Roundcube Dev Team |
7 | |
8 | Licensed under the GNU General Public License version 3 or |
9 | any later version with exceptions for skins & plugins. |
10 | See the README file for a full license statement. |
11 | |
12 | PURPOSE: |
13 | Database wrapper class that implements PHP PDO functions |
14 | for MySQL database |
15 +-----------------------------------------------------------------------+
16 | Author: Aleksander Machniak <alec@alec.pl> |
17 +-----------------------------------------------------------------------+
18 */
19
20 /**
21 * Database independent query interface
22 *
23 * This is a wrapper for the PHP PDO
24 *
25 * @package Framework
26 * @subpackage Database
27 */
28 class rcube_db_mysql extends rcube_db
29 {
30 public $db_provider = 'mysql';
31
32 /**
33 * Object constructor
34 *
35 * @param string $db_dsnw DSN for read/write operations
36 * @param string $db_dsnr Optional DSN for read only operations
37 * @param bool $pconn Enables persistent connections
38 */
39 public function __construct($db_dsnw, $db_dsnr = '', $pconn = false)
40 {
41 parent::__construct($db_dsnw, $db_dsnr, $pconn);
42
43 // SQL identifiers quoting
44 $this->options['identifier_start'] = '`';
45 $this->options['identifier_end'] = '`';
46 }
47
48 /**
49 * Driver-specific configuration of database connection
50 *
51 * @param array $dsn DSN for DB connections
52 * @param PDO $dbh Connection handler
53 */
54 protected function conn_configure($dsn, $dbh)
55 {
56 $dbh->query("SET NAMES 'utf8'");
57 }
58
59 /**
60 * Abstract SQL statement for value concatenation
61 *
62 * @return string SQL statement to be used in query
63 */
64 public function concat(/* col1, col2, ... */)
65 {
66 $args = func_get_args();
67
68 if (is_array($args[0])) {
69 $args = $args[0];
70 }
71
72 return 'CONCAT(' . join(', ', $args) . ')';
73 }
74
75 /**
76 * Returns PDO DSN string from DSN array
77 *
78 * @param array $dsn DSN parameters
79 *
80 * @return string Connection string
81 */
82 protected function dsn_string($dsn)
83 {
84 $params = array();
85 $result = 'mysql:';
86
87 if ($dsn['database']) {
88 $params[] = 'dbname=' . $dsn['database'];
89 }
90
91 if ($dsn['hostspec']) {
92 $params[] = 'host=' . $dsn['hostspec'];
93 }
94
95 if ($dsn['port']) {
96 $params[] = 'port=' . $dsn['port'];
97 }
98
99 if ($dsn['socket']) {
100 $params[] = 'unix_socket=' . $dsn['socket'];
101 }
102
103 $params[] = 'charset=utf8';
104
105 if (!empty($params)) {
106 $result .= implode(';', $params);
107 }
108
109 return $result;
110 }
111
112 /**
113 * Returns driver-specific connection options
114 *
115 * @param array $dsn DSN parameters
116 *
117 * @return array Connection options
118 */
119 protected function dsn_options($dsn)
120 {
121 $result = parent::dsn_options($dsn);
122
123 if (!empty($dsn['key'])) {
124 $result[PDO::MYSQL_ATTR_SSL_KEY] = $dsn['key'];
125 }
126
127 if (!empty($dsn['cipher'])) {
128 $result[PDO::MYSQL_ATTR_SSL_CIPHER] = $dsn['cipher'];
129 }
130
131 if (!empty($dsn['cert'])) {
132 $result[PDO::MYSQL_ATTR_SSL_CERT] = $dsn['cert'];
133 }
134
135 if (!empty($dsn['capath'])) {
136 $result[PDO::MYSQL_ATTR_SSL_CAPATH] = $dsn['capath'];
137 }
138
139 if (!empty($dsn['ca'])) {
140 $result[PDO::MYSQL_ATTR_SSL_CA] = $dsn['ca'];
141 }
142
143 // Always return matching (not affected only) rows count
144 $result[PDO::MYSQL_ATTR_FOUND_ROWS] = true;
145
146 // Enable AUTOCOMMIT mode (#1488902)
147 $result[PDO::ATTR_AUTOCOMMIT] = true;
148
149 return $result;
150 }
151
152 /**
153 * Returns list of tables in a database
154 *
155 * @return array List of all tables of the current database
156 */
157 public function list_tables()
158 {
159 // get tables if not cached
160 if ($this->tables === null) {
161 // first fetch current database name
162 $d = $this->query("SELECT database()");
163 $d = $this->fetch_array($d);
164
165 // get list of tables in current database
166 $q = $this->query("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES"
167 . " WHERE TABLE_SCHEMA = ? AND TABLE_TYPE = 'BASE TABLE'"
168 . " ORDER BY TABLE_NAME", $d ? $d[0] : '');
169
170 $this->tables = $q ? $q->fetchAll(PDO::FETCH_COLUMN, 0) : array();
171 }
172
173 return $this->tables;
174 }
175
176 /**
177 * Get database runtime variables
178 *
179 * @param string $varname Variable name
180 * @param mixed $default Default value if variable is not set
181 *
182 * @return mixed Variable value or default
183 */
184 public function get_variable($varname, $default = null)
185 {
186 if (!isset($this->variables)) {
187 $this->variables = array();
188 }
189
190 if (array_key_exists($varname, $this->variables)) {
191 return $this->variables[$varname];
192 }
193
194 // configured value has higher prio
195 $conf_value = rcube::get_instance()->config->get('db_' . $varname);
196 if ($conf_value !== null) {
197 return $this->variables[$varname] = $conf_value;
198 }
199
200 $result = $this->query('SHOW VARIABLES LIKE ?', $varname);
201
202 while ($row = $this->fetch_array($result)) {
203 $this->variables[$row[0]] = $row[1];
204 }
205
206 // not found, use default
207 if (!isset($this->variables[$varname])) {
208 $this->variables[$varname] = $default;
209 }
210
211 return $this->variables[$varname];
212 }
213
214 /**
215 * Handle DB errors, re-issue the query on deadlock errors from InnoDB row-level locking
216 *
217 * @param string Query that triggered the error
218 * @return mixed Result to be stored and returned
219 */
220 protected function handle_error($query)
221 {
222 $error = $this->dbh->errorInfo();
223
224 // retry after "Deadlock found when trying to get lock" errors
225 $retries = 2;
226 while ($error[1] == 1213 && $retries >= 0) {
227 usleep(50000); // wait 50 ms
228 $result = $this->dbh->query($query);
229 if ($result !== false) {
230 return $result;
231 }
232 $error = $this->dbh->errorInfo();
233 $retries--;
234 }
235
236 return parent::handle_error($query);
237 }
238
239 }