Query/core/abstract/abstract_driver.php

509 lines
12 KiB
PHP
Raw Normal View History

2012-03-15 09:25:18 -04:00
<?php
/**
* Query
*
* Free Query Builder / Database Abstraction Layer
*
2012-04-20 13:17:39 -04:00
* @package Query
* @author Timothy J. Warren
2014-01-02 12:36:50 -05:00
* @copyright Copyright (c) 2012 - 2014
2012-03-15 09:25:18 -04:00
* @link https://github.com/aviat4ion/Query
2012-04-20 13:17:39 -04:00
* @license http://philsturgeon.co.uk/code/dbad-license
2012-03-15 09:25:18 -04:00
*/
// --------------------------------------------------------------------------
2014-04-02 17:08:50 -04:00
namespace Query\Driver;
2012-03-15 09:25:18 -04:00
/**
* Base Database class
*
* Extends PDO to simplify cross-database issues
2012-04-20 13:17:39 -04:00
*
* @package Query
2014-03-31 16:01:58 -04:00
* @subpackage Drivers
2012-03-15 09:25:18 -04:00
*/
2014-04-02 17:08:50 -04:00
abstract class Abstract_Driver extends \PDO implements Driver_Interface {
2012-03-15 09:25:18 -04:00
2014-03-31 16:01:58 -04:00
/**
* Reference to the last executed query
* @var PDOStatement
*/
2012-03-15 09:25:18 -04:00
protected $statement;
2012-08-09 12:15:36 -04:00
2014-03-31 16:01:58 -04:00
/**
* Character to escape indentifiers
* @var string
*/
2012-04-10 14:06:34 -04:00
protected $escape_char = '"';
2012-08-09 12:15:36 -04:00
2014-03-31 16:01:58 -04:00
/**
* Reference to sql class
* @var SQL_Interface
*/
public $sql;
2012-08-09 12:15:36 -04:00
2014-03-31 16:01:58 -04:00
/**
* Reference to util class
* @var DB_Util
*/
public $util;
2012-03-15 09:25:18 -04:00
2014-03-31 16:01:58 -04:00
/**
* Last query executed
* @var string
*/
2012-09-27 13:41:29 -04:00
public $last_query;
2013-02-28 12:22:55 -05:00
2014-03-31 16:01:58 -04:00
/**
* Prefix to apply to table names
* @var string
*/
2012-11-07 08:42:34 -05:00
public $table_prefix = '';
2012-09-27 13:41:29 -04:00
2012-03-15 09:25:18 -04:00
/**
* PDO constructor wrapper
*
* @param string $dsn
* @param string $username
* @param string $password
* @param array $driver_options
2012-03-15 09:25:18 -04:00
*/
public function __construct($dsn, $username=NULL, $password=NULL, array $driver_options=array())
2012-03-15 09:25:18 -04:00
{
// Set PDO to display errors as exceptions, and apply driver options
2014-04-02 17:08:50 -04:00
$driver_options[\PDO::ATTR_ERRMODE] = \PDO::ERRMODE_EXCEPTION;
2012-03-15 09:25:18 -04:00
parent::__construct($dsn, $username, $password, $driver_options);
2012-08-09 12:15:36 -04:00
// Load the sql and util class for the driver
foreach(array('sql', 'util') as $sub)
{
$class = get_class($this) . "_{$sub}";
$this->$sub = new $class($this);
}
2012-03-15 09:25:18 -04:00
}
2012-04-10 14:06:34 -04:00
// --------------------------------------------------------------------------
// ! Concrete functions that can be overridden in child classes
// --------------------------------------------------------------------------
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
/**
* Simplifies prepared statements for database queries
*
* @param string $sql
* @param array $data
* @return PDOStatement PDOStatement / FALSE
2014-02-11 14:29:41 -05:00
* @throws InvalidArgumentException
2012-03-15 09:25:18 -04:00
*/
public function prepare_query($sql, $data)
{
// Prepare the sql, save the statement for easy access later
$this->statement = $this->prepare($sql);
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
if( ! (is_array($data) || is_object($data)))
{
2014-04-02 17:08:50 -04:00
throw new \InvalidArgumentException("Invalid data argument");
2012-03-15 09:25:18 -04:00
}
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
// Bind the parameters
foreach($data as $k => $value)
2012-03-15 09:25:18 -04:00
{
// Parameters are 1-based, the data is 0-based
// So, if the key is numeric, add 1
if(is_numeric($k)) $k++;
$this->statement->bindValue($k, $value);
2012-03-15 09:25:18 -04:00
}
2012-04-10 14:06:34 -04:00
return $this->statement;
2012-03-15 09:25:18 -04:00
}
// -------------------------------------------------------------------------
/**
* Create and execute a prepared statement with the provided parameters
*
* @param string $sql
* @param array $params
* @return PDOStatement
*/
public function prepare_execute($sql, $params)
2012-04-10 14:06:34 -04:00
{
2012-03-15 09:25:18 -04:00
$this->statement = $this->prepare_query($sql, $params);
$this->statement->execute();
return $this->statement;
}
// -------------------------------------------------------------------------
/**
* Returns number of rows affected by an INSERT, UPDATE, DELETE type query
*
* @return int
*/
public function affected_rows()
2012-03-15 09:25:18 -04:00
{
// Return number of rows affected
return $this->statement->rowCount();
}
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-04-10 14:06:34 -04:00
2012-11-07 08:42:34 -05:00
/**
* Quote database table name, and set prefix
2013-02-28 12:22:55 -05:00
*
* @param string $table
2012-11-07 08:42:34 -05:00
* @return string
*/
public function quote_table($table)
2013-02-28 12:22:55 -05:00
{
// Add the prefix to the table name
// before quoting it
if ( ! empty($this->table_prefix))
2012-11-07 08:42:34 -05:00
{
// Split indentifier by period, will split into:
// database.schema.table OR
// schema.table OR
// database.table OR
// table
$idents = explode('.', $table);
$segments = count($idents);
// Quote the last item, and add the database prefix
$idents[$segments - 1] = $this->_prefix(end($idents));
// Rejoin
$table = implode('.', $idents);
2012-11-07 08:42:34 -05:00
}
2013-02-28 12:22:55 -05:00
2012-11-07 08:42:34 -05:00
// Finally, quote the table
return $this->quote_ident($table);
}
2013-02-28 12:22:55 -05:00
2012-11-07 08:42:34 -05:00
// --------------------------------------------------------------------------
2013-02-28 12:22:55 -05:00
/**
2012-11-07 08:42:34 -05:00
* Sets the table prefix on the passed string
2013-02-28 12:22:55 -05:00
*
2012-11-07 08:42:34 -05:00
* @param string $str
* @return string
*/
protected function _prefix($str)
{
// Don't prefix an already prefixed table
if (strpos($str, $this->table_prefix) !== FALSE)
{
return $str;
}
2013-02-28 12:22:55 -05:00
2012-11-07 08:42:34 -05:00
return $this->table_prefix.$str;
}
2013-02-28 12:22:55 -05:00
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
/**
* Surrounds the string with the databases identifier escape characters
*
* @param mixed $ident
* @return string
*/
public function quote_ident($ident)
{
2012-03-15 09:25:18 -04:00
if (is_array($ident))
{
2012-11-07 08:42:34 -05:00
return array_map(array($this, __METHOD__), $ident);
2012-03-15 09:25:18 -04:00
}
2012-08-09 12:15:36 -04:00
// Handle comma-separated identifiers
if (strpos($ident, ',') !== FALSE)
{
$parts = explode(',', $ident);
$parts = array_map('mb_trim', $parts);
2012-11-07 08:42:34 -05:00
$parts = array_map(array($this, __METHOD__), $parts);
$ident = implode(',', $parts);
}
2012-03-15 09:25:18 -04:00
// Split each identifier by the period
$hiers = explode('.', $ident);
$hiers = array_map('mb_trim', $hiers);
2012-03-15 09:25:18 -04:00
// Re-compile the string
$raw = implode('.', array_map(array($this, '_quote'), $hiers));
// Fix functions
$funcs = array();
preg_match_all("#{$this->escape_char}([a-zA-Z0-9_]+(\((.*?)\))){$this->escape_char}#iu", $raw, $funcs, PREG_SET_ORDER);
foreach($funcs as $f)
{
// Unquote the function
$raw = str_replace($f[0], $f[1], $raw);
// Quote the inside identifiers
$raw = str_replace($f[3], $this->quote_ident($f[3]), $raw);
}
return $raw;
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Helper method for quote_ident
*
* @param mixed $str
* @return mixed
*/
2014-02-14 22:08:19 -05:00
public function _quote($str)
{
// Check that the current value is a string,
// and is not already quoted before quoting
// that value, otherwise, return the original value
return (
strpos($str, $this->escape_char) !== 0
&& strrpos($str, $this->escape_char) !== 0
&& is_string($str)
&& ! is_numeric($str)
2014-02-11 14:29:41 -05:00
)
? "{$this->escape_char}{$str}{$this->escape_char}"
: $str;
2012-03-15 09:25:18 -04:00
}
// -------------------------------------------------------------------------
/**
2012-04-10 14:06:34 -04:00
* Return schemas for databases that list them
*
* @return array
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
public function get_schemas()
{
return NULL;
2012-04-10 14:06:34 -04:00
}
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Return list of tables for the current database
2012-04-10 14:06:34 -04:00
*
2012-03-15 09:25:18 -04:00
* @return array
*/
2012-04-10 14:06:34 -04:00
public function get_tables()
{
return $this->driver_query('table_list');
2012-04-10 14:06:34 -04:00
}
// -------------------------------------------------------------------------
2012-04-03 16:54:33 -04:00
/**
* Return list of dbs for the current connection, if possible
*
* @return array
*/
2012-04-10 14:06:34 -04:00
public function get_dbs()
{
return $this->driver_query('db_list');
2012-04-10 14:06:34 -04:00
}
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Return list of views for the current database
*
* @return array
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
public function get_views()
{
return $this->driver_query('view_list');
2012-04-10 14:06:34 -04:00
}
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Return list of sequences for the current database, if they exist
*
* @return array
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
public function get_sequences()
{
return $this->driver_query('sequence_list');
2012-04-10 14:06:34 -04:00
}
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Return list of function for the current database
*
2012-03-15 09:25:18 -04:00
* @return array
*/
2012-04-10 14:06:34 -04:00
public function get_functions()
{
return $this->driver_query('function_list', FALSE);
2012-04-10 14:06:34 -04:00
}
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Return list of stored procedures for the current database
2012-03-15 09:25:18 -04:00
*
2012-04-10 14:06:34 -04:00
* @return array
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
public function get_procedures()
{
return $this->driver_query('procedure_list', FALSE);
2012-04-10 14:06:34 -04:00
}
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Return list of triggers for the current database
2012-03-15 09:25:18 -04:00
*
2012-04-10 14:06:34 -04:00
* @return array
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
public function get_triggers()
{
return $this->driver_query('trigger_list', FALSE);
2012-04-10 14:06:34 -04:00
}
2012-03-15 09:25:18 -04:00
2012-04-10 14:06:34 -04:00
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Retreives an array of non-user-created tables for
* the connection/database
*
* @return array
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
public function get_system_tables()
{
return $this->driver_query('system_table_list');
2012-04-10 14:06:34 -04:00
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Retrieve column information for the current database table
*
* @param string $table
* @return array
*/
public function get_columns($table)
{
return $this->driver_query($this->sql->column_list($table), FALSE);
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Retrieve list of data types for the database
*
* @return array
*/
public function get_types()
{
return $this->driver_query('type_list', FALSE);
}
2012-04-10 14:06:34 -04:00
// -------------------------------------------------------------------------
/**
* Method to simplify retreiving db results for meta-data queries
*
* @param string|array|null $query
* @param bool $filtered_index
* @return array
*/
public function driver_query($query, $filtered_index=TRUE)
{
// Call the appropriate method, if it exists
if (method_exists($this->sql, $query))
{
$query = $this->sql->$query();
}
// Return if the values are returned instead of a query,
// or if the query doesn't apply to the driver
if ( ! is_string($query)) return $query;
// Run the query!
$res = $this->query($query);
2014-04-02 17:08:50 -04:00
$flag = ($filtered_index) ? \PDO::FETCH_NUM : \PDO::FETCH_ASSOC;
$all = $res->fetchAll($flag);
2014-04-02 17:08:50 -04:00
return ($filtered_index) ? \db_filter($all, 0) : $all;
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
/**
* Return the number of rows returned for a SELECT query
*
* @see http://us3.php.net/manual/en/pdostatement.rowcount.php#87110
* @return int
*/
public function num_rows()
{
2012-09-27 13:41:29 -04:00
$regex = '/^SELECT\s+(?:ALL\s+|DISTINCT\s+)?(?:.*?)\s+FROM\s+(.*)$/i';
$output = array();
if (preg_match($regex, $this->last_query, $output) > 0)
{
2014-02-17 19:31:06 -05:00
$stmt = $this->query("SELECT COUNT(*) FROM {$output[1]}");
return (int) $stmt->fetchColumn();
2012-09-27 13:41:29 -04:00
}
return NULL;
}
2012-04-10 14:06:34 -04:00
// -------------------------------------------------------------------------
// ! Abstract public functions to implement in child classes
2012-04-10 14:06:34 -04:00
// -------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Empty the passed table
*
* @param string $table
* @return void
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
abstract public function truncate($table);
// --------------------------------------------------------------------------
/**
* Create sql for batch insert
*
* @param string $table
* @param array $data
* @return array
*/
public function insert_batch($table, $data=array())
{
if ( ! is_array(current($data))) return NULL;
$vals = array(); // Values for insertion
$table = $this->quote_table($table);
$fields = array_keys(current($data));
$sql = "INSERT INTO {$table} ("
. implode(',', $this->quote_ident($fields))
. ") VALUES ";
// Create the placeholder groups
$params = array_fill(0, count($fields), '?');
$param_string = "(" . implode(',', $params) . ")";
$param_list = array_fill(0, count($data), $param_string);
// For each grouping, add the values
foreach($data as $group)
{
$vals = array_merge($vals, array_values($group));
}
// Add the param groups
$sql .= implode(',', $param_list);
return array($sql, $vals);
}
2012-03-15 09:25:18 -04:00
}
2014-02-11 14:29:41 -05:00
// End of db_pdo.php