Query/classes/query_builder.php

1437 lines
31 KiB
PHP
Raw Normal View History

2012-03-15 09:25:18 -04:00
<?php
/**
* Query
*
* Free Query Builder / Database Abstraction Layer
*
2012-04-23 13:28:49 -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-23 13:28:49 -04:00
* @license http://philsturgeon.co.uk/code/dbad-license
2012-03-15 09:25:18 -04:00
*/
// --------------------------------------------------------------------------
/**
* Convienience class for creating sql queries - also the class that
2012-03-15 09:25:18 -04:00
* instantiates the specific db driver
2012-04-20 13:17:39 -04:00
*
* @package Query
* @subpackage Query
2012-03-15 09:25:18 -04:00
*/
2012-12-18 16:19:52 -05:00
class Query_Builder implements iQuery_Builder {
2012-03-15 09:25:18 -04:00
2012-04-30 14:49:30 -04:00
// --------------------------------------------------------------------------
// ! SQL Clause Strings
// --------------------------------------------------------------------------
// Compiled 'select' clause
protected $select_string = '';
2012-08-09 12:15:36 -04:00
// Compiled 'from' clause
2012-12-18 16:19:52 -05:00
protected $from_string;
2012-08-09 12:15:36 -04:00
// Compiled arguments for insert / update
2012-12-18 16:19:52 -05:00
protected $set_string;
2012-08-09 12:15:36 -04:00
// Order by clause
2012-12-18 16:19:52 -05:00
protected $order_string;
2012-08-09 12:15:36 -04:00
// Group by clause
2012-12-18 16:19:52 -05:00
protected $group_string;
2012-08-09 12:15:36 -04:00
2012-04-30 14:49:30 -04:00
// --------------------------------------------------------------------------
// ! SQL Clause Arrays
// --------------------------------------------------------------------------
// Keys for insert/update statement
protected $set_array_keys = array();
2012-08-09 12:15:36 -04:00
// Key/val pairs for order by clause
protected $order_array = array();
2012-08-09 12:15:36 -04:00
// Key/val pairs for group by clause
protected $group_array = array();
2012-08-09 12:15:36 -04:00
2012-04-30 14:49:30 -04:00
// --------------------------------------------------------------------------
// ! Other Class vars
// --------------------------------------------------------------------------
// Values to apply to prepared statements
2012-12-18 16:19:52 -05:00
protected $values = array();
// Values to apply to where clauses in prepared statements
2012-12-18 16:19:52 -05:00
protected $where_values = array();
// Value for limit string
2012-12-18 16:19:52 -05:00
protected $limit;
2012-08-09 12:15:36 -04:00
// Value for offset in limit string
2012-12-18 16:19:52 -05:00
protected $offset;
// Alias to $this->db->sql
2012-05-08 15:37:36 -04:00
public $sql;
// Query component order mapping
// for complex select queries
//
// Format:
// array(
// 'type' => 'where',
// 'conjunction' => ' AND ',
// 'string' => 'k=?'
// )
protected $query_map = array();
2012-08-09 12:15:36 -04:00
// Map for having clause
2012-12-18 16:19:52 -05:00
protected $having_map;
2012-03-15 09:25:18 -04:00
// Convenience property for connection management
2012-04-12 13:44:31 -04:00
public $conn_name = "";
// List of sql queries executed
2012-09-12 09:55:04 -04:00
public $queries;
2014-02-21 15:08:00 -05:00
2014-02-04 20:59:30 -05:00
// Whether to do only an explain on the query
protected $explain;
2014-03-17 19:34:48 -04:00
// Subclass instances
public $db;
protected $parser;
public $util;
2012-09-12 09:55:04 -04:00
// --------------------------------------------------------------------------
// ! Methods
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Constructor
*
* @param DB_PDO $db
* @param object $params - the connection parameters
2012-03-15 09:25:18 -04:00
*/
2013-02-07 16:21:02 -05:00
public function __construct($db, $params)
2012-03-15 09:25:18 -04:00
{
$this->db = $db;
2012-08-09 12:15:36 -04:00
2012-10-31 16:21:15 -04:00
// Set the connection name property, if applicable
if (isset($params->name))
{
$this->conn_name = $params->name;
}
// Instantiate the Query Parser
$this->parser = new Query_Parser();
2013-02-28 12:22:55 -05:00
$this->queries['total_time'] = 0;
2012-10-31 16:21:15 -04:00
// Make things just slightly shorter
2013-02-07 16:21:02 -05:00
$this->sql = $this->db->sql;
$this->util = $this->db->util;
2012-10-31 16:21:15 -04:00
}
2014-02-21 15:08:00 -05:00
2014-02-17 20:10:47 -05:00
// --------------------------------------------------------------------------
2014-02-21 15:08:00 -05:00
2014-02-17 20:10:47 -05:00
/**
* Destructor
*/
public function __destruct()
{
$this->db = NULL;
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
// ! Select Queries
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Specifies rows to select in a query
*
* @param string $fields
* @return $this
*/
public function select($fields)
{
// Split fields by comma
$fields_array = explode(",", $fields);
2012-11-07 08:42:34 -05:00
$fields_array = array_map('mb_trim', $fields_array);
2012-03-15 09:25:18 -04:00
// Split on 'As'
foreach ($fields_array as $key => $field)
2012-03-15 09:25:18 -04:00
{
if (stripos($field, 'as') !== FALSE)
{
$fields_array[$key] = preg_split('` as `i', $field);
2012-11-07 08:42:34 -05:00
$fields_array[$key] = array_map('mb_trim', $fields_array[$key]);
2012-03-15 09:25:18 -04:00
}
}
// Quote the identifiers
$safe_array = $this->db->quote_ident($fields_array);
2012-03-15 09:25:18 -04:00
unset($fields_array);
// Join the strings back together
for($i = 0, $c = count($safe_array); $i < $c; $i++)
{
if (is_array($safe_array[$i]))
{
$safe_array[$i] = implode(' AS ', $safe_array[$i]);
}
}
2014-02-14 10:38:25 -05:00
$this->select_string .= implode(', ', $safe_array);
2012-03-15 09:25:18 -04:00
unset($safe_array);
return $this;
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
2012-04-30 15:29:45 -04:00
* Method to simplify select_ methods
*
* @param string $field
* @param string $as
2012-04-30 15:29:45 -04:00
* @return string
*/
2012-12-18 16:19:52 -05:00
protected function _select($field, $as = FALSE)
{
// Escape the identifiers
$field = $this->db->quote_ident($field);
2012-08-09 12:15:36 -04:00
$as = ($as !== FALSE)
? $this->db->quote_ident($as)
: $field;
2012-08-09 12:15:36 -04:00
2012-04-30 15:29:45 -04:00
return "({$field}) AS {$as} ";
}
2012-08-09 12:15:36 -04:00
2012-04-30 15:29:45 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-04-30 15:29:45 -04:00
/**
* Selects the maximum value of a field from a query
*
* @param string $field
* @param string $as
* @return $this
*/
public function select_max($field, $as=FALSE)
{
// Create the select string
2012-10-31 16:21:15 -04:00
$this->select_string .= ' MAX'.$this->_select($field, $as);
return $this;
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Selects the minimum value of a field from a query
*
* @param string $field
* @param string $as
* @return $this
*/
public function select_min($field, $as=FALSE)
2012-08-09 12:15:36 -04:00
{
// Create the select string
2012-10-31 16:21:15 -04:00
$this->select_string .= ' MIN'.$this->_select($field, $as);
return $this;
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Selects the average value of a field from a query
*
* @param string $field
* @param string $as
* @return $this
*/
public function select_avg($field, $as=FALSE)
{
// Create the select string
2012-10-31 16:21:15 -04:00
$this->select_string .= ' AVG'.$this->_select($field, $as);
return $this;
}
2012-08-09 12:15:36 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Selects the sum of a field from a query
*
* @param string $field
* @param string $as
* @return $this
*/
public function select_sum($field, $as=FALSE)
{
// Create the select string
2012-10-31 16:21:15 -04:00
$this->select_string .= ' SUM'.$this->_select($field, $as);
return $this;
}
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Adds the 'distinct' keyword to a query
*
* @return $this
*/
public function distinct()
{
// Prepend the keyword to the select string
2012-10-31 16:21:15 -04:00
$this->select_string = ' DISTINCT '.$this->select_string;
return $this;
}
2014-02-21 15:08:00 -05:00
2014-02-04 20:59:30 -05:00
// --------------------------------------------------------------------------
2014-02-21 15:08:00 -05:00
2014-02-04 20:59:30 -05:00
/**
* Tell the database to give you the query plan instead of result set
*
* @return $this
*/
public function explain()
{
$this->explain = TRUE;
return $this;
}
2012-08-09 12:15:36 -04:00
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Specify the database table to select from
*
2012-11-07 08:42:34 -05:00
* @param string $tblname
2012-03-15 09:25:18 -04:00
* @return $this
*/
2012-11-07 08:42:34 -05:00
public function from($tblname)
2012-03-15 09:25:18 -04:00
{
// Split identifiers on spaces
2014-02-14 10:38:25 -05:00
$ident_array = explode(' ', mb_trim($tblname));
2012-11-07 08:42:34 -05:00
$ident_array = array_map('mb_trim', $ident_array);
// Quote the identifiers
2012-11-07 08:42:34 -05:00
$ident_array[0] = $this->db->quote_table($ident_array[0]);
$ident_array = $this->db->quote_ident($ident_array);
2012-03-15 09:25:18 -04:00
// Paste it back together
$this->from_string = implode(' ', $ident_array);
2012-03-15 09:25:18 -04:00
return $this;
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
// ! 'Like' methods
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-03-15 09:25:18 -04:00
/**
2012-04-30 16:06:06 -04:00
* Simplify 'like' methods
2012-03-15 09:25:18 -04:00
*
* @param string $field
* @param mixed $val
* @param string $pos
2012-04-30 16:06:06 -04:00
* @param string $like
* @param string $conj
2012-05-01 13:45:11 -04:00
* @return $this
2012-03-15 09:25:18 -04:00
*/
2012-12-18 16:19:52 -05:00
protected function _like($field, $val, $pos, $like='LIKE', $conj='AND')
{
$field = $this->db->quote_ident($field);
2012-03-15 09:25:18 -04:00
// Add the like string into the order map
2012-04-30 16:06:06 -04:00
$l = $field. " {$like} ?";
2012-03-15 09:25:18 -04:00
if ($pos == 'before')
{
$val = "%{$val}";
}
elseif ($pos == 'after')
{
$val = "{$val}%";
}
else
{
$val = "%{$val}%";
}
2012-08-09 12:15:36 -04:00
2012-03-15 09:25:18 -04:00
$this->query_map[] = array(
'type' => 'like',
'conjunction' => (empty($this->query_map)) ? ' WHERE ' : " {$conj} ",
2012-03-15 09:25:18 -04:00
'string' => $l
);
2012-08-09 12:15:36 -04:00
2012-03-15 09:25:18 -04:00
// Add to the values array
2012-10-24 20:19:39 -04:00
$this->where_values[] = $val;
2012-08-09 12:15:36 -04:00
2012-05-01 13:45:11 -04:00
return $this;
2012-08-09 12:15:36 -04:00
}
2012-04-30 16:06:06 -04:00
// --------------------------------------------------------------------------
2012-04-30 16:06:06 -04:00
/**
* Creates a Like clause in the sql statement
*
* @param string $field
* @param mixed $val
* @param string $pos
* @return $this
*/
public function like($field, $val, $pos='both')
{
2012-05-01 13:45:11 -04:00
return $this->_like($field, $val, $pos, 'LIKE', 'AND');
2012-03-15 09:25:18 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Generates an OR Like clause
*
* @param string $field
* @param mixed $val
* @param string $pos
* @return $this
*/
public function or_like($field, $val, $pos='both')
{
2012-05-01 13:45:11 -04:00
return $this->_like($field, $val, $pos, 'LIKE', 'OR');
2012-03-15 09:25:18 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Generates a NOT LIKE clause
*
* @param string $field
* @param mixed $val
* @param string $pos
* @return $this
*/
public function not_like($field, $val, $pos='both')
{
2012-05-01 13:45:11 -04:00
return $this->_like($field, $val, $pos, 'NOT LIKE', 'AND');
2012-03-15 09:25:18 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Generates a OR NOT LIKE clause
*
* @param string $field
* @param mixed $val
* @param string $pos
2012-04-20 13:17:39 -04:00
* @return $this
2012-03-15 09:25:18 -04:00
*/
public function or_not_like($field, $val, $pos='both')
{
2012-05-01 13:45:11 -04:00
return $this->_like($field, $val, $pos, 'NOT LIKE', 'OR');
2012-03-15 09:25:18 -04:00
}
2012-08-09 12:15:36 -04:00
2012-05-01 13:45:11 -04:00
// --------------------------------------------------------------------------
// ! Having methods
2012-04-16 13:43:41 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
/**
2012-05-01 13:45:11 -04:00
* Simplify building having clauses
2012-04-16 13:43:41 -04:00
*
* @param mixed $key
* @param mixed $val
2012-05-01 13:45:11 -04:00
* @param string $conj
2012-04-16 13:43:41 -04:00
* @return $this
*/
2012-12-18 16:19:52 -05:00
protected function _having($key, $val=array(), $conj='AND')
2012-04-16 13:43:41 -04:00
{
$where = $this->_where($key, $val);
2012-08-09 12:15:36 -04:00
// Create key/value placeholders
foreach($where as $f => $val)
{
// Split each key by spaces, in case there
// is an operator such as >, <, !=, etc.
$f_array = explode(' ', trim($f));
$item = $this->db->quote_ident($f_array[0]);
// Simple key value, or an operator
2012-09-12 09:55:04 -04:00
$item .= (count($f_array) === 1) ? '=?' : " {$f_array[1]} ?";
// Put in the query map for select statements
$this->having_map[] = array(
2012-05-01 13:45:11 -04:00
'conjunction' => ( ! empty($this->having_map)) ? " {$conj} " : ' HAVING ',
'string' => $item
);
}
2012-08-09 12:15:36 -04:00
return $this;
2012-04-16 13:43:41 -04:00
}
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-05-01 13:45:11 -04:00
/**
* Generates a 'Having' clause
*
* @param mixed $key
* @param mixed $val
* @return $this
*/
public function having($key, $val=array())
{
return $this->_having($key, $val, 'AND');
}
2012-08-09 12:15:36 -04:00
2012-05-01 13:45:11 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
/**
* Generates a 'Having' clause prefixed with 'OR'
*
* @param mixed $key
* @param mixed $val
* @return $this
*/
public function or_having($key, $val=array())
{
2012-05-01 13:45:11 -04:00
return $this->_having($key, $val, 'OR');
2012-04-16 13:43:41 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
// ! 'Where' methods
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Do all the repeditive stuff for where/having type methods
2012-03-15 09:25:18 -04:00
*
* @param mixed $key
* @param mixed $val
* @return array
*/
2012-12-18 16:19:52 -05:00
protected function _where($key, $val=array())
2012-03-15 09:25:18 -04:00
{
$where = array();
2012-03-15 09:25:18 -04:00
// Key and value passed? Add them to the where array
if (is_scalar($key) && is_scalar($val))
{
$where[$key] = $val;
2012-10-24 20:19:39 -04:00
$this->where_values[] = $val;
2012-03-15 09:25:18 -04:00
}
// Array or object, loop through and add to the where array
elseif ( ! is_scalar($key))
{
foreach($key as $k => $v)
2012-03-15 09:25:18 -04:00
{
$where[$k] = $v;
2012-10-24 20:19:39 -04:00
$this->where_values[] = $v;
2012-03-15 09:25:18 -04:00
}
}
2012-03-15 09:25:18 -04:00
return $where;
}
2012-08-09 12:15:36 -04:00
2012-04-30 16:06:06 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-05-01 13:45:11 -04:00
/**
* Simplify generating where string
*
* @param mixed $key
* @param mixed $val
* @param string $conj
* @return $this
*/
2012-12-18 16:19:52 -05:00
protected function _where_string($key, $val=array(), $conj='AND')
2012-05-01 13:45:11 -04:00
{
$where = $this->_where($key, $val);
// Create key/value placeholders
foreach($where as $f => $val)
2012-05-01 13:45:11 -04:00
{
// Split each key by spaces, in case there
// is an operator such as >, <, !=, etc.
$f_array = explode(' ', trim($f));
$item = $this->db->quote_ident($f_array[0]);
2012-05-01 13:45:11 -04:00
// Simple key value, or an operator
$item .= (count($f_array) === 1) ? '=?' : " {$f_array[1]} ?";
2014-02-21 15:08:00 -05:00
// Get the type of the first item in the query map
$first_item = end($this->query_map);
2014-02-21 15:08:00 -05:00
// Determine the correct conjunction
if (empty($this->query_map))
{
2014-02-14 10:38:25 -05:00
$conj = "\nWHERE ";
}
elseif ($first_item['type'] === 'group_start')
{
$conj = '';
}
2014-02-21 15:08:00 -05:00
else
{
$conj = " {$conj} ";
}
2012-05-01 13:45:11 -04:00
// Put in the query map for select statements
$this->query_map[] = array(
'type' => 'where',
'conjunction' => $conj,
2012-05-01 13:45:11 -04:00
'string' => $item
);
}
return $this;
}
2012-08-09 12:15:36 -04:00
2012-05-01 13:45:11 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-04-30 16:06:06 -04:00
/**
* Simplify where_in methods
*
* @param mixed $key
* @param mixed $val
* @param string $in - The (not) in fragment
* @param string $conj - The where in conjunction
2012-05-01 13:45:11 -04:00
* @return $this
2012-04-30 16:06:06 -04:00
*/
2012-12-18 16:19:52 -05:00
protected function _where_in($key, $val=array(), $in='IN', $conj='AND')
2012-04-30 16:06:06 -04:00
{
$key = $this->db->quote_ident($key);
2012-04-30 16:06:06 -04:00
$params = array_fill(0, count($val), '?');
foreach($val as $v)
2012-04-30 16:06:06 -04:00
{
2012-10-24 20:19:39 -04:00
$this->where_values[] = $v;
2012-04-30 16:06:06 -04:00
}
$string = $key . " {$in} (".implode(',', $params).') ';
2012-04-30 16:06:06 -04:00
$this->query_map[] = array(
'type' => 'where_in',
'conjunction' => ( ! empty($this->query_map)) ? " {$conj} " : ' WHERE ',
'string' => $string
);
2012-08-09 12:15:36 -04:00
2012-05-01 13:45:11 -04:00
return $this;
2012-04-30 16:06:06 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Specify condition(s) in the where clause of a query
* Note: this function works with key / value, or a
2012-03-15 09:25:18 -04:00
* passed array with key / value pairs
*
* @param mixed $key
2012-03-15 09:25:18 -04:00
* @param mixed $val
2014-02-04 20:59:30 -05:00
* @param mixed $escape
2012-03-15 09:25:18 -04:00
* @return $this
*/
2014-02-04 20:59:30 -05:00
public function where($key, $val=array(), $escape=NULL)
2012-03-15 09:25:18 -04:00
{
2012-05-01 13:45:11 -04:00
return $this->_where_string($key, $val, 'AND');
2012-03-15 09:25:18 -04:00
}
// --------------------------------------------------------------------------
/**
* Where clause prefixed with "OR"
*
2012-05-01 13:45:11 -04:00
* @param string $key
2012-03-15 09:25:18 -04:00
* @param mixed $val
* @return $this
*/
2012-05-01 13:45:11 -04:00
public function or_where($key, $val=array())
2012-03-15 09:25:18 -04:00
{
2012-05-01 13:45:11 -04:00
return $this->_where_string($key, $val, 'OR');
2012-03-15 09:25:18 -04:00
}
// --------------------------------------------------------------------------
/**
* Where clause with 'IN' statement
*
* @param mixed $field
* @param mixed $val
* @return $this
*/
public function where_in($field, $val=array())
{
2012-05-01 13:45:11 -04:00
return $this->_where_in($field, $val);
2012-03-15 09:25:18 -04:00
}
// --------------------------------------------------------------------------
/**
* Where in statement prefixed with "or"
*
* @param string $field
* @param mixed $val
* @return $this
*/
public function or_where_in($field, $val=array())
{
2012-05-01 13:45:11 -04:00
return $this->_where_in($field, $val, 'IN', 'OR');
2012-03-15 09:25:18 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
/**
* WHERE NOT IN (FOO) clause
*
* @param string $field
* @param mixed $val
* @return $this
*/
public function where_not_in($field, $val=array())
{
2012-05-01 13:45:11 -04:00
return $this->_where_in($field, $val, 'NOT IN', 'AND');
2012-03-15 09:25:18 -04:00
}
// --------------------------------------------------------------------------
/**
* OR WHERE NOT IN (FOO) clause
*
2012-03-15 09:25:18 -04:00
* @param string $field
* @param mixed $val
* @return $this
*/
public function or_where_not_in($field, $val=array())
{
2012-05-01 13:45:11 -04:00
return $this->_where_in($field, $val, 'NOT IN', 'OR');
2012-03-15 09:25:18 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
// ! Other Query Modifier methods
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
/**
* Sets values for inserts / updates / deletes
*
* @param mixed $key
* @param mixed $val
* @return $this
*/
2012-04-30 14:49:30 -04:00
public function set($key, $val = NULL)
2012-04-16 13:43:41 -04:00
{
// Plain key, value pair
if (is_scalar($key) && is_scalar($val))
{
2012-04-30 14:49:30 -04:00
$this->set_array_keys[] = $key;
2012-04-16 13:43:41 -04:00
$this->values[] = $val;
}
// Object or array
2014-02-14 22:08:19 -05:00
elseif (is_array($key) || is_object($key))
2012-04-16 13:43:41 -04:00
{
foreach($key as $k => $v)
2012-04-16 13:43:41 -04:00
{
2012-04-30 14:49:30 -04:00
$this->set_array_keys[] = $k;
$this->values[] = $v;
2012-04-16 13:43:41 -04:00
}
}
// Use the keys of the array to make the insert/update string
// Escape the field names
2014-02-14 22:08:19 -05:00
$this->set_array_keys = array_map(array($this->db, '_quote'), $this->set_array_keys);
2012-04-16 13:43:41 -04:00
// Generate the "set" string
$this->set_string = implode('=?,', $this->set_array_keys);
2012-04-16 13:43:41 -04:00
$this->set_string .= '=?';
return $this;
}
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Creates a join phrase in a compiled query
*
* @param string $table
* @param string $condition
* @param string $type
* @return $this
*/
public function join($table, $condition, $type='')
{
2012-11-07 08:42:34 -05:00
// Prefix and quote table name
$table = explode(' ', mb_trim($table));
$table[0] = $this->db->quote_table($table[0]);
$table = $this->db->quote_ident($table);
$table = implode(' ', $table);
2012-08-09 12:15:36 -04:00
// Parse out the join condition
2012-10-31 16:21:15 -04:00
$parts = $this->parser->parse_join($condition);
2012-08-09 12:15:36 -04:00
$count = count($parts['identifiers']);
// Go through and quote the identifiers
for($i=0; $i <= $count; $i++)
{
if (in_array($parts['combined'][$i], $parts['identifiers']) && ! is_numeric($parts['combined'][$i]))
{
$parts['combined'][$i] = $this->db->quote_ident($parts['combined'][$i]);
2012-08-09 12:15:36 -04:00
}
}
$parsed_condition = implode('', $parts['combined']);
2012-08-09 12:15:36 -04:00
$condition = $table . ' ON ' . $parsed_condition;
2012-03-15 09:25:18 -04:00
$this->query_map[] = array(
'type' => 'join',
2014-02-14 10:38:25 -05:00
'conjunction' => "\n" . strtoupper($type) . ' JOIN ',
2012-03-15 09:25:18 -04:00
'string' => $condition,
);
2012-03-15 09:25:18 -04:00
return $this;
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Group the results by the selected field(s)
*
* @param mixed $field
* @return $this
*/
public function group_by($field)
{
if ( ! is_scalar($field))
{
$new_group_array = array_map(array($this->db, 'quote_ident'), $field);
$this->group_array = array_merge($this->group_array, $new_group_array);
2012-03-15 09:25:18 -04:00
}
else
{
$this->group_array[] = $this->db->quote_ident($field);
2012-03-15 09:25:18 -04:00
}
$this->group_string = ' GROUP BY ' . implode(',', $this->group_array);
2012-03-15 09:25:18 -04:00
return $this;
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Order the results by the selected field(s)
*
* @param string $field
* @param string $type
* @return $this
*/
public function order_by($field, $type="")
{
// Random case
if (stripos($type, 'rand') !== FALSE)
{
$type = (($rand = $this->sql->random()) !== FALSE ) ? $rand : 'ASC';
}
2012-03-15 09:25:18 -04:00
// Set fields for later manipulation
$field = $this->db->quote_ident($field);
2012-03-15 09:25:18 -04:00
$this->order_array[$field] = $type;
2012-03-15 09:25:18 -04:00
$order_clauses = array();
2012-03-15 09:25:18 -04:00
// Flatten key/val pairs into an array of space-separated pairs
foreach($this->order_array as $k => $v)
2012-03-15 09:25:18 -04:00
{
$order_clauses[] = $k . ' ' . strtoupper($v);
}
2012-03-15 09:25:18 -04:00
// Set the final string
$this->order_string = (empty($rand))
2014-02-14 10:38:25 -05:00
? "\nORDER BY ".implode(', ', $order_clauses)
: "\nORDER BY".$rand;
2012-03-15 09:25:18 -04:00
return $this;
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Set a limit on the current sql statement
*
* @param int $limit
* @param int $offset
* @return Query_Builder
2012-03-15 09:25:18 -04:00
*/
public function limit($limit, $offset=FALSE)
{
$this->limit = $limit;
$this->offset = $offset;
2012-03-15 09:25:18 -04:00
return $this;
}
// --------------------------------------------------------------------------
// ! Query Grouping Methods
// --------------------------------------------------------------------------
/**
* Adds a paren to the current query for query grouping
*
* @return $this
*/
public function group_start()
{
$this->query_map[] = array(
'type' => 'group_start',
'conjunction' => (empty($this->query_map)) ? ' WHERE ' : ' ',
'string' => '('
);
return $this;
}
// --------------------------------------------------------------------------
/**
* Adds a paren to the current query for query grouping,
* prefixed with 'OR'
*
* @return $this
*/
public function or_group_start()
{
$this->query_map[] = array(
'type' => 'group_start',
'conjunction' => '',
'string' => ' OR ('
);
return $this;
}
// --------------------------------------------------------------------------
/**
* Adds a paren to the current query for query grouping,
* prefixed with 'OR NOT'
*
* @return $this
*/
public function or_not_group_start()
{
$this->query_map[] = array(
'type' => 'group_start',
'conjunction' => '',
'string' => ' OR NOT ('
);
return $this;
}
// --------------------------------------------------------------------------
/**
* Ends a query group
*
* @return $this
*/
public function group_end()
{
$this->query_map[] = array(
'type' => 'group_end',
'conjunction' => '',
'string' => ')'
);
return $this;
}
2012-08-09 12:15:36 -04:00
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
// ! Query execution methods
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Select and retrieve all records from the current table, and/or
* execute current compiled query
*
* @param $table
* @param int $limit
* @param int $offset
* @return object
*/
public function get($table='', $limit=FALSE, $offset=FALSE)
{
// Set the table
if ( ! empty($table))
{
$this->from($table);
}
// Set the limit, if it exists
if ($limit !== FALSE)
{
$this->limit($limit, $offset);
}
return $this->_run("get", $table);
2012-03-15 09:25:18 -04:00
}
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
/**
* Convience method for get() with a where clause
*
* @param string $table
* @param array $where
* @param int $limit
* @param int $offset
* @return object
*/
public function get_where($table, $where=array(), $limit=FALSE, $offset=FALSE)
{
// Create the where clause
$this->where($where);
2012-08-09 12:15:36 -04:00
// Return the result
return $this->get($table, $limit, $offset);
}
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
/**
* Retreive the number of rows in the selected table
*
* @param string $table
* @return int
*/
public function count_all($table)
{
$sql = 'SELECT * FROM '.$this->db->quote_table($table);
$res = $this->db->query($sql);
return (int) count($res->fetchAll());
2012-04-16 13:43:41 -04:00
}
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
// --------------------------------------------------------------------------
2012-08-09 12:15:36 -04:00
2012-04-16 13:43:41 -04:00
/**
* Retrieve the number of results for the generated query - used
* in place of the get() method
*
* @param string $table
* @return int
*/
public function count_all_results($table='')
{
// Set the table
if ( ! empty($table))
{
$this->from($table);
}
2012-08-09 12:15:36 -04:00
$result = $this->_run('get', $table);
$rows = $result->fetchAll();
2012-08-09 12:15:36 -04:00
2012-04-17 11:43:06 -04:00
return (int) count($rows);
2012-04-16 13:43:41 -04:00
}
2012-08-09 12:15:36 -04:00
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
/**
* Creates an insert clause, and executes it
*
* @param string $table
* @param mixed $data
* @return mixed
*/
public function insert($table, $data=array())
{
// No use duplicating logic!
if ( ! empty($data))
{
$this->set($data);
}
return $this->_run("insert", $table);
2012-03-15 09:25:18 -04:00
}
2014-02-21 15:08:00 -05:00
// --------------------------------------------------------------------------
2014-02-21 15:08:00 -05:00
/**
* Create sql for batch insert
*
* @param string $table
* @param array $data
* @return string
*/
public function insert_batch($table, $data=array())
{
// Get the generated values and sql string
list($sql, $data) = $this->db->insert_batch($table, $data);
2014-02-21 15:08:00 -05:00
if ( ! is_null($sql))
{
return $this->_run('', $table, $sql, $data);
}
2014-02-21 15:08:00 -05:00
return NULL;
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
/**
* Creates an update clause, and executes it
*
* @param string $table
* @param mixed $data
* @return mixed
*/
public function update($table, $data=array())
{
// No use duplicating logic!
if ( ! empty($data))
{
$this->set($data);
}
return $this->_run("update", $table);
2012-03-15 09:25:18 -04:00
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
/**
* Deletes data from a table
*
* @param string $table
* @param mixed $where
* @return mixed
*/
public function delete($table, $where='')
{
// Set the where clause
if ( ! empty($where))
{
$this->where($where);
}
return $this->_run("delete", $table);
2012-03-15 09:25:18 -04:00
}
// --------------------------------------------------------------------------
2012-12-18 16:19:52 -05:00
// ! SQL Returning Methods
// --------------------------------------------------------------------------
/**
* Returns the generated 'select' sql query
*
* @param string $table
* @param bool $reset
* @return string
*/
public function get_compiled_select($table='', $reset=TRUE)
{
// Set the table
if ( ! empty($table))
{
$this->from($table);
}
return $this->_get_compile('select', $table, $reset);
}
// --------------------------------------------------------------------------
/**
* Returns the generated 'insert' sql query
*
* @param string $table
* @param bool $reset
* @return string
*/
public function get_compiled_insert($table, $reset=TRUE)
{
return $this->_get_compile('insert', $table, $reset);
}
// --------------------------------------------------------------------------
/**
2012-12-18 16:19:52 -05:00
* Returns the generated 'update' sql query
*
* @param string $table
* @param bool $reset
* @return string
*/
public function get_compiled_update($table='', $reset=TRUE)
{
return $this->_get_compile('update', $table, $reset);
}
// --------------------------------------------------------------------------
/**
2012-12-18 16:19:52 -05:00
* Returns the generated 'delete' sql query
*
* @param string $table
* @param bool $reset
* @return string
*/
public function get_compiled_delete($table="", $reset=TRUE)
{
return $this->_get_compile('delete', $table, $reset);
}
// --------------------------------------------------------------------------
/**
* Helper function for returning sql strings
*
* @param string $type
* @param string $table
* @param bool $reset
* @resturn string
*/
2012-12-18 16:19:52 -05:00
protected function _get_compile($type, $table, $reset)
{
$sql = $this->_compile($type, $table);
// Reset the query builder for the next query
if ($reset)
{
$this->reset_query();
}
return $sql;
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
// ! Miscellaneous Methods
// --------------------------------------------------------------------------
/**
* Clear out the class variables, so the next query can be run
*
* @return void
2012-03-15 09:25:18 -04:00
*/
public function reset_query()
2012-03-15 09:25:18 -04:00
{
$null_properties = array(
'select_string',
'from_string',
'set_string',
'order_string',
'group_string',
'limit',
'offset',
'explain'
);
$array_properties = array(
'set_array_keys',
'order_array',
'group_array',
'values',
'where_values',
'query_map',
'having_map'
);
// Reset strings and booleans
foreach($null_properties as $var)
{
$this->$var = NULL;
}
2012-08-09 12:15:36 -04:00
// Reset arrays
foreach($array_properties as $var)
{
$this->$var = array();
2012-03-15 09:25:18 -04:00
}
}
2012-03-15 09:25:18 -04:00
// --------------------------------------------------------------------------
2012-10-24 20:19:39 -04:00
/**
* Executes the compiled query
*
* @param string $type
* @param string $table
* @param string $sql
* @param mixed $vals
2012-10-24 20:19:39 -04:00
* @return mixed
*/
protected function _run($type, $table, $sql=NULL, $vals=NULL)
2012-10-24 20:19:39 -04:00
{
if (is_null($sql))
{
$sql = $this->_compile($type, $table);
}
2014-02-21 15:08:00 -05:00
if (is_null($vals))
{
$vals = array_merge($this->values, (array) $this->where_values);
}
2014-02-21 15:08:00 -05:00
$evals = (is_array($vals)) ? $vals : array();
2013-02-28 12:22:55 -05:00
$start_time = microtime(TRUE);
2014-02-21 15:08:00 -05:00
if (empty($vals))
2014-02-14 22:08:19 -05:00
{
$res = $this->db->query($sql);
}
else
{
$res = $this->db->prepare_execute($sql, $vals);
}
2013-02-28 12:22:55 -05:00
$end_time = microtime(TRUE);
$total_time = number_format($end_time - $start_time, 5);
// Add the interpreted query to the list of executed queries
foreach($evals as $k => &$v)
{
2014-02-21 15:08:00 -05:00
$v = ( ! is_numeric($v)) ? htmlentities($this->db->quote($v), ENT_NOQUOTES, 'utf-8', FALSE) : $v;
}
2014-02-14 10:38:25 -05:00
$esql = str_replace('?', "%s", $sql);
2013-02-28 12:22:55 -05:00
array_unshift($vals, $esql);
array_unshift($evals, $esql);
2014-02-21 15:08:00 -05:00
2013-02-28 12:22:55 -05:00
$this->queries[] = array(
'time' => $total_time,
'sql' => call_user_func_array('sprintf', $evals),
2013-02-28 12:22:55 -05:00
);
$this->queries['total_time'] += $total_time;
array_shift($vals);
// Set the last query to get rowcounts properly
$this->db->last_query = $sql;
// Reset class state for next query
$this->reset_query();
return $res;
2012-10-24 20:19:39 -04:00
}
2012-10-24 20:19:39 -04:00
// --------------------------------------------------------------------------
/**
* Calls a function further down the inheritence chain
*
* @param string $name
* @param array $params
* @return mixed
2014-02-14 22:08:19 -05:00
* @throws BadMethodCallException
*/
public function __call($name, $params)
{
if (method_exists($this->db, $name))
{
return call_user_func_array(array($this->db, $name), $params);
}
2014-02-14 22:08:19 -05:00
throw new BadMethodCallException("Method does not exist");
}
// --------------------------------------------------------------------------
2012-03-15 09:25:18 -04:00
/**
* Sub-method for generating sql strings
2012-03-15 09:25:18 -04:00
*
* @param string $type
* @param string $table
* @return $string
*/
protected function _compile_type($type='', $table='')
2012-03-15 09:25:18 -04:00
{
if ($type === 'insert')
{
$param_count = count($this->set_array_keys);
$params = array_fill(0, $param_count, '?');
$sql = "INSERT INTO {$table} ("
. implode(',', $this->set_array_keys)
. ")\nVALUES (".implode(',', $params).')';
}
elseif ($type === 'update')
2012-03-15 09:25:18 -04:00
{
$sql = "UPDATE {$table}\nSET {$this->set_string}";
}
elseif ($type === 'delete')
{
$sql = "DELETE FROM {$table}";
}
else // GET queries
{
$sql = "SELECT * \nFROM {$this->from_string}";
// Set the select string
if ( ! empty($this->select_string))
{
// Replace the star with the selected fields
$sql = str_replace('*', $this->select_string, $sql);
}
}
2012-03-15 09:25:18 -04:00
return $sql;
}
// --------------------------------------------------------------------------
/**
* String together the sql statements for sending to the db
*
* @param string $type
* @param string $table
* @return $string
*/
protected function _compile($type='', $table='')
{
// Get the base clause for the query
$sql = $this->_compile_type($type, $this->db->quote_table($table));
2012-11-07 08:42:34 -05:00
// Set the where clause
if ( ! empty($this->query_map))
{
foreach($this->query_map as $k => $q)
{
$sql .= $q['conjunction'] . $q['string'];
}
}
2012-03-15 09:25:18 -04:00
2012-11-07 08:42:34 -05:00
// Set the group_by clause
if ( ! empty($this->group_string))
{
$sql .= $this->group_string;
}
2012-11-07 08:42:34 -05:00
// Set the order_by clause
if ( ! empty($this->order_string))
{
$sql .= $this->order_string;
}
2012-11-07 08:42:34 -05:00
// Set the having clause
if ( ! empty($this->having_map))
{
foreach($this->having_map as $h)
{
$sql .= $h['conjunction'] . $h['string'];
}
2012-03-15 09:25:18 -04:00
}
2012-11-07 08:42:34 -05:00
// Set the limit via the class variables
if (isset($this->limit) && is_numeric($this->limit))
{
$sql = $this->sql->limit($sql, $this->limit, $this->offset);
}
2014-02-21 15:08:00 -05:00
2014-02-04 20:59:30 -05:00
// See what needs to happen to only return the query plan
if (isset($this->explain) && $this->explain === TRUE)
{
$sql = $this->sql->explain($sql);
}
2012-08-09 12:15:36 -04:00
2012-03-15 09:25:18 -04:00
return $sql;
}
}
2014-02-04 20:59:30 -05:00
// End of query_builder.php