Query/db_pdo.php

359 lines
7.6 KiB
PHP
Raw Normal View History

2012-03-15 09:25:18 -04:00
<?php
/**
* Query
*
* Free Query Builder / Database Abstraction Layer
*
* @author Timothy J. Warren
* @copyright Copyright (c) 2012
* @link https://github.com/aviat4ion/Query
* @license http://philsturgeon.co.uk/code/dbad-license
*/
// --------------------------------------------------------------------------
/**
* Base Database class
*
* Extends PDO to simplify cross-database issues
2012-04-10 14:06:34 -04:00
*
* @abstract
2012-03-15 09:25:18 -04:00
*/
abstract class DB_PDO extends PDO {
public $manip;
protected $statement;
2012-04-10 14:06:34 -04:00
protected $escape_char = '"';
2012-03-15 09:25:18 -04:00
/**
* PDO constructor wrapper
*/
public function __construct($dsn, $username=NULL, $password=NULL, $driver_options=array())
{
parent::__construct($dsn, $username, $password, $driver_options);
2012-04-10 14:06:34 -04:00
$this->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
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
2012-03-15 09:25:18 -04:00
/**
* Simplifies prepared statements for database queries
*
* @param string $sql
* @param array $data
* @return mixed PDOStatement / FALSE
*/
public function prepare_query($sql, $data)
{
// Prepare the sql
$query = $this->prepare($sql);
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
if( ! (is_object($query) || is_resource($query)))
{
$this->get_last_error();
return FALSE;
}
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
// Set the statement in the class variable for easy later access
$this->statement =& $query;
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
if( ! (is_array($data) || is_object($data)))
{
trigger_error("Invalid data argument");
return FALSE;
}
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
// Bind the parameters
foreach($data as $k => $value)
{
if(is_numeric($k))
{
$k++;
}
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
$res = $query->bindValue($k, $value);
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
if( ! $res)
{
trigger_error("Parameter not successfully bound");
return FALSE;
}
}
2012-04-10 14:06:34 -04:00
return $query;
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;
}
// -------------------------------------------------------------------------
/**
* Retreives the data from a select query
*
* @param PDOStatement $statement
* @return array
*/
public function get_query_data($statement)
{
$this->statement =& $statement;
// Execute the query
$this->statement->execute();
// Return the data array fetched
return $this->statement->fetchAll(PDO::FETCH_ASSOC);
}
// -------------------------------------------------------------------------
/**
* Returns number of rows affected by an INSERT, UPDATE, DELETE type query
*
* @param PDOStatement $statement
* @return int
*/
public function affected_rows($statement='')
{
if ( ! empty($statement))
{
2012-04-10 14:06:34 -04:00
$this->statement = $statement;
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-03-15 09:25:18 -04:00
/**
* Return the last error for the current database connection
*
* @return string
*/
public function get_last_error()
{
$info = $this->errorInfo();
2012-04-10 14:06:34 -04:00
2012-03-15 09:25:18 -04:00
echo "Error: <pre>{$info[0]}:{$info[1]}\n{$info[2]}</pre>";
}
// --------------------------------------------------------------------------
/**
* Surrounds the string with the databases identifier escape characters
*
* @param mixed $ident
* @return string
*/
public function quote_ident($ident)
{
if (is_array($ident))
{
return array_map(array($this, 'quote_ident'), $ident);
}
// Split each identifier by the period
$hiers = explode('.', $ident);
2012-04-10 14:06:34 -04:00
return $this->escape_char .
implode("{$this->escape_char}.{$this->escape_char}", $hiers) .
$this->escape_char;
2012-03-15 09:25:18 -04:00
}
// -------------------------------------------------------------------------
/**
* Deletes all the rows from a table. Does the same as the truncate
* method if the database does not support 'TRUNCATE';
*
* @param string $table
* @return mixed
*/
public function empty_table($table)
{
$sql = 'DELETE FROM '.$this->quote_ident($table);
return $this->query($sql);
}
// -------------------------------------------------------------------------
/**
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 FALSE;
}
// -------------------------------------------------------------------------
/**
* Method to simplify retreiving db results for meta-data queries
*
* @param string $sql
* @param bool $filtered_index
* @return mixed
*/
protected function driver_query($sql, $filtered_index=TRUE)
{
if ($sql === FALSE)
{
return FALSE;
}
$res = $this->query($sql);
$flag = ($filtered_index) ? PDO::FETCH_NUM : PDO::FETCH_ASSOC;
$all = $res->fetchAll($flag);
return ($filtered_index) ? db_filter($all, 0) : $all;
}
// -------------------------------------------------------------------------
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($this->sql->table_list());
}
// -------------------------------------------------------------------------
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($this->sql->db_list());
}
// -------------------------------------------------------------------------
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($this->sql->view_list());
}
// -------------------------------------------------------------------------
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($this->sql->sequence_list());
}
// -------------------------------------------------------------------------
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($this->sql->function_list(), FALSE);
}
// -------------------------------------------------------------------------
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($this->sql->procedure_list(), FALSE);
}
// -------------------------------------------------------------------------
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($this->sql->trigger_list(), FALSE);
}
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($this->sql->system_table_list());
}
// -------------------------------------------------------------------------
// ! Abstract public functions to override in child classes
// -------------------------------------------------------------------------
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);
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Return the number of rows for the last SELECT query
2012-03-15 09:25:18 -04:00
*
2012-04-10 14:06:34 -04:00
* @return int
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
abstract public function num_rows();
2012-03-15 09:25:18 -04:00
/**
2012-04-10 14:06:34 -04:00
* Connect to a different database
2012-03-15 09:25:18 -04:00
*
2012-04-10 14:06:34 -04:00
* @param string $name
* @return void
2012-03-15 09:25:18 -04:00
*/
2012-04-10 14:06:34 -04:00
abstract public function switch_db($name);
2012-03-15 09:25:18 -04:00
}
// End of db_pdo.php