1439 lines
48 KiB
PHP
1439 lines
48 KiB
PHP
<?php
|
|
/*
|
|
* $Id$
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
* This software consists of voluntary contributions made by many individuals
|
|
* and is licensed under the LGPL. For more information, see
|
|
* <http://www.phpdoctrine.com>.
|
|
*/
|
|
Doctrine::autoload('Doctrine_Query_Abstract');
|
|
/**
|
|
* Doctrine_Query
|
|
*
|
|
* @package Doctrine
|
|
* @license http://www.opensource.org/licenses/lgpl-license.php LGPL
|
|
* @category Object Relational Mapping
|
|
* @link www.phpdoctrine.com
|
|
* @since 1.0
|
|
* @version $Revision$
|
|
* @author Konsta Vesterinen <kvesteri@cc.hut.fi>
|
|
*/
|
|
class Doctrine_Query extends Doctrine_Query_Abstract implements Countable
|
|
{
|
|
const STATE_CLEAN = 1;
|
|
|
|
const STATE_DIRTY = 2;
|
|
|
|
const STATE_DIRECT = 3;
|
|
|
|
const STATE_LOCKED = 4;
|
|
|
|
|
|
protected $subqueryAliases = array();
|
|
/**
|
|
* @param boolean $needsSubquery
|
|
*/
|
|
protected $needsSubquery = false;
|
|
/**
|
|
* @param boolean $isSubquery whether or not this query object is a subquery of another
|
|
* query object
|
|
*/
|
|
protected $isSubquery;
|
|
|
|
protected $isLimitSubqueryUsed = false;
|
|
/**
|
|
* @var array $_neededTableAliases an array containing the needed table aliases
|
|
*/
|
|
protected $_neededTables = array();
|
|
/**
|
|
* @var array $pendingFields
|
|
*/
|
|
protected $pendingFields = array();
|
|
/**
|
|
* @var array $pendingSubqueries SELECT part subqueries, these are called pending subqueries since
|
|
* they cannot be parsed directly (some queries might be correlated)
|
|
*/
|
|
protected $pendingSubqueries = array();
|
|
/**
|
|
* @var array $_parsers an array of parser objects, each DQL query part has its own parser
|
|
*/
|
|
protected $_parsers = array();
|
|
/**
|
|
* @var array $_enumParams an array containing the keys of the parameters that should be enumerated
|
|
*/
|
|
protected $_enumParams = array();
|
|
|
|
/**
|
|
* @var array $_dqlParts an array containing all DQL query parts
|
|
*/
|
|
protected $_dqlParts = array(
|
|
'select' => array(),
|
|
'forUpdate' => false,
|
|
'from' => array(),
|
|
'set' => array(),
|
|
'join' => array(),
|
|
'where' => array(),
|
|
'groupby' => array(),
|
|
'having' => array(),
|
|
'orderby' => array(),
|
|
'limit' => array(),
|
|
'offset' => array(),
|
|
);
|
|
/**
|
|
* @var array $_pendingJoinConditions an array containing pending joins
|
|
*/
|
|
protected $_pendingJoinConditions = array();
|
|
|
|
protected $_expressionMap = array();
|
|
|
|
protected $_state = Doctrine_Query::STATE_CLEAN;
|
|
|
|
/**
|
|
* create
|
|
* returns a new Doctrine_Query object
|
|
*
|
|
* @param Doctrine_Connection $conn optional connection parameter
|
|
* @return Doctrine_Query
|
|
*/
|
|
public static function create($conn = null)
|
|
{
|
|
return new Doctrine_Query($conn);
|
|
}
|
|
public function reset()
|
|
{
|
|
$this->_pendingJoinConditions = array();
|
|
$this->pendingSubqueries = array();
|
|
$this->pendingFields = array();
|
|
$this->_neededTables = array();
|
|
$this->_expressionMap = array();
|
|
$this->subqueryAliases = array();
|
|
$this->needsSubquery = false;
|
|
$this->isLimitSubqueryUsed = false;
|
|
}
|
|
/**
|
|
* setOption
|
|
*
|
|
* @param string $name option name
|
|
* @param string $value option value
|
|
* @return Doctrine_Query this object
|
|
*/
|
|
public function setOption($name, $value)
|
|
{
|
|
if ( ! isset($this->_options[$name])) {
|
|
throw new Doctrine_Query_Exception('Unknown option ' . $name);
|
|
}
|
|
$this->_options[$name] = $value;
|
|
}
|
|
/**
|
|
* addPendingJoinCondition
|
|
*
|
|
* @param string $componentAlias component alias
|
|
* @param string $joinCondition dql join condition
|
|
* @return Doctrine_Query this object
|
|
*/
|
|
public function addPendingJoinCondition($componentAlias, $joinCondition)
|
|
{
|
|
$this->_pendingJoins[$componentAlias] = $joinCondition;
|
|
}
|
|
/**
|
|
* addEnumParam
|
|
* sets input parameter as an enumerated parameter
|
|
*
|
|
* @param string $key the key of the input parameter
|
|
* @return Doctrine_Query
|
|
*/
|
|
public function addEnumParam($key, $table = null, $column = null)
|
|
{
|
|
$array = (isset($table) || isset($column)) ? array($table, $column) : array();
|
|
|
|
if ($key === '?') {
|
|
$this->_enumParams[] = $array;
|
|
} else {
|
|
$this->_enumParams[$key] = $array;
|
|
}
|
|
}
|
|
/**
|
|
* getEnumParams
|
|
* get all enumerated parameters
|
|
*
|
|
* @return array all enumerated parameters
|
|
*/
|
|
public function getEnumParams()
|
|
{
|
|
return $this->_enumParams;
|
|
}
|
|
/**
|
|
* limitSubqueryUsed
|
|
*
|
|
* @return boolean
|
|
*/
|
|
public function isLimitSubqueryUsed()
|
|
{
|
|
return $this->isLimitSubqueryUsed;
|
|
}
|
|
/**
|
|
* convertEnums
|
|
* convert enum parameters to their integer equivalents
|
|
*
|
|
* @return array converted parameter array
|
|
*/
|
|
public function convertEnums($params)
|
|
{
|
|
foreach ($this->_enumParams as $key => $values) {
|
|
if (isset($params[$key])) {
|
|
if ( ! empty($values)) {
|
|
$params[$key] = $values[0]->enumIndex($values[1], $params[$key]);
|
|
}
|
|
}
|
|
}
|
|
return $params;
|
|
}
|
|
/**
|
|
* isSubquery
|
|
* if $bool parameter is set this method sets the value of
|
|
* Doctrine_Query::$isSubquery. If this value is set to true
|
|
* the query object will not load the primary key fields of the selected
|
|
* components.
|
|
*
|
|
* If null is given as the first parameter this method retrieves the current
|
|
* value of Doctrine_Query::$isSubquery.
|
|
*
|
|
* @param boolean $bool whether or not this query acts as a subquery
|
|
* @return Doctrine_Query|bool
|
|
*/
|
|
public function isSubquery($bool = null)
|
|
{
|
|
if ($bool === null) {
|
|
return $this->isSubquery;
|
|
}
|
|
|
|
$this->isSubquery = (bool) $bool;
|
|
return $this;
|
|
}
|
|
/**
|
|
* getAggregateAlias
|
|
*
|
|
* @param string $dqlAlias the dql alias of an aggregate value
|
|
* @return string
|
|
*/
|
|
public function getAggregateAlias($dqlAlias)
|
|
{
|
|
if (isset($this->aggregateMap[$dqlAlias])) {
|
|
// mark the expression as used
|
|
$this->_expressionMap[$dqlAlias][1] = true;
|
|
|
|
return $this->aggregateMap[$dqlAlias];
|
|
}
|
|
if ( ! empty($this->pendingAggregates)) {
|
|
$this->processPendingAggregates();
|
|
|
|
return $this->getAggregateAlias($dqlAlias);
|
|
}
|
|
throw new Doctrine_Query_Exception('Unknown aggregate alias ' . $dqlAlias);
|
|
}
|
|
/**
|
|
* getParser
|
|
* parser lazy-loader
|
|
*
|
|
* @throws Doctrine_Query_Exception if unknown parser name given
|
|
* @return Doctrine_Query_Part
|
|
*/
|
|
public function getParser($name)
|
|
{
|
|
if ( ! isset($this->_parsers[$name])) {
|
|
$class = 'Doctrine_Query_' . ucwords(strtolower($name));
|
|
|
|
Doctrine::autoload($class);
|
|
|
|
if ( ! class_exists($class)) {
|
|
throw new Doctrine_Query_Exception('Unknown parser ' . $name);
|
|
}
|
|
|
|
$this->_parsers[$name] = new $class($this);
|
|
}
|
|
|
|
return $this->_parsers[$name];
|
|
}
|
|
/**
|
|
* parseQueryPart
|
|
* parses given DQL query part
|
|
*
|
|
* @param string $queryPartName the name of the query part
|
|
* @param string $queryPart query part to be parsed
|
|
* @param boolean $append whether or not to append the query part to its stack
|
|
* if false is given, this method will overwrite
|
|
* the given query part stack with $queryPart
|
|
* @return Doctrine_Query this object
|
|
*/
|
|
public function parseQueryPart($queryPartName, $queryPart, $append = false)
|
|
{
|
|
if ($this->_state === self::STATE_LOCKED) {
|
|
throw new Doctrine_Query_Exception('This query object is locked. No query parts can be manipulated.');
|
|
}
|
|
|
|
// sanity check
|
|
if ($queryPart === '' || $queryPart === null) {
|
|
throw new Doctrine_Query_Exception('Empty ' . $queryPartName . ' part given.');
|
|
}
|
|
|
|
// add query part to the dql part array
|
|
if ($append) {
|
|
$this->_dqlParts[$queryPartName][] = $queryPart;
|
|
} else {
|
|
$this->_dqlParts[$queryPartName] = array($queryPart);
|
|
}
|
|
|
|
if ($this->_state === self::STATE_DIRECT) {
|
|
$parser = $this->getParser($queryPartName);
|
|
|
|
$sql = $parser->parse($queryPart);
|
|
|
|
if (isset($sql)) {
|
|
if ($append) {
|
|
$this->addQueryPart($queryPartName, $sql);
|
|
} else {
|
|
$this->setQueryPart($queryPartName, $sql);
|
|
}
|
|
}
|
|
}
|
|
|
|
$this->_state = Doctrine_Query::STATE_DIRTY;
|
|
|
|
return $this;
|
|
}
|
|
/**
|
|
* getDqlPart
|
|
* returns the given DQL query part
|
|
*
|
|
* @param string $queryPart the name of the query part
|
|
* @return string the DQL query part
|
|
*/
|
|
public function getDqlPart($queryPart)
|
|
{
|
|
if ( ! isset($this->_dqlParts[$queryPart])) {
|
|
throw new Doctrine_Query_Exception('Unknown query part ' . $queryPart);
|
|
}
|
|
|
|
return $this->_dqlParts[$queryPart];
|
|
}
|
|
/**
|
|
* getDql
|
|
* returns the DQL query associated with this object
|
|
*
|
|
* the query is built from $_dqlParts
|
|
*
|
|
* @return string the DQL query
|
|
*/
|
|
public function getDql()
|
|
{
|
|
$q = '';
|
|
$q .= ( ! empty($this->_dqlParts['select']))? 'SELECT ' . implode(', ', $this->_dqlParts['select']) : '';
|
|
$q .= ( ! empty($this->_dqlParts['from']))? ' FROM ' . implode(' ', $this->_dqlParts['from']) : '';
|
|
$q .= ( ! empty($this->_dqlParts['where']))? ' WHERE ' . implode(' AND ', $this->_dqlParts['where']) : '';
|
|
$q .= ( ! empty($this->_dqlParts['groupby']))? ' GROUP BY ' . implode(', ', $this->_dqlParts['groupby']) : '';
|
|
$q .= ( ! empty($this->_dqlParts['having']))? ' HAVING ' . implode(' AND ', $this->_dqlParts['having']) : '';
|
|
$q .= ( ! empty($this->_dqlParts['orderby']))? ' ORDER BY ' . implode(', ', $this->_dqlParts['orderby']) : '';
|
|
$q .= ( ! empty($this->_dqlParts['limit']))? ' LIMIT ' . implode(' ', $this->_dqlParts['limit']) : '';
|
|
$q .= ( ! empty($this->_dqlParts['offset']))? ' OFFSET ' . implode(' ', $this->_dqlParts['offset']) : '';
|
|
|
|
return $q;
|
|
}
|
|
/**
|
|
* processPendingFields
|
|
* the fields in SELECT clause cannot be parsed until the components
|
|
* in FROM clause are parsed, hence this method is called everytime a
|
|
* specific component is being parsed.
|
|
*
|
|
* @throws Doctrine_Query_Exception if unknown component alias has been given
|
|
* @param string $componentAlias the alias of the component
|
|
* @return void
|
|
*/
|
|
public function processPendingFields($componentAlias)
|
|
{
|
|
$tableAlias = $this->getTableAlias($componentAlias);
|
|
$table = $this->_aliasMap[$componentAlias]['table'];
|
|
|
|
if (isset($this->pendingFields[$componentAlias])) {
|
|
$fields = $this->pendingFields[$componentAlias];
|
|
|
|
// check for wildcards
|
|
if (in_array('*', $fields)) {
|
|
$fields = $table->getColumnNames();
|
|
} else {
|
|
// only auto-add the primary key fields if this query object is not
|
|
// a subquery of another query object
|
|
if ( ! $this->isSubquery) {
|
|
$fields = array_unique(array_merge($table->getPrimaryKeys(), $fields));
|
|
}
|
|
}
|
|
}
|
|
foreach ($fields as $name) {
|
|
$name = $table->getColumnName($name);
|
|
|
|
$this->parts['select'][] = $this->_conn->quoteIdentifier($tableAlias . '.' . $name)
|
|
. ' AS '
|
|
. $this->_conn->quoteIdentifier($tableAlias . '__' . $name);
|
|
}
|
|
|
|
$this->neededTables[] = $tableAlias;
|
|
|
|
}
|
|
/**
|
|
* parseSelect
|
|
* parses the query select part and
|
|
* adds selected fields to pendingFields array
|
|
*
|
|
* @param string $dql
|
|
*/
|
|
public function parseSelect($dql)
|
|
{
|
|
$refs = Doctrine_Tokenizer::bracketExplode($dql, ',');
|
|
|
|
$pos = strpos(trim($refs[0]), ' ');
|
|
$first = substr($refs[0], 0, $pos);
|
|
|
|
if ($first === 'DISTINCT') {
|
|
$this->parts['distinct'] = true;
|
|
|
|
$refs[0] = substr($refs[0], ++$pos);
|
|
}
|
|
|
|
foreach ($refs as $reference) {
|
|
$reference = trim($reference);
|
|
if (strpos($reference, '(') !== false) {
|
|
if (substr($reference, 0, 1) === '(') {
|
|
// subselect found in SELECT part
|
|
$this->parseSubselect($reference);
|
|
} else {
|
|
$this->parseAggregateFunction($reference);
|
|
}
|
|
} else {
|
|
|
|
|
|
$e = explode('.', $reference);
|
|
if (count($e) > 2) {
|
|
$this->pendingFields[] = $reference;
|
|
} else {
|
|
$this->pendingFields[$e[0]][] = $e[1];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
/**
|
|
* parseSubselect
|
|
*
|
|
* parses the subquery found in DQL SELECT part and adds the
|
|
* parsed form into $pendingSubqueries stack
|
|
*
|
|
* @param string $reference
|
|
* @return void
|
|
*/
|
|
public function parseSubselect($reference)
|
|
{
|
|
$e = Doctrine_Tokenizer::bracketExplode($reference, ' ');
|
|
$alias = $e[1];
|
|
|
|
if (count($e) > 2) {
|
|
if (strtoupper($e[1]) !== 'AS') {
|
|
throw new Doctrine_Query_Exception('Syntax error near: ' . $reference);
|
|
}
|
|
$alias = $e[2];
|
|
}
|
|
|
|
$subquery = substr($e[0], 1, -1);
|
|
|
|
$this->pendingSubqueries[] = array($subquery, $alias);
|
|
}
|
|
/**
|
|
* parseAggregateFunction
|
|
* parses an aggregate function and returns the parsed form
|
|
*
|
|
* @see Doctrine_Expression
|
|
* @param string $expr DQL aggregate function
|
|
* @throws Doctrine_Query_Exception if unknown aggregate function given
|
|
* @return array parsed form of given function
|
|
*/
|
|
public function parseAggregateFunction($expr, $nestedCall = false)
|
|
{
|
|
$e = Doctrine_Tokenizer::bracketExplode($expr, ' ');
|
|
$func = $e[0];
|
|
|
|
$pos = strpos($func, '(');
|
|
if ($pos === false) {
|
|
return $expr;
|
|
}
|
|
|
|
// get the name of the function
|
|
$name = substr($func, 0, $pos);
|
|
$argStr = substr($func, ($pos + 1), -1);
|
|
|
|
$args = array();
|
|
// parse args
|
|
foreach (Doctrine_Tokenizer::bracketExplode($argStr, ',') as $expr) {
|
|
$args[] = $this->parseAggregateFunction($expr, true);
|
|
}
|
|
|
|
// convert DQL function to its RDBMS specific equivalent
|
|
try {
|
|
$expr = call_user_func_array(array($this->_conn->expression, $name), $args);
|
|
} catch(Doctrine_Expression_Exception $e) {
|
|
throw new Doctrine_Query_Exception('Unknown function ' . $func . '.');
|
|
}
|
|
|
|
if ( ! $nestedCall) {
|
|
// try to find all component references
|
|
preg_match_all("/[a-z0-9_]+\.[a-z0-9_]+[\.[a-z0-9]+]*/i", $argStr, $m);
|
|
|
|
if (isset($e[1])) {
|
|
if (strtoupper($e[1]) === 'AS') {
|
|
if ( ! isset($e[2])) {
|
|
throw new Doctrine_Query_Exception('Missing aggregate function alias.');
|
|
}
|
|
$alias = $e[2];
|
|
} else {
|
|
$alias = $e[1];
|
|
}
|
|
} else {
|
|
$alias = substr($expr, 0, strpos($expr, '('));
|
|
}
|
|
|
|
$this->pendingAggregates[] = array($expr, $m[0], $alias);
|
|
}
|
|
|
|
return $expr;
|
|
}
|
|
/**
|
|
* processPendingSubqueries
|
|
* processes pending subqueries
|
|
*
|
|
* subqueries can only be processed when the query is fully constructed
|
|
* since some subqueries may be correlated
|
|
*
|
|
* @return void
|
|
*/
|
|
public function processPendingSubqueries()
|
|
{
|
|
foreach ($this->pendingSubqueries as $value) {
|
|
list($dql, $alias) = $value;
|
|
|
|
$subquery = $this->createSubquery();
|
|
|
|
$sql = $subquery->parseQuery($dql, false)->getQuery();
|
|
|
|
reset($this->_aliasMap);
|
|
$componentAlias = key($this->_aliasMap);
|
|
$tableAlias = $this->getTableAlias($componentAlias);
|
|
|
|
$sqlAlias = $tableAlias . '__' . count($this->aggregateMap);
|
|
|
|
$this->parts['select'][] = '(' . $sql . ') AS ' . $this->_conn->quoteIdentifier($sqlAlias);
|
|
|
|
$this->aggregateMap[$alias] = $sqlAlias;
|
|
$this->_aliasMap[$componentAlias]['agg'][] = $alias;
|
|
}
|
|
$this->pendingSubqueries = array();
|
|
}
|
|
/**
|
|
* processPendingAggregates
|
|
* processes pending aggregate values for given component alias
|
|
*
|
|
* @return void
|
|
*/
|
|
public function processPendingAggregates()
|
|
{
|
|
// iterate trhough all aggregates
|
|
foreach ($this->pendingAggregates as $aggregate) {
|
|
list ($expression, $components, $alias) = $aggregate;
|
|
|
|
$tableAliases = array();
|
|
|
|
// iterate through the component references within the aggregate function
|
|
if ( ! empty ($components)) {
|
|
foreach ($components as $component) {
|
|
|
|
if (is_numeric($component)) {
|
|
continue;
|
|
}
|
|
|
|
$e = explode('.', $component);
|
|
|
|
$field = array_pop($e);
|
|
$componentAlias = implode('.', $e);
|
|
|
|
// check the existence of the component alias
|
|
if ( ! isset($this->_aliasMap[$componentAlias])) {
|
|
throw new Doctrine_Query_Exception('Unknown component alias ' . $componentAlias);
|
|
}
|
|
|
|
$table = $this->_aliasMap[$componentAlias]['table'];
|
|
|
|
$field = $table->getColumnName($field);
|
|
|
|
// check column existence
|
|
if ( ! $table->hasColumn($field)) {
|
|
throw new Doctrine_Query_Exception('Unknown column ' . $field);
|
|
}
|
|
|
|
$tableAlias = $this->getTableAlias($componentAlias);
|
|
|
|
$tableAliases[$tableAlias] = true;
|
|
|
|
// build sql expression
|
|
|
|
$identifier = $this->_conn->quoteIdentifier($tableAlias . '.' . $field);
|
|
$expression = str_replace($component, $identifier, $expression);
|
|
}
|
|
}
|
|
|
|
if (count($tableAliases) !== 1) {
|
|
$componentAlias = reset($this->tableAliases);
|
|
$tableAlias = key($this->tableAliases);
|
|
}
|
|
|
|
$index = count($this->aggregateMap);
|
|
$sqlAlias = $this->_conn->quoteIdentifier($tableAlias . '__' . $index);
|
|
|
|
$this->parts['select'][] = $expression . ' AS ' . $sqlAlias;
|
|
|
|
$this->aggregateMap[$alias] = $sqlAlias;
|
|
$this->_expressionMap[$alias][0] = $expression;
|
|
|
|
$this->_aliasMap[$componentAlias]['agg'][$index] = $alias;
|
|
|
|
$this->neededTables[] = $tableAlias;
|
|
}
|
|
// reset the state
|
|
$this->pendingAggregates = array();
|
|
}
|
|
/**
|
|
* getQueryBase
|
|
* returns the base of the generated sql query
|
|
* On mysql driver special strategy has to be used for DELETE statements
|
|
*
|
|
* @return string the base of the generated sql query
|
|
*/
|
|
public function getQueryBase()
|
|
{
|
|
switch ($this->type) {
|
|
case self::DELETE:
|
|
$q = 'DELETE FROM ';
|
|
break;
|
|
case self::UPDATE:
|
|
$q = 'UPDATE ';
|
|
break;
|
|
case self::SELECT:
|
|
$distinct = ($this->parts['distinct']) ? 'DISTINCT ' : '';
|
|
|
|
$q = 'SELECT ' . $distinct . implode(', ', $this->parts['select']) . ' FROM ';
|
|
break;
|
|
}
|
|
return $q;
|
|
}
|
|
/**
|
|
* buildFromPart
|
|
* builds the from part of the query and returns it
|
|
*
|
|
* @return string the query sql from part
|
|
*/
|
|
public function buildFromPart()
|
|
{
|
|
$q = '';
|
|
foreach ($this->parts['from'] as $k => $part) {
|
|
if ($k === 0) {
|
|
$q .= $part;
|
|
continue;
|
|
}
|
|
// preserve LEFT JOINs only if needed
|
|
|
|
if (substr($part, 0, 9) === 'LEFT JOIN') {
|
|
$e = explode(' ', $part);
|
|
|
|
$aliases = array_merge($this->subqueryAliases,
|
|
array_keys($this->neededTables));
|
|
|
|
if( ! in_array($e[3], $aliases) &&
|
|
! in_array($e[2], $aliases) &&
|
|
|
|
! empty($this->pendingFields)) {
|
|
continue;
|
|
}
|
|
|
|
}
|
|
|
|
if (isset($this->_pendingJoinConditions[$k])) {
|
|
$parser = new Doctrine_Query_JoinCondition($this);
|
|
|
|
if (strpos($part, ' ON ') !== false) {
|
|
$part .= ' AND ';
|
|
} else {
|
|
$part .= ' ON ';
|
|
}
|
|
$part .= $parser->parse($this->_pendingJoinConditions[$k]);
|
|
|
|
unset($this->_pendingJoinConditions[$k]);
|
|
}
|
|
|
|
$q .= ' ' . $part;
|
|
|
|
$this->parts['from'][$k] = $part;
|
|
}
|
|
return $q;
|
|
}
|
|
/**
|
|
* preQuery
|
|
*
|
|
* Empty template method to provide Query subclasses with the possibility
|
|
* to hook into the query building procedure, doing any custom / specialized
|
|
* query building procedures that are neccessary.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function preQuery()
|
|
{
|
|
|
|
}
|
|
/**
|
|
* postQuery
|
|
*
|
|
* Empty template method to provide Query subclasses with the possibility
|
|
* to hook into the query building procedure, doing any custom / specialized
|
|
* post query procedures (for example logging) that are neccessary.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function postQuery()
|
|
{
|
|
|
|
}
|
|
/**
|
|
* builds the sql query from the given parameters and applies things such as
|
|
* column aggregation inheritance and limit subqueries if needed
|
|
*
|
|
* @param array $params an array of prepared statement params (needed only in mysql driver
|
|
* when limit subquery algorithm is used)
|
|
* @return string the built sql query
|
|
*/
|
|
public function getQuery($params = array())
|
|
{
|
|
if ($this->_state !== self::STATE_DIRTY) {
|
|
return $this->_sql;
|
|
}
|
|
|
|
$parts = $this->_dqlParts;
|
|
|
|
// reset the state
|
|
if ( ! $this->isSubquery()) {
|
|
$this->_aliasMap = array();
|
|
$this->pendingAggregates = array();
|
|
$this->aggregateMap = array();
|
|
}
|
|
$this->reset();
|
|
|
|
// parse the DQL parts
|
|
foreach ($this->_dqlParts as $queryPartName => $queryParts) {
|
|
|
|
$this->removeQueryPart($queryPartName);
|
|
|
|
if (is_array($queryParts) && ! empty($queryParts)) {
|
|
|
|
foreach ($queryParts as $queryPart) {
|
|
$parser = $this->getParser($queryPartName);
|
|
|
|
|
|
$sql = $parser->parse($queryPart);
|
|
|
|
if (isset($sql)) {
|
|
if ($queryPartName == 'limit' ||
|
|
$queryPartName == 'offset') {
|
|
|
|
$this->setQueryPart($queryPartName, $sql);
|
|
} else {
|
|
$this->addQueryPart($queryPartName, $sql);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
$params = $this->convertEnums($params);
|
|
|
|
$this->_state = self::STATE_DIRECT;
|
|
|
|
// invoke the preQuery hook
|
|
$this->preQuery();
|
|
$this->_state = self::STATE_CLEAN;
|
|
|
|
$this->_dqlParts = $parts;
|
|
|
|
if (empty($this->parts['from'])) {
|
|
return false;
|
|
}
|
|
|
|
$needsSubQuery = false;
|
|
$subquery = '';
|
|
$map = reset($this->_aliasMap);
|
|
$table = $map['table'];
|
|
$rootAlias = key($this->_aliasMap);
|
|
|
|
if ( ! empty($this->parts['limit']) && $this->needsSubquery && $table->getAttribute(Doctrine::ATTR_QUERY_LIMIT) == Doctrine::LIMIT_RECORDS) {
|
|
$this->isLimitSubqueryUsed = true;
|
|
$needsSubQuery = true;
|
|
}
|
|
|
|
// process all pending SELECT part subqueries
|
|
$this->processPendingSubqueries();
|
|
$this->processPendingAggregates();
|
|
|
|
// build the basic query
|
|
|
|
$q = $this->getQueryBase();
|
|
$q .= $this->buildFromPart();
|
|
|
|
if ( ! empty($this->parts['set'])) {
|
|
$q .= ' SET ' . implode(', ', $this->parts['set']);
|
|
}
|
|
|
|
|
|
$string = $this->applyInheritance();
|
|
|
|
// apply inheritance to WHERE part
|
|
if ( ! empty($string)) {
|
|
$this->parts['where'][] = '(' . $string . ')';
|
|
}
|
|
|
|
|
|
$modifyLimit = true;
|
|
if ( ! empty($this->parts['limit']) || ! empty($this->parts['offset'])) {
|
|
|
|
if ($needsSubQuery) {
|
|
$subquery = $this->getLimitSubquery();
|
|
|
|
|
|
switch (strtolower($this->_conn->getName())) {
|
|
case 'mysql':
|
|
// mysql doesn't support LIMIT in subqueries
|
|
$list = $this->_conn->execute($subquery, $params)->fetchAll(Doctrine::FETCH_COLUMN);
|
|
$subquery = implode(', ', array_map(array($this->_conn, 'quote'), $list));
|
|
break;
|
|
case 'pgsql':
|
|
// pgsql needs special nested LIMIT subquery
|
|
$subquery = 'SELECT doctrine_subquery_alias.' . $table->getIdentifier(). ' FROM (' . $subquery . ') AS doctrine_subquery_alias';
|
|
break;
|
|
}
|
|
|
|
$field = $this->getTableAlias($rootAlias) . '.' . $table->getIdentifier();
|
|
|
|
// only append the subquery if it actually contains something
|
|
if ($subquery !== '') {
|
|
array_unshift($this->parts['where'], $this->_conn->quoteIdentifier($field) . ' IN (' . $subquery . ')');
|
|
}
|
|
|
|
$modifyLimit = false;
|
|
}
|
|
}
|
|
|
|
$q .= ( ! empty($this->parts['where']))? ' WHERE ' . implode(' AND ', $this->parts['where']) : '';
|
|
$q .= ( ! empty($this->parts['groupby']))? ' GROUP BY ' . implode(', ', $this->parts['groupby']) : '';
|
|
$q .= ( ! empty($this->parts['having']))? ' HAVING ' . implode(' AND ', $this->parts['having']): '';
|
|
$q .= ( ! empty($this->parts['orderby']))? ' ORDER BY ' . implode(', ', $this->parts['orderby']) : '';
|
|
|
|
if ($modifyLimit) {
|
|
|
|
$q = $this->_conn->modifyLimitQuery($q, $this->parts['limit'], $this->parts['offset']);
|
|
}
|
|
|
|
// return to the previous state
|
|
if ( ! empty($string)) {
|
|
array_pop($this->parts['where']);
|
|
}
|
|
if ($needsSubQuery) {
|
|
array_shift($this->parts['where']);
|
|
}
|
|
$this->_sql = $q;
|
|
|
|
return $q;
|
|
}
|
|
/**
|
|
* getLimitSubquery
|
|
* this is method is used by the record limit algorithm
|
|
*
|
|
* when fetching one-to-many, many-to-many associated data with LIMIT clause
|
|
* an additional subquery is needed for limiting the number of returned records instead
|
|
* of limiting the number of sql result set rows
|
|
*
|
|
* @return string the limit subquery
|
|
*/
|
|
public function getLimitSubquery()
|
|
{
|
|
$map = reset($this->_aliasMap);
|
|
$table = $map['table'];
|
|
$componentAlias = key($this->_aliasMap);
|
|
|
|
// get short alias
|
|
$alias = $this->getTableAlias($componentAlias);
|
|
$primaryKey = $alias . '.' . $table->getIdentifier();
|
|
|
|
// initialize the base of the subquery
|
|
$subquery = 'SELECT DISTINCT ' . $this->_conn->quoteIdentifier($primaryKey);
|
|
|
|
$driverName = $this->_conn->getAttribute(Doctrine::ATTR_DRIVER_NAME);
|
|
|
|
|
|
// pgsql needs the order by fields to be preserved in select clause
|
|
if ($driverName == 'pgsql') {
|
|
foreach ($this->parts['orderby'] as $part) {
|
|
$part = trim($part);
|
|
$e = Doctrine_Tokenizer::bracketExplode($part, ' ');
|
|
$part = trim($e[0]);
|
|
|
|
if (strpos($part, '.') === false) {
|
|
continue;
|
|
}
|
|
|
|
// don't add functions
|
|
if (strpos($part, '(') !== false) {
|
|
continue;
|
|
}
|
|
|
|
// don't add primarykey column (its already in the select clause)
|
|
if ($part !== $primaryKey) {
|
|
$subquery .= ', ' . $part;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ($driverName == 'mysql' || $driverName == 'pgsql') {
|
|
foreach ($this->_expressionMap as $dqlAlias => $expr) {
|
|
if (isset($expr[1])) {
|
|
$subquery .= ', ' . $expr[0] . ' AS ' . $this->aggregateMap[$dqlAlias];
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
$subquery .= ' FROM';
|
|
|
|
|
|
foreach ($this->parts['from'] as $part) {
|
|
// preserve LEFT JOINs only if needed
|
|
if (substr($part, 0, 9) === 'LEFT JOIN') {
|
|
$e = explode(' ', $part);
|
|
|
|
if (empty($this->parts['orderby']) && empty($this->parts['where'])) {
|
|
continue;
|
|
}
|
|
}
|
|
|
|
$subquery .= ' ' . $part;
|
|
}
|
|
|
|
// all conditions must be preserved in subquery
|
|
$subquery .= ( ! empty($this->parts['where']))? ' WHERE ' . implode(' AND ', $this->parts['where']) : '';
|
|
$subquery .= ( ! empty($this->parts['groupby']))? ' GROUP BY ' . implode(', ', $this->parts['groupby']) : '';
|
|
$subquery .= ( ! empty($this->parts['having']))? ' HAVING ' . implode(' AND ', $this->parts['having']) : '';
|
|
|
|
$subquery .= ( ! empty($this->parts['orderby']))? ' ORDER BY ' . implode(', ', $this->parts['orderby']) : '';
|
|
|
|
// add driver specific limit clause
|
|
$subquery = $this->_conn->modifyLimitQuery($subquery, $this->parts['limit'], $this->parts['offset']);
|
|
|
|
$parts = Doctrine_Tokenizer::quoteExplode($subquery, ' ', "'", "'");
|
|
|
|
foreach ($parts as $k => $part) {
|
|
if (strpos($part, ' ') !== false) {
|
|
continue;
|
|
}
|
|
|
|
$part = trim($part, "\"'`");
|
|
|
|
if ($this->hasTableAlias($part)) {
|
|
$parts[$k] = $this->_conn->quoteIdentifier($this->generateNewTableAlias($part));
|
|
continue;
|
|
}
|
|
|
|
if (strpos($part, '.') === false) {
|
|
continue;
|
|
}
|
|
preg_match_all("/[a-zA-Z0-9_]+\.[a-z0-9_]+/i", $part, $m);
|
|
|
|
foreach ($m[0] as $match) {
|
|
$e = explode('.', $match);
|
|
$e[0] = $this->generateNewTableAlias($e[0]);
|
|
|
|
$parts[$k] = str_replace($match, implode('.', $e), $parts[$k]);
|
|
}
|
|
}
|
|
|
|
if ($driverName == 'mysql' || $driverName == 'pgsql') {
|
|
foreach ($parts as $k => $part) {
|
|
if (strpos($part, "'") !== false) {
|
|
continue;
|
|
}
|
|
if (strpos($part, '__') == false) {
|
|
continue;
|
|
}
|
|
|
|
preg_match_all("/[a-zA-Z0-9_]+\_\_[a-z0-9_]+/i", $part, $m);
|
|
|
|
foreach ($m[0] as $match) {
|
|
$e = explode('__', $match);
|
|
$e[0] = $this->generateNewTableAlias($e[0]);
|
|
|
|
$parts[$k] = str_replace($match, implode('__', $e), $parts[$k]);
|
|
}
|
|
}
|
|
}
|
|
|
|
$subquery = implode(' ', $parts);
|
|
return $subquery;
|
|
}
|
|
/**
|
|
* tokenizeQuery
|
|
* splits the given dql query into an array where keys
|
|
* represent different query part names and values are
|
|
* arrays splitted using sqlExplode method
|
|
*
|
|
* example:
|
|
*
|
|
* parameter:
|
|
* $query = "SELECT u.* FROM User u WHERE u.name LIKE ?"
|
|
* returns:
|
|
* array('select' => array('u.*'),
|
|
* 'from' => array('User', 'u'),
|
|
* 'where' => array('u.name', 'LIKE', '?'))
|
|
*
|
|
* @param string $query DQL query
|
|
* @throws Doctrine_Query_Exception if some generic parsing error occurs
|
|
* @return array an array containing the query string parts
|
|
*/
|
|
public function tokenizeQuery($query)
|
|
{
|
|
$e = Doctrine_Tokenizer::sqlExplode($query, ' ');
|
|
|
|
foreach($e as $k=>$part) {
|
|
$part = trim($part);
|
|
switch(strtolower($part)) {
|
|
case 'delete':
|
|
case 'update':
|
|
case 'select':
|
|
case 'set':
|
|
case 'from':
|
|
case 'where':
|
|
case 'limit':
|
|
case 'offset':
|
|
case 'having':
|
|
$p = $part;
|
|
$parts[$part] = array();
|
|
break;
|
|
case 'order':
|
|
case 'group':
|
|
$i = ($k + 1);
|
|
if(isset($e[$i]) && strtolower($e[$i]) === "by") {
|
|
$p = $part;
|
|
$parts[$part] = array();
|
|
} else
|
|
$parts[$p][] = $part;
|
|
break;
|
|
case "by":
|
|
continue;
|
|
default:
|
|
if( ! isset($p))
|
|
throw new Doctrine_Query_Exception("Couldn't parse query.");
|
|
|
|
$parts[$p][] = $part;
|
|
}
|
|
}
|
|
return $parts;
|
|
}
|
|
/**
|
|
* DQL PARSER
|
|
* parses a DQL query
|
|
* first splits the query in parts and then uses individual
|
|
* parsers for each part
|
|
*
|
|
* @param string $query DQL query
|
|
* @param boolean $clear whether or not to clear the aliases
|
|
* @throws Doctrine_Query_Exception if some generic parsing error occurs
|
|
* @return Doctrine_Query
|
|
*/
|
|
public function parseQuery($query, $clear = true)
|
|
{
|
|
if ($clear) {
|
|
$this->clear();
|
|
}
|
|
|
|
$query = trim($query);
|
|
$query = str_replace("\n", ' ', $query);
|
|
$query = str_replace("\r", ' ', $query);
|
|
|
|
$parts = $this->tokenizeQuery($query);
|
|
|
|
foreach($parts as $k => $part) {
|
|
$part = implode(' ', $part);
|
|
$k = strtolower($k);
|
|
switch ($k) {
|
|
case 'create':
|
|
$this->type = self::CREATE;
|
|
break;
|
|
case 'insert':
|
|
$this->type = self::INSERT;
|
|
break;
|
|
case 'delete':
|
|
$this->type = self::DELETE;
|
|
break;
|
|
case 'select':
|
|
$this->type = self::SELECT;
|
|
$this->parseQueryPart($k, $part);
|
|
break;
|
|
case 'update':
|
|
$this->type = self::UPDATE;
|
|
$k = 'from';
|
|
case 'from':
|
|
$this->parseQueryPart($k, $part);
|
|
break;
|
|
case 'set':
|
|
$this->parseQueryPart($k, $part, true);
|
|
break;
|
|
case 'group':
|
|
case 'order':
|
|
$k .= 'by';
|
|
case 'where':
|
|
case 'having':
|
|
case 'limit':
|
|
case 'offset':
|
|
$this->parseQueryPart($k, $part);
|
|
break;
|
|
}
|
|
}
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function load($path, $loadFields = true)
|
|
{
|
|
// parse custom join conditions
|
|
$e = explode(' ON ', $path);
|
|
|
|
$joinCondition = '';
|
|
|
|
if (count($e) > 1) {
|
|
$joinCondition = $e[1];
|
|
$overrideJoin = true;
|
|
$path = $e[0];
|
|
} else {
|
|
$e = explode(' WITH ', $path);
|
|
|
|
if (count($e) > 1) {
|
|
$joinCondition = $e[1];
|
|
$path = $e[0];
|
|
}
|
|
$overrideJoin = false;
|
|
}
|
|
|
|
$tmp = explode(' ', $path);
|
|
$componentAlias = $originalAlias = (count($tmp) > 1) ? end($tmp) : null;
|
|
|
|
$e = preg_split("/[.:]/", $tmp[0], -1);
|
|
|
|
$fullPath = $tmp[0];
|
|
$prevPath = '';
|
|
$fullLength = strlen($fullPath);
|
|
|
|
if (isset($this->_aliasMap[$e[0]])) {
|
|
$table = $this->_aliasMap[$e[0]]['table'];
|
|
$componentAlias = $e[0];
|
|
|
|
$prevPath = $parent = array_shift($e);
|
|
}
|
|
|
|
foreach ($e as $key => $name) {
|
|
// get length of the previous path
|
|
$length = strlen($prevPath);
|
|
|
|
// build the current component path
|
|
$prevPath = ($prevPath) ? $prevPath . '.' . $name : $name;
|
|
|
|
$delimeter = substr($fullPath, $length, 1);
|
|
|
|
// if an alias is not given use the current path as an alias identifier
|
|
if (strlen($prevPath) === $fullLength && isset($originalAlias)) {
|
|
$componentAlias = $originalAlias;
|
|
} else {
|
|
$componentAlias = $prevPath;
|
|
}
|
|
|
|
// if the current alias already exists, skip it
|
|
if (isset($this->_aliasMap[$componentAlias])) {
|
|
continue;
|
|
}
|
|
|
|
if ( ! isset($table)) {
|
|
// process the root of the path
|
|
|
|
$table = $this->loadRoot($name, $componentAlias);
|
|
} else {
|
|
$join = ($delimeter == ':') ? 'INNER JOIN ' : 'LEFT JOIN ';
|
|
|
|
$relation = $table->getRelation($name);
|
|
$table = $relation->getTable();
|
|
$this->_aliasMap[$componentAlias] = array('table' => $table,
|
|
'parent' => $parent,
|
|
'relation' => $relation);
|
|
if ( ! $relation->isOneToOne()) {
|
|
$this->needsSubquery = true;
|
|
}
|
|
|
|
$localAlias = $this->getTableAlias($parent, $table->getTableName());
|
|
$foreignAlias = $this->getTableAlias($componentAlias, $relation->getTable()->getTableName());
|
|
$localSql = $this->_conn->quoteIdentifier($table->getTableName())
|
|
. ' '
|
|
. $this->_conn->quoteIdentifier($localAlias);
|
|
|
|
$foreignSql = $this->_conn->quoteIdentifier($relation->getTable()->getTableName())
|
|
. ' '
|
|
. $this->_conn->quoteIdentifier($foreignAlias);
|
|
|
|
$map = $relation->getTable()->inheritanceMap;
|
|
|
|
if ( ! $loadFields || ! empty($map) || $joinCondition) {
|
|
$this->subqueryAliases[] = $foreignAlias;
|
|
}
|
|
|
|
if ($relation instanceof Doctrine_Relation_Association) {
|
|
$asf = $relation->getAssociationFactory();
|
|
|
|
$assocTableName = $asf->getTableName();
|
|
|
|
if( ! $loadFields || ! empty($map) || $joinCondition) {
|
|
$this->subqueryAliases[] = $assocTableName;
|
|
}
|
|
|
|
$assocPath = $prevPath . '.' . $asf->getComponentName();
|
|
|
|
$assocAlias = $this->getTableAlias($assocPath, $asf->getTableName());
|
|
|
|
$queryPart = $join . $assocTableName . ' ' . $assocAlias;
|
|
|
|
|
|
$queryPart .= ' ON ' . $localAlias
|
|
. '.'
|
|
. $table->getIdentifier()
|
|
. ' = '
|
|
. $assocAlias . '.' . $relation->getLocal();
|
|
|
|
if ($relation->isEqual()) {
|
|
// equal nest relation needs additional condition
|
|
$queryPart .= ' OR ' . $localAlias
|
|
. '.'
|
|
. $table->getIdentifier()
|
|
. ' = '
|
|
. $assocAlias . '.' . $relation->getForeign();
|
|
|
|
}
|
|
|
|
$this->parts['from'][] = $queryPart;
|
|
|
|
$queryPart = $join . $foreignSql;
|
|
|
|
if ( ! $overrideJoin) {
|
|
$queryPart .= ' ON ';
|
|
|
|
if ($relation->isEqual()) {
|
|
$queryPart .= '(';
|
|
}
|
|
|
|
$queryPart .= $this->_conn->quoteIdentifier($foreignAlias . '.' . $relation->getTable()->getIdentifier())
|
|
. ' = '
|
|
. $this->_conn->quoteIdentifier($assocAlias . '.' . $relation->getForeign());
|
|
|
|
if ($relation->isEqual()) {
|
|
$queryPart .= ' OR '
|
|
. $this->_conn->quoteIdentifier($foreignAlias . '.' . $table->getIdentifier())
|
|
. ' = '
|
|
. $this->_conn->quoteIdentifier($assocAlias . '.' . $relation->getLocal())
|
|
. ') AND '
|
|
. $this->_conn->quoteIdentifier($foreignAlias . '.' . $table->getIdentifier())
|
|
. ' != '
|
|
. $this->_conn->quoteIdentifier($localAlias . '.' . $table->getIdentifier());
|
|
}
|
|
}
|
|
} else {
|
|
|
|
$queryPart = $join . $foreignSql;
|
|
|
|
if ( ! $overrideJoin) {
|
|
$queryPart .= ' ON '
|
|
. $this->_conn->quoteIdentifier($localAlias . '.' . $relation->getLocal())
|
|
. ' = '
|
|
. $this->_conn->quoteIdentifier($foreignAlias . '.' . $relation->getForeign());
|
|
}
|
|
|
|
}
|
|
$this->parts['from'][$componentAlias] = $queryPart;
|
|
if ( ! empty($joinCondition)) {
|
|
$this->_pendingJoinConditions[$componentAlias] = $joinCondition;
|
|
}
|
|
}
|
|
if ($loadFields) {
|
|
|
|
$restoreState = false;
|
|
// load fields if necessary
|
|
if ($loadFields && empty($this->pendingFields)
|
|
&& empty($this->pendingAggregates)
|
|
&& empty($this->pendingSubqueries)) {
|
|
|
|
$this->pendingFields[$componentAlias] = array('*');
|
|
|
|
$restoreState = true;
|
|
}
|
|
|
|
if(isset($this->pendingFields[$componentAlias])) {
|
|
$this->processPendingFields($componentAlias);
|
|
}
|
|
|
|
if ($restoreState) {
|
|
$this->pendingFields = array();
|
|
$this->pendingAggregates = array();
|
|
}
|
|
}
|
|
$parent = $prevPath;
|
|
}
|
|
|
|
return $this->_aliasMap[$componentAlias];
|
|
}
|
|
|
|
/**
|
|
* loadRoot
|
|
*
|
|
* @param string $name
|
|
* @param string $componentAlias
|
|
*/
|
|
public function loadRoot($name, $componentAlias)
|
|
{
|
|
// get the connection for the component
|
|
$this->_conn = Doctrine_Manager::getInstance()
|
|
->getConnectionForComponent($name);
|
|
|
|
$table = $this->_conn->getTable($name);
|
|
$tableName = $table->getTableName();
|
|
|
|
// get the short alias for this table
|
|
$tableAlias = $this->getTableAlias($componentAlias, $tableName);
|
|
// quote table name
|
|
$queryPart = $this->_conn->quoteIdentifier($tableName);
|
|
|
|
if ($this->type === self::SELECT) {
|
|
$queryPart .= ' ' . $this->_conn->quoteIdentifier($tableAlias);
|
|
}
|
|
|
|
$this->parts['from'][] = $queryPart;
|
|
$this->tableAliases[$tableAlias] = $componentAlias;
|
|
$this->_aliasMap[$componentAlias] = array('table' => $table);
|
|
|
|
return $table;
|
|
}
|
|
/**
|
|
* count
|
|
* fetches the count of the query
|
|
*
|
|
* This method executes the main query without all the
|
|
* selected fields, ORDER BY part, LIMIT part and OFFSET part.
|
|
*
|
|
* Example:
|
|
* Main query:
|
|
* SELECT u.*, p.phonenumber FROM User u
|
|
* LEFT JOIN u.Phonenumber p
|
|
* WHERE p.phonenumber = '123 123' LIMIT 10
|
|
*
|
|
* The modified DQL query:
|
|
* SELECT COUNT(DISTINCT u.id) FROM User u
|
|
* LEFT JOIN u.Phonenumber p
|
|
* WHERE p.phonenumber = '123 123'
|
|
*
|
|
* @param array $params an array of prepared statement parameters
|
|
* @return integer the count of this query
|
|
*/
|
|
public function count($params = array())
|
|
{
|
|
$this->getQuery();
|
|
|
|
// initialize temporary variables
|
|
$where = $this->parts['where'];
|
|
$having = $this->parts['having'];
|
|
$groupby = $this->parts['groupby'];
|
|
$map = reset($this->_aliasMap);
|
|
$componentAlias = key($this->_aliasMap);
|
|
$table = $map['table'];
|
|
|
|
// build the query base
|
|
$q = 'SELECT COUNT(DISTINCT ' . $this->getTableAlias($componentAlias)
|
|
. '.' . implode(',', (array) $table->getIdentifier())
|
|
. ') AS num_results';
|
|
|
|
foreach ($this->parts['select'] as $field) {
|
|
if (strpos($field, '(') !== false) {
|
|
$q .= ', ' . $field;
|
|
}
|
|
}
|
|
|
|
$q .= ' FROM ' . $this->buildFromPart();
|
|
|
|
// append column aggregation inheritance (if needed)
|
|
$string = $this->applyInheritance();
|
|
|
|
if ( ! empty($string)) {
|
|
$where[] = $string;
|
|
}
|
|
// append conditions
|
|
$q .= ( ! empty($where)) ? ' WHERE ' . implode(' AND ', $where) : '';
|
|
$q .= ( ! empty($groupby)) ? ' GROUP BY ' . implode(', ', $groupby) : '';
|
|
$q .= ( ! empty($having)) ? ' HAVING ' . implode(' AND ', $having): '';
|
|
|
|
if ( ! is_array($params)) {
|
|
$params = array($params);
|
|
}
|
|
// append parameters
|
|
$params = array_merge($this->_params, $params);
|
|
|
|
$results = $this->getConnection()->fetchAll($q, $params);
|
|
|
|
if (count($results) > 1) {
|
|
$count = 0;
|
|
foreach ($results as $result) {
|
|
$count += $result['num_results'];
|
|
}
|
|
} else {
|
|
$count = isset($results[0]) ? $results[0]['num_results']:0;
|
|
}
|
|
|
|
return (int) $count;
|
|
}
|
|
|
|
/**
|
|
* query
|
|
* query the database with DQL (Doctrine Query Language)
|
|
*
|
|
* @param string $query DQL query
|
|
* @param array $params prepared statement parameters
|
|
* @see Doctrine::FETCH_* constants
|
|
* @return mixed
|
|
*/
|
|
public function query($query, $params = array())
|
|
{
|
|
$this->parseQuery($query);
|
|
|
|
return $this->execute($params);
|
|
}
|
|
}
|