include/DbSimple/Mysql.php

   1 <?php
   2 /**
   3  * DbSimple_Mysql: MySQL database.
   4  * (C) Dk Lab, http://en.dklab.ru
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  * See http://www.gnu.org/copyleft/lesser.html
  11  *
  12  * Placeholders end blobs are emulated.
  13  *
  14  * @author Dmitry Koterov, http://forum.dklab.ru/users/DmitryKoterov/
  15  * @author Konstantin Zhinko, http://forum.dklab.ru/users/KonstantinGinkoTit/
  16  *
  17  * @version 2.x $Id: Mysql.php 247 2008-08-18 21:17:08Z dk $
  18  */
  19 require_once dirname(__FILE__) . '/Generic.php';
  20
  21
  22 /**
  23  * Database class for MySQL.
  24  */
  25 class DbSimple_Mysql extends DbSimple_Generic_Database
  26 {
  27     var $link;
  28
  29     /**
  30      * constructor(string $dsn)
  31      * Connect to MySQL.
  32      */
  33     function DbSimple_Mysql($dsn)
  34     {
  35         $p = DbSimple_Generic::parseDSN($dsn);
  36         if (!is_callable('mysql_connect')) {
  37             return $this->_setLastError("-1", "MySQL extension is not loaded", "mysql_connect");
  38         }
  39         $ok = $this->link = @mysql_connect(
  40             $p['host'] . (empty($p['port'])? "" : ":".$p['port']),
  41             $p['user'],
  42             $p['pass'],
  43             true
  44         );
  45         $this->_resetLastError();
  46         if (!$ok) return $this->_setDbError('mysql_connect()');
  47         $ok = @mysql_select_db(preg_replace('{^/}s', '', $p['path']), $this->link);
  48         if (!$ok) return $this->_setDbError('mysql_select_db()');
  49         if (isset($p["charset"])) {
  50             $this->query('SET NAMES ?', $p["charset"]);
  51         }
  52     }
  53
  54
  55     function _performEscape($s, $isIdent=false)
  56     {
  57         if (!$isIdent) {
  58             return "'" . mysql_real_escape_string($s, $this->link) . "'";
  59         } else {
  60             return "`" . str_replace('`', '``', $s) . "`";
  61         }
  62     }
  63
  64
  65     function _performTransaction($parameters=null)
  66     {
  67         return $this->query('BEGIN');
  68     }
  69
  70
  71     function& _performNewBlob($blobid=null)
  72     {
  73         $obj =& new DbSimple_Mysql_Blob($this, $blobid);
  74         return $obj;
  75     }
  76
  77
  78     function _performGetBlobFieldNames($result)
  79     {
  80         $blobFields = array();
  81         for ($i=mysql_num_fields($result)-1; $i>=0; $i--) {
  82             $type = mysql_field_type($result, $i);
  83             if (strpos($type, "BLOB") !== false) $blobFields[] = mysql_field_name($result, $i);
  84         }
  85         return $blobFields;
  86     }
  87
  88
  89     function _performGetPlaceholderIgnoreRe()
  90     {
  91         return '
  92             "   (?> [^"\\\\]+|\\\\"|\\\\)*    "   |
  93             \'  (?> [^\'\\\\]+|\\\\\'|\\\\)* \'   |
  94             `   (?> [^`]+ | ``)*              `   |   # backticks
  95             /\* .*?                          \*/      # comments
  96         ';
  97     }
  98
  99
 100     function _performCommit()
 101     {
 102         return $this->query('COMMIT');
 103     }
 104
 105
 106     function _performRollback()
 107     {
 108         return $this->query('ROLLBACK');
 109     }
 110
 111
 112     function _performTransformQuery(&$queryMain, $how)
 113     {
 114         // If we also need to calculate total number of found rows...
 115         switch ($how) {
 116             // Prepare total calculation (if possible)
 117             case 'CALC_TOTAL':
 118                 $m = null;
 119                 if (preg_match('/^(\s* SELECT)(.*)/six', $queryMain[0], $m)) {
 120                     if ($this->_calcFoundRowsAvailable()) {
 121                         $queryMain[0] = $m[1] . ' SQL_CALC_FOUND_ROWS' . $m[2];
 122                     }
 123                 }
 124                 return true;
 125
 126             // Perform total calculation.
 127             case 'GET_TOTAL':
 128                 // Built-in calculation available?
 129                 if ($this->_calcFoundRowsAvailable()) {
 130                     $queryMain = array('SELECT FOUND_ROWS()');
 131                 }
 132                 // Else use manual calculation.
 133                 // TODO: GROUP BY ... -> COUNT(DISTINCT ...)
 134                 $re = '/^
 135                     (?> -- [^\r\n]* | \s+)*
 136                     (\s* SELECT \s+)                                      #1
 137                     (.*?)                                                 #2
 138                     (\s+ FROM \s+ .*?)                                    #3
 139                         ((?:\s+ ORDER \s+ BY \s+ .*?)?)                   #4
 140                         ((?:\s+ LIMIT \s+ \S+ \s* (?:, \s* \S+ \s*)? )?)  #5
 141                 $/six';
 142                 $m = null;
 143                 if (preg_match($re, $queryMain[0], $m)) {
 144                     $query[0] = $m[1] . $this->_fieldList2Count($m[2]) . " AS C" . $m[3];
 145                     $skipTail = substr_count($m[4] . $m[5], '?');
 146                     if ($skipTail) array_splice($query, -$skipTail);
 147                 }
 148                 return true;
 149         }
 150
 151         return false;
 152     }
 153
 154
 155     function _performQuery($queryMain)
 156     {
 157         $this->_lastQuery = $queryMain;
 158         $this->_expandPlaceholders($queryMain, false);
 159         $result = @mysql_query($queryMain[0], $this->link);
 160         if ($result === false) return $this->_setDbError($queryMain[0]);
 161         if (!is_resource($result)) {
 162             if (preg_match('/^\s* INSERT \s+/six', $queryMain[0])) {
 163                 // INSERT queries return generated ID.
 164                 return @mysql_insert_id($this->link);
 165             }
 166             // Non-SELECT queries return number of affected rows, SELECT - resource.
 167             return @mysql_affected_rows($this->link);
 168         }
 169         return $result;
 170     }
 171
 172
 173     function _performFetch($result)
 174     {
 175         $row = @mysql_fetch_assoc($result);
 176         if (mysql_error()) return $this->_setDbError($this->_lastQuery);
 177         if ($row === false) return null;
 178         return $row;
 179     }
 180
 181
 182     function _setDbError($query)
 183     {
 184         return $this->_setLastError(mysql_errno($this->link), mysql_error($this->link), $query);
 185     }
 186
 187
 188     function _calcFoundRowsAvailable()
 189     {
 190         $ok = version_compare(mysql_get_server_info($this->link), '4.0') >= 0;
 191         return $ok;
 192     }
 193 }
 194
 195
 196 class DbSimple_Mysql_Blob extends DbSimple_Generic_Blob
 197 {
 198     // MySQL does not support separate BLOB fetching.
 199     var $blobdata = null;
 200     var $curSeek = 0;
 201
 202     function DbSimple_Mysql_Blob(&$database, $blobdata=null)
 203     {
 204         $this->blobdata = $blobdata;
 205         $this->curSeek = 0;
 206     }
 207
 208     function read($len)
 209     {
 210         $p = $this->curSeek;
 211         $this->curSeek = min($this->curSeek + $len, strlen($this->blobdata));
 212         return substr($this->blobdata, $this->curSeek, $len);
 213     }
 214
 215     function write($data)
 216     {
 217         $this->blobdata .= $data;
 218     }
 219
 220     function close()
 221     {
 222         return $this->blobdata;
 223     }
 224
 225     function length()
 226     {
 227         return strlen($this->blobdata);
 228     }
 229 }
 230 ?>