2 /***************************************************************************
3 * Copyright (C) 2003-2009 Polytechnique.org *
4 * http://opensource.polytechnique.org/ *
6 * This program is free software; you can redistribute it and/or modify *
7 * it under the terms of the GNU General Public License as published by *
8 * the Free Software Foundation; either version 2 of the License, or *
9 * (at your option) any later version. *
11 * This program is distributed in the hope that it will be useful, *
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
14 * GNU General Public License for more details. *
16 * You should have received a copy of the GNU General Public License *
17 * along with this program; if not, write to the Free Software *
19 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA *
20 **************************************************************************/
24 private static $mysqli = null
;
26 public static function connect()
29 self
::$mysqli = new mysqli($globals->dbhost
, $globals->dbuser
, $globals->dbpwd
, $globals->dbdb
);
30 if ($globals->debug
& DEBUG_BT
) {
31 $bt = new PlBacktrace('MySQL');
32 if (mysqli_connect_errno()) {
33 $bt->newEvent("MySQLI connection", 0, mysqli_connect_error());
37 self
::$mysqli->autocommit(true
);
38 self
::$mysqli->set_charset($globals->dbcharset
);
42 public static function _prepare($args)
44 $query = array_map(Array('XDB', 'escape'), $args);
45 $query[0] = str_replace('{?}', '%s', str_replace('%', '%%', $args[0]));
46 return call_user_func_array('sprintf', $query);
49 public static function _reformatQuery($query)
51 $query = preg_split("/\n\\s*/", trim($query));
53 foreach ($query as $key=>$line) {
55 if (preg_match('/^([A-Z]+(?:\s+(?:JOIN|BY|FROM|INTO))?)\s+(.*)/u', $line, $matches)
56 && $matches[1] != 'AND' && $matches[1] != 'OR')
58 $local = strlen($matches[1]);
59 $line = $matches[1] . ' ' . $matches[2];
60 $length = max($length, $local);
62 $query[$key] = array($line, $local);
65 foreach ($query as $array) {
66 list($line, $local) = $array;
67 $local = max(0, $length - $local);
68 $res .= str_repeat(' ', $local) . $line . "\n";
69 $length +
= 2 * (substr_count($line, '(') - substr_count($line, ')'));
74 public static function _query($query)
78 if (!self
::$mysqli && !self
::connect()) {
79 header($_SERVER['SERVER_PROTOCOL'] . ' 500 Internal Server Error');
80 Platal
::page()->kill('Impossible de se connecter à la base de données.');
84 if ($globals->debug
& DEBUG_BT
) {
86 if (strpos($query, 'FOUND_ROWS()') === false
) {
87 $res = self
::$mysqli->query("EXPLAIN $query");
89 while ($row = $res->fetch_assoc()) {
95 PlBacktrace
::$bt['MySQL']->start(XDB
::_reformatQuery($query));
98 $res = XDB
::$mysqli->query($query);
100 if ($globals->debug
& DEBUG_BT
) {
101 PlBacktrace
::$bt['MySQL']->stop(@$res->num_rows ?
$res->num_rows
: self
::$mysqli->affected_rows
,
102 self
::$mysqli->error
,
106 if ($res === false
) {
107 header($_SERVER['SERVER_PROTOCOL'] . ' 500 Internal Server Error');
108 if (strpos($query, 'INSERT') === false
&& strpos($query, 'UPDATE') === false
109 && strpos($query, 'REPLACE') === false
&& strpos($query, 'DELETE') === false
) {
110 $text = 'Erreur lors de l\'interrogation de la base de données';
112 $text = 'Erreur lors de l\'écriture dans la base de données';
114 if ($globals->debug
) {
115 $text .= '<pre>' . pl_entities(XDB
::_reformatQuery($query)) . '</pre>';
117 $file = fopen($globals->spoolroot
. '/spool/tmp/query_errors', 'a');
118 fwrite($file, '<pre>' . pl_entities(XDB
::_reformatQuery($query)) . '</pre>'
119 . '<pre>' . XDB
::$mysqli->error
. '</pre>' . "\n");
122 Platal
::page()->kill($text);
128 private static function queryv($query)
130 return new XOrgDBResult(self
::_prepare($query));
133 public static function query()
135 return self
::queryv(func_get_args());
138 public static function format()
140 return self
::_prepare(func_get_args());
143 public static function execute()
146 $args = func_get_args();
147 if ($globals->mode
!= 'rw' && !strpos($args[0], 'logger')) {
150 return self
::_query(XDB
::_prepare($args));
153 public static function iterator()
155 return new XOrgDBIterator(self
::_prepare(func_get_args()));
158 public static function iterRow()
160 return new XOrgDBIterator(self
::_prepare(func_get_args()), MYSQL_NUM
);
163 private static function findQuery($params, $default = array())
165 for ($i = 0 ; $i < count($default) ; ++
$i) {
167 foreach (array('insert', 'select', 'replace', 'delete', 'update') as $kwd) {
168 if (stripos($params[0], $kwd) !== false
) {
176 $default[$i] = array_shift($params);
179 return array($default, $params);
182 /** Fetch all rows returned by the given query.
183 * This functions can take 2 optional arguments (cf XOrgDBResult::fetchAllRow()).
184 * Optional arguments are given *before* the query.
186 public static function fetchAllRow()
188 list($args, $query) = self
::findQuery(func_get_args(), array(false
, false
));
189 return self
::queryv($query)->fetchAllRow($args[0], $args[1]);
192 /** Fetch all rows returned by the given query.
193 * This functions can take 2 optional arguments (cf XOrgDBResult::fetchAllAssoc()).
194 * Optional arguments are given *before* the query.
196 public static function fetchAllAssoc()
198 list($args, $query) = self
::findQuery(func_get_args(), array(false
, false
));
199 return self
::queryv($query)->fetchAllAssoc($args[0], $args[1]);
202 public static function fetchOneCell()
204 list($args, $query) = self
::findQuery(func_get_args());
205 return self
::queryv($query)->fetchOneCell();
208 public static function fetchOneRow()
210 list($args, $query) = self
::findQuery(func_get_args());
211 return self
::queryv($query)->fetchOneRow();
214 public static function fetchOneAssoc()
216 list($args, $query) = self
::findQuery(func_get_args());
217 return self
::queryv($query)->fetchOneAssoc();
220 /** Fetch a column from the result of the given query.
221 * This functions can take 1 optional arguments (cf XOrgDBResult::fetchColumn()).
222 * Optional arguments are given *before* the query.
224 public static function fetchColumn()
226 list($args, $query) = self
::findQuery(func_get_args(), array(0));
227 return self
::queryv($query)->fetchColumn();
230 public static function insertId()
232 return self
::$mysqli->insert_id
;
235 public static function errno()
237 return self
::$mysqli->errno
;
240 public static function error()
242 return self
::$mysqli->error
;
245 public static function affectedRows()
247 return self
::$mysqli->affected_rows
;
250 public static function escape($var)
252 switch (gettype($var)) {
262 return "'".addslashes($var)."'";
268 if ($var instanceof PlFlagSet
) {
269 return "'" . addslashes($var->flags()) . "'";
272 return "'".addslashes(serialize($var))."'";
275 die(var_export($var, true
).' is not a valid for a database entry');
285 public function __construct($query)
287 $this->_res
= XDB
::_query($query);
290 public function free()
298 protected function _fetchRow()
300 return $this->_res ?
$this->_res
->fetch_row() : null
;
303 protected function _fetchAssoc()
305 return $this->_res ?
$this->_res
->fetch_assoc() : null
;
308 public function fetchAllRow($id = false
, $keep_array = false
)
314 while (($data = $this->_res
->fetch_row())) {
318 if (!$keep_array && count($data) == 1) {
320 $result[$key] = current($data);
322 $result[$key] = $data;
332 public function fetchAllAssoc($id = false
, $keep_array = false
)
338 while (($data = $this->_res
->fetch_assoc())) {
342 if (!$keep_array && count($data) == 1) {
344 $result[$key] = current($data);
346 $result[$key] = $data;
356 public function fetchOneAssoc()
358 $tmp = $this->_fetchAssoc();
363 public function fetchOneRow()
365 $tmp = $this->_fetchRow();
370 public function fetchOneCell()
372 $tmp = $this->_fetchRow();
377 public function fetchColumn($key = 0)
380 if (is_numeric($key)) {
381 while($tmp = $this->_fetchRow()) {
385 while($tmp = $this->_fetchAssoc()) {
393 public function fetchOneField()
395 return $this->_res ?
$this->_res
->fetch_field() : null
;
398 public function fetchFields()
401 while ($res[] = $this->fetchOneField());
405 public function numRows()
407 return $this->_res ?
$this->_res
->num_rows
: 0;
410 public function fieldCount()
412 return $this->_res ?
$this->_res
->field_count
: 0;
416 require_once dirname(__FILE__
) . '/pliterator.php';
418 class XOrgDBIterator
extends XOrgDBResult
implements PlIterator
425 private $_mode = MYSQL_ASSOC
;
427 public function __construct($query, $mode = MYSQL_ASSOC
)
429 parent
::__construct($query);
431 $this->_total
= $this->numRows();
433 $this->_fields
= $this->fieldCount();
434 $this->_mode
= $mode;
437 public function next()
440 if ($this->_pos
> $this->_total
) {
445 return $this->_mode
!= MYSQL_ASSOC ?
$this->_fetchRow() : $this->_fetchAssoc();
448 public function first()
450 return $this->_pos
== 1;
453 public function last()
455 return $this->_pos
== $this->_total
;
458 public function total()
460 return $this->_total
;
463 public function nextField()
466 if ($this->_fpos
> $this->_fields
) {
469 return $this->fetchOneField();
472 public function firstField()
474 return $this->_fpos
== 1;
477 public function lastField()
479 return $this->_fpos
== $this->_fields
;
482 public function totalFields()
484 return $this->_fields
;
488 // vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8: