/system/database/drivers/mysql/mysql_driver.php
PHP | 532 lines | 214 code | 80 blank | 238 comment | 22 complexity | e349e9242e89ce8d36563e883eea5c33 MD5 | raw file
- <?php
- /**
- * 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 - 2013, 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
- */
- defined('BASEPATH') OR exit('No direct script access allowed');
- /**
- * MySQL 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_mysql_driver extends CI_DB {
- /**
- * Database driver
- *
- * @var string
- */
- public $dbdriver = 'mysql';
- /**
- * Compression flag
- *
- * @var bool
- */
- public $compress = FALSE;
- /**
- * DELETE hack flag
- *
- * 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.
- *
- * @var bool
- */
- public $delete_hack = TRUE;
- // --------------------------------------------------------------------
- /**
- * Identifier escape character
- *
- * @var string
- */
- protected $_escape_char = '`';
- // --------------------------------------------------------------------
- /**
- * Class constructor
- *
- * @param array $params
- * @return void
- */
- public function __construct($params)
- {
- parent::__construct($params);
- if ( ! empty($this->port))
- {
- $this->hostname .= ':'.$this->port;
- }
- }
- // --------------------------------------------------------------------
- /**
- * Non-persistent database connection
- *
- * @param bool $persistent
- * @return resource
- */
- public function db_connect($persistent = FALSE)
- {
- $client_flags = ($this->compress === FALSE) ? 0 : MYSQL_CLIENT_COMPRESS;
- if ($this->encrypt === TRUE)
- {
- $client_flags = $client_flags | MYSQL_CLIENT_SSL;
- }
- $this->conn_id = ($persistent === TRUE)
- ? @mysql_pconnect($this->hostname, $this->username, $this->password, $client_flags)
- : @mysql_connect($this->hostname, $this->username, $this->password, TRUE, $client_flags);
- // ----------------------------------------------------------------
- // Select the DB... assuming a database name is specified in the config file
- if ($this->database !== '' && ! $this->db_select())
- {
- log_message('error', 'Unable to select database: '.$this->database);
- return ($this->db_debug === TRUE)
- ? $this->display_error('db_unable_to_select', $this->database)
- : FALSE;
- }
- return $this->conn_id;
- }
- // --------------------------------------------------------------------
- /**
- * Persistent database connection
- *
- * @return resource
- */
- public function db_pconnect()
- {
- return $this->db_connect(TRUE);
- }
- // --------------------------------------------------------------------
- /**
- * 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 (mysql_ping($this->conn_id) === FALSE)
- {
- $this->conn_id = FALSE;
- }
- }
- // --------------------------------------------------------------------
- /**
- * Select the database
- *
- * @param string $database
- * @return bool
- */
- public function db_select($database = '')
- {
- if ($database === '')
- {
- $database = $this->database;
- }
- if (@mysql_select_db($database, $this->conn_id))
- {
- $this->database = $database;
- return TRUE;
- }
- return FALSE;
- }
- // --------------------------------------------------------------------
- /**
- * Set client character set
- *
- * @param string $charset
- * @return bool
- */
- protected function _db_set_charset($charset)
- {
- return @mysql_set_charset($charset, $this->conn_id);
- }
- // --------------------------------------------------------------------
- /**
- * Database version number
- *
- * @return string
- */
- public function version()
- {
- if (isset($this->data_cache['version']))
- {
- return $this->data_cache['version'];
- }
- elseif ( ! $this->conn_id)
- {
- $this->initialize();
- }
- if ( ! $this->conn_id OR ($version = @mysql_get_server_info($this->conn_id)) === FALSE)
- {
- return FALSE;
- }
- return $this->data_cache['version'] = $version;
- }
- // --------------------------------------------------------------------
- /**
- * Execute the query
- *
- * @param string $sql an SQL query
- * @return mixed
- */
- protected function _execute($sql)
- {
- return @mysql_query($this->_prep_query($sql), $this->conn_id);
- }
- // --------------------------------------------------------------------
- /**
- * Prep the query
- *
- * If needed, each database adapter can prep the query string
- *
- * @param string $sql an SQL query
- * @return string
- */
- protected function _prep_query($sql)
- {
- // mysql_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 trim($sql).' WHERE 1=1';
- }
- return $sql;
- }
- // --------------------------------------------------------------------
- /**
- * Begin Transaction
- *
- * @param bool $test_mode
- * @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;
- }
- // --------------------------------------------------------------------
- /**
- * Platform-dependant string escape
- *
- * @param string
- * @return string
- */
- protected function _escape_str($str)
- {
- return is_resource($this->conn_id)
- ? mysql_real_escape_string($str, $this->conn_id)
- : addslashes($str);
- }
- // --------------------------------------------------------------------
- /**
- * Affected Rows
- *
- * @return int
- */
- public function affected_rows()
- {
- return @mysql_affected_rows($this->conn_id);
- }
- // --------------------------------------------------------------------
- /**
- * Insert ID
- *
- * @return int
- */
- public function insert_id()
- {
- return @mysql_insert_id($this->conn_id);
- }
- // --------------------------------------------------------------------
- /**
- * List table query
- *
- * Generates a platform-specific query string so that the table names can be fetched
- *
- * @param bool $prefix_limit
- * @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 $table
- * @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 $table
- * @return array
- */
- public function field_data($table = '')
- {
- if ($table === '')
- {
- return ($this->db_debug) ? $this->display_error('db_field_param_missing') : FALSE;
- }
- if (($query = $this->query('SHOW COLUMNS FROM '.$this->protect_identifiers($table, TRUE, NULL, FALSE))) === FALSE)
- {
- return FALSE;
- }
- $query = $query->result_object();
- $retval = array();
- for ($i = 0, $c = count($query); $i < $c; $i++)
- {
- $retval[$i] = new stdClass();
- $retval[$i]->name = $query[$i]->Field;
- sscanf($query[$i]->Type, '%[a-z](%d)',
- $retval[$i]->type,
- $retval[$i]->max_length
- );
- $retval[$i]->default = $query[$i]->Default;
- $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()
- {
- return array('code' => mysql_errno($this->conn_id), 'message' => mysql_error($this->conn_id));
- }
- // --------------------------------------------------------------------
- /**
- * Update_Batch statement
- *
- * Generates a platform-specific batch update string from the supplied data
- *
- * @param string $table Table name
- * @param array $values Update data
- * @param string $index WHERE key
- * @return string
- */
- protected function _update_batch($table, $values, $index)
- {
- $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, ';
- }
- $this->where($index.' IN('.implode(',', $ids).')', NULL, FALSE);
- return 'UPDATE '.$table.' SET '.substr($cases, 0, -2).$this->_compile_wh('qb_where');
- }
- // --------------------------------------------------------------------
- /**
- * 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()
- {
- @mysql_close($this->conn_id);
- }
- }
- /* End of file mysql_driver.php */
- /* Location: ./system/database/drivers/mysql/mysql_driver.php */