/system/database/drivers/mysqli/mysqli_driver.php
PHP | 510 lines | 220 code | 78 blank | 212 comment | 25 complexity | 7d76e959925faaf4618a304c3871a659 MD5 | raw file
- <?php if ( ! defined('BASEPATH')) exit('No direct script access allowed');
- /**
- * CodeIgniter
- *
- * An open source application development framework for PHP 5.2.4 or newer
- *
- * NOTICE OF LICENSE
- *
- * Licensed under the Open Software License version 3.0
- *
- * This source file is subject to the Open Software License (OSL 3.0) that is
- * bundled with this package in the files license.txt / license.rst. It is
- * also available through the world wide web at this URL:
- * http://opensource.org/licenses/OSL-3.0
- * If you did not receive a copy of the license and are unable to obtain it
- * through the world wide web, please send an email to
- * licensing@ellislab.com so we can send you a copy immediately.
- *
- * @package CodeIgniter
- * @author EllisLab Dev Team
- * @copyright Copyright (c) 2008 - 2012, EllisLab, Inc. (http://ellislab.com/)
- * @license http://opensource.org/licenses/OSL-3.0 Open Software License (OSL 3.0)
- * @link http://codeigniter.com
- * @since Version 1.0
- * @filesource
- */
- /**
- * MySQLi Database Adapter Class
- *
- * Note: _DB is an extender class that the app controller
- * creates dynamically based on whether the query builder
- * class is being used or not.
- *
- * @package CodeIgniter
- * @subpackage Drivers
- * @category Database
- * @author EllisLab Dev Team
- * @link http://codeigniter.com/user_guide/database/
- */
- class CI_DB_mysqli_driver extends CI_DB {
- public $dbdriver = 'mysqli';
- // The character used for escaping
- protected $_escape_char = '`';
- protected $_random_keyword = ' RAND()'; // database specific random keyword
- /**
- * Whether to use the MySQL "delete hack" which allows the number
- * of affected rows to be shown. Uses a preg_replace when enabled,
- * adding a bit more processing to all queries.
- */
- public $delete_hack = TRUE;
- /**
- * Non-persistent database connection
- *
- * @return object
- */
- public function db_connect()
- {
- // Use MySQL client compression?
- if ($this->compress === TRUE)
- {
- $port = empty($this->port) ? NULL : $this->port;
- $mysqli = new mysqli();
- @$mysqli->real_connect($this->hostname, $this->username, $this->password, $this->database, $port, NULL, MYSQLI_CLIENT_COMPRESS);
- return $mysqli;
- }
- return empty($this->port)
- ? @new mysqli($this->hostname, $this->username, $this->password, $this->database)
- : @new mysqli($this->hostname, $this->username, $this->password, $this->database, $this->port);
- }
- // --------------------------------------------------------------------
- /**
- * Persistent database connection
- *
- * @return object
- */
- public function db_pconnect()
- {
- // Persistent connection support was added in PHP 5.3.0
- if ( ! is_php('5.3'))
- {
- return $this->db_connect();
- }
- // Use MySQL client compression?
- if ($this->compress === TRUE)
- {
- $port = empty($this->port) ? NULL : $this->port;
- $mysqli = mysqli_init();
- $mysqli->real_connect('p:'.$this->hostname, $this->username, $this->password, $this->database, $port, NULL, MYSQLI_CLIENT_COMPRESS);
- return $mysqli;
- }
- return empty($this->port)
- ? @new mysqli('p:'.$this->hostname, $this->username, $this->password, $this->database)
- : @new mysqli('p:'.$this->hostname, $this->username, $this->password, $this->database, $this->port);
- }
- // --------------------------------------------------------------------
- /**
- * Reconnect
- *
- * Keep / reestablish the db connection if no queries have been
- * sent for a length of time exceeding the server's idle timeout
- *
- * @return void
- */
- public function reconnect()
- {
- if ($this->conn_id !== FALSE && $this->conn_id->ping() === FALSE)
- {
- $this->conn_id = FALSE;
- }
- }
- // --------------------------------------------------------------------
- /**
- * Select the database
- *
- * @param string database name
- * @return bool
- */
- public function db_select($database = '')
- {
- if ($database === '')
- {
- $database = $this->database;
- }
- if (@$this->conn_id->select_db($database))
- {
- $this->database = $database;
- return TRUE;
- }
- return FALSE;
- }
- // --------------------------------------------------------------------
- /**
- * Set client character set
- *
- * @param string
- * @return bool
- */
- protected function _db_set_charset($charset)
- {
- return @$this->conn_id->set_charset($charset);
- }
- // --------------------------------------------------------------------
- /**
- * Database version number
- *
- * @return string
- */
- public function version()
- {
- return isset($this->data_cache['version'])
- ? $this->data_cache['version']
- : $this->data_cache['version'] = $this->conn_id->server_info;
- }
- // --------------------------------------------------------------------
- /**
- * Execute the query
- *
- * @param string an SQL query
- * @return mixed
- */
- protected function _execute($sql)
- {
- return @$this->conn_id->query($this->_prep_query($sql));
- }
- // --------------------------------------------------------------------
- /**
- * Prep the query
- *
- * If needed, each database adapter can prep the query string
- *
- * @param string an SQL query
- * @return string
- */
- protected function _prep_query($sql)
- {
- // mysqli_affected_rows() returns 0 for "DELETE FROM TABLE" queries. This hack
- // modifies the query so that it a proper number of affected rows is returned.
- if ($this->delete_hack === TRUE && preg_match('/^\s*DELETE\s+FROM\s+(\S+)\s*$/i', $sql))
- {
- return preg_replace('/^\s*DELETE\s+FROM\s+(\S+)\s*$/', 'DELETE FROM \\1 WHERE 1=1', $sql);
- }
- return $sql;
- }
- // --------------------------------------------------------------------
- /**
- * Begin Transaction
- *
- * @return bool
- */
- public function trans_begin($test_mode = FALSE)
- {
- // When transactions are nested we only begin/commit/rollback the outermost ones
- if ( ! $this->trans_enabled OR $this->_trans_depth > 0)
- {
- return TRUE;
- }
- // Reset the transaction failure flag.
- // If the $test_mode flag is set to TRUE transactions will be rolled back
- // even if the queries produce a successful result.
- $this->_trans_failure = ($test_mode === TRUE);
- $this->simple_query('SET AUTOCOMMIT=0');
- $this->simple_query('START TRANSACTION'); // can also be BEGIN or BEGIN WORK
- return TRUE;
- }
- // --------------------------------------------------------------------
- /**
- * Commit Transaction
- *
- * @return bool
- */
- public function trans_commit()
- {
- // When transactions are nested we only begin/commit/rollback the outermost ones
- if ( ! $this->trans_enabled OR $this->_trans_depth > 0)
- {
- return TRUE;
- }
- $this->simple_query('COMMIT');
- $this->simple_query('SET AUTOCOMMIT=1');
- return TRUE;
- }
- // --------------------------------------------------------------------
- /**
- * Rollback Transaction
- *
- * @return bool
- */
- public function trans_rollback()
- {
- // When transactions are nested we only begin/commit/rollback the outermost ones
- if ( ! $this->trans_enabled OR $this->_trans_depth > 0)
- {
- return TRUE;
- }
- $this->simple_query('ROLLBACK');
- $this->simple_query('SET AUTOCOMMIT=1');
- return TRUE;
- }
- // --------------------------------------------------------------------
- /**
- * Escape String
- *
- * @param string
- * @param bool whether or not the string will be used in a LIKE condition
- * @return string
- */
- public function escape_str($str, $like = FALSE)
- {
- if (is_array($str))
- {
- foreach ($str as $key => $val)
- {
- $str[$key] = $this->escape_str($val, $like);
- }
- return $str;
- }
- $str = is_object($this->conn_id) ? $this->conn_id->real_escape_string($str) : addslashes($str);
- // escape LIKE condition wildcards
- if ($like === TRUE)
- {
- return str_replace(array($this->_like_escape_chr, '%', '_'),
- array($this->_like_escape_chr.$this->_like_escape_chr, $this->_like_escape_chr.'%', $this->_like_escape_chr.'_'),
- $str);
- }
- return $str;
- }
- // --------------------------------------------------------------------
- /**
- * Affected Rows
- *
- * @return int
- */
- public function affected_rows()
- {
- return $this->conn_id->affected_rows;
- }
- // --------------------------------------------------------------------
- /**
- * Insert ID
- *
- * @return int
- */
- public function insert_id()
- {
- return $this->conn_id->insert_id;
- }
- // --------------------------------------------------------------------
- /**
- * List table query
- *
- * Generates a platform-specific query string so that the table names can be fetched
- *
- * @param bool
- * @return string
- */
- protected function _list_tables($prefix_limit = FALSE)
- {
- $sql = 'SHOW TABLES FROM '.$this->escape_identifiers($this->database);
- if ($prefix_limit !== FALSE && $this->dbprefix !== '')
- {
- return $sql." LIKE '".$this->escape_like_str($this->dbprefix)."%'";
- }
- return $sql;
- }
- // --------------------------------------------------------------------
- /**
- * Show column query
- *
- * Generates a platform-specific query string so that the column names can be fetched
- *
- * @param string the table name
- * @return string
- */
- protected function _list_columns($table = '')
- {
- return 'SHOW COLUMNS FROM '.$this->protect_identifiers($table, TRUE, NULL, FALSE);
- }
- // --------------------------------------------------------------------
- /**
- * Returns an object with field data
- *
- * @param string the table name
- * @return object
- */
- public function field_data($table = '')
- {
- if ($table === '')
- {
- return ($this->db_debug) ? $this->display_error('db_field_param_missing') : FALSE;
- }
- $query = $this->query('DESCRIBE '.$this->protect_identifiers($table, TRUE, NULL, FALSE));
- $query = $query->result_object();
- $retval = array();
- for ($i = 0, $c = count($query); $i < $c; $i++)
- {
- preg_match('/([a-z]+)(\(\d+\))?/', $query[$i]->Type, $matches);
- $retval[$i] = new stdClass();
- $retval[$i]->name = $query[$i]->Field;
- $retval[$i]->type = empty($matches[1]) ? NULL : $matches[1];
- $retval[$i]->default = $query[$i]->Default;
- $retval[$i]->max_length = empty($matches[2]) ? NULL : preg_replace('/[^\d]/', '', $matches[2]);
- $retval[$i]->primary_key = (int) ($query[$i]->Key === 'PRI');
- }
- return $retval;
- }
- // --------------------------------------------------------------------
- /**
- * Error
- *
- * Returns an array containing code and message of the last
- * database error that has occured.
- *
- * @return array
- */
- public function error()
- {
- if ( ! empty($this->conn_id->connect_errno))
- {
- return array(
- 'code' => $this->conn_id->connect_errno,
- 'message' => is_php('5.2.9') ? $this->conn_id->connect_error : mysqli_connect_error()
- );
- }
- return array('code' => $this->conn_id->errno, 'message' => $this->conn_id->error);
- }
- // --------------------------------------------------------------------
- /**
- * Update_Batch statement
- *
- * Generates a platform-specific batch update string from the supplied data
- *
- * @param string the table name
- * @param array the update data
- * @param array the where clause
- * @return string
- */
- protected function _update_batch($table, $values, $index, $where = NULL)
- {
- $ids = array();
- foreach ($values as $key => $val)
- {
- $ids[] = $val[$index];
- foreach (array_keys($val) as $field)
- {
- if ($field !== $index)
- {
- $final[$field][] = 'WHEN '.$index.' = '.$val[$index].' THEN '.$val[$field];
- }
- }
- }
- $cases = '';
- foreach ($final as $k => $v)
- {
- $cases .= $k.' = CASE '."\n"
- .implode("\n", $v)."\n"
- .'ELSE '.$k.' END, ';
- }
- $where = ($where !== '' && count($where) > 0) ? implode(' ', $where).' AND ' : '';
- return 'UPDATE '.$table.' SET '.substr($cases, 0, -2)
- .' WHERE '.(($where !== '' && count($where) > 0) ? implode(' ', $where).' AND ' : '')
- .$index.' IN('.implode(',', $ids).')';
- }
- // --------------------------------------------------------------------
- /**
- * FROM tables
- *
- * Groups tables in FROM clauses if needed, so there is no confusion
- * about operator precedence.
- *
- * @return string
- */
- protected function _from_tables()
- {
- if ( ! empty($this->qb_join) && count($this->qb_from) > 1)
- {
- return '('.implode(', ', $this->qb_from).')';
- }
- return implode(', ', $this->qb_from);
- }
- // --------------------------------------------------------------------
- /**
- * Close DB Connection
- *
- * @return void
- */
- protected function _close()
- {
- $this->conn_id->close();
- }
- }
- /* End of file mysqli_driver.php */
- /* Location: ./system/database/drivers/mysqli/mysqli_driver.php */