12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085 |
- <?php
- /*
- * 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.doctrine-project.org>.
- */
-
- namespace Doctrine\ORM;
-
- use Doctrine\ORM\Query\Expr;
-
- /**
- * This class is responsible for building DQL query strings via an object oriented
- * PHP interface.
- *
- * @since 2.0
- * @author Guilherme Blanco <guilhermeblanco@hotmail.com>
- * @author Jonathan Wage <jonwage@gmail.com>
- * @author Roman Borschel <roman@code-factory.org>
- */
- class QueryBuilder
- {
- /* The query types. */
- const SELECT = 0;
- const DELETE = 1;
- const UPDATE = 2;
-
- /** The builder states. */
- const STATE_DIRTY = 0;
- const STATE_CLEAN = 1;
-
- /**
- * @var EntityManager The EntityManager used by this QueryBuilder.
- */
- private $_em;
-
- /**
- * @var array The array of DQL parts collected.
- */
- private $_dqlParts = array(
- 'select' => array(),
- 'from' => array(),
- 'join' => array(),
- 'set' => array(),
- 'where' => null,
- 'groupBy' => array(),
- 'having' => null,
- 'orderBy' => array()
- );
-
- /**
- * @var integer The type of query this is. Can be select, update or delete.
- */
- private $_type = self::SELECT;
-
- /**
- * @var integer The state of the query object. Can be dirty or clean.
- */
- private $_state = self::STATE_CLEAN;
-
- /**
- * @var string The complete DQL string for this query.
- */
- private $_dql;
-
- /**
- * @var array The query parameters.
- */
- private $_params = array();
-
- /**
- * @var array The parameter type map of this query.
- */
- private $_paramTypes = array();
-
- /**
- * @var integer The index of the first result to retrieve.
- */
- private $_firstResult = null;
-
- /**
- * @var integer The maximum number of results to retrieve.
- */
- private $_maxResults = null;
-
- /**
- * Initializes a new <tt>QueryBuilder</tt> that uses the given <tt>EntityManager</tt>.
- *
- * @param EntityManager $em The EntityManager to use.
- */
- public function __construct(EntityManager $em)
- {
- $this->_em = $em;
- }
-
- /**
- * Gets an ExpressionBuilder used for object-oriented construction of query expressions.
- * This producer method is intended for convenient inline usage. Example:
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->where($qb->expr()->eq('u.id', 1));
- * </code>
- *
- * For more complex expression construction, consider storing the expression
- * builder object in a local variable.
- *
- * @return Expr
- */
- public function expr()
- {
- return $this->_em->getExpressionBuilder();
- }
-
- /**
- * Get the type of the currently built query.
- *
- * @return integer
- */
- public function getType()
- {
- return $this->_type;
- }
-
- /**
- * Get the associated EntityManager for this query builder.
- *
- * @return EntityManager
- */
- public function getEntityManager()
- {
- return $this->_em;
- }
-
- /**
- * Get the state of this query builder instance.
- *
- * @return integer Either QueryBuilder::STATE_DIRTY or QueryBuilder::STATE_CLEAN.
- */
- public function getState()
- {
- return $this->_state;
- }
-
- /**
- * Get the complete DQL string formed by the current specifications of this QueryBuilder.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * echo $qb->getDql(); // SELECT u FROM User u
- * </code>
- *
- * @return string The DQL query string.
- */
- public function getDQL()
- {
- if ($this->_dql !== null && $this->_state === self::STATE_CLEAN) {
- return $this->_dql;
- }
-
- $dql = '';
-
- switch ($this->_type) {
- case self::DELETE:
- $dql = $this->_getDQLForDelete();
- break;
-
- case self::UPDATE:
- $dql = $this->_getDQLForUpdate();
- break;
-
- case self::SELECT:
- default:
- $dql = $this->_getDQLForSelect();
- break;
- }
-
- $this->_state = self::STATE_CLEAN;
- $this->_dql = $dql;
-
- return $dql;
- }
-
- /**
- * Constructs a Query instance from the current specifications of the builder.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u');
- * $q = $qb->getQuery();
- * $results = $q->execute();
- * </code>
- *
- * @return Query
- */
- public function getQuery()
- {
- return $this->_em->createQuery($this->getDQL())
- ->setParameters($this->_params, $this->_paramTypes)
- ->setFirstResult($this->_firstResult)
- ->setMaxResults($this->_maxResults);
- }
-
- /**
- * Gets the FIRST root alias of the query. This is the first entity alias involved
- * in the construction of the query.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u');
- *
- * echo $qb->getRootAlias(); // u
- * </code>
- *
- * @deprecated Please use $qb->getRootAliases() instead.
- * @return string $rootAlias
- */
- public function getRootAlias()
- {
- $aliases = $this->getRootAliases();
- return $aliases[0];
- }
-
- /**
- * Gets the root aliases of the query. This is the entity aliases involved
- * in the construction of the query.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u');
- *
- * $qb->getRootAliases(); // array('u')
- * </code>
- *
- * @return array $rootAliases
- */
- public function getRootAliases()
- {
- $aliases = array();
-
- foreach ($this->_dqlParts['from'] as &$fromClause) {
- if (is_string($fromClause)) {
- $spacePos = strrpos($fromClause, ' ');
- $from = substr($fromClause, 0, $spacePos);
- $alias = substr($fromClause, $spacePos + 1);
-
- $fromClause = new Query\Expr\From($from, $alias);
- }
-
- $aliases[] = $fromClause->getAlias();
- }
-
- return $aliases;
- }
-
- /**
- * Gets the root entities of the query. This is the entity aliases involved
- * in the construction of the query.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u');
- *
- * $qb->getRootEntities(); // array('User')
- * </code>
- *
- * @return array $rootEntities
- */
- public function getRootEntities()
- {
- $entities = array();
-
- foreach ($this->_dqlParts['from'] as &$fromClause) {
- if (is_string($fromClause)) {
- $spacePos = strrpos($fromClause, ' ');
- $from = substr($fromClause, 0, $spacePos);
- $alias = substr($fromClause, $spacePos + 1);
-
- $fromClause = new Query\Expr\From($from, $alias);
- }
-
- $entities[] = $fromClause->getFrom();
- }
-
- return $entities;
- }
-
- /**
- * Sets a query parameter for the query being constructed.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->where('u.id = :user_id')
- * ->setParameter(':user_id', 1);
- * </code>
- *
- * @param string|integer $key The parameter position or name.
- * @param mixed $value The parameter value.
- * @param string|null $type PDO::PARAM_* or \Doctrine\DBAL\Types\Type::* constant
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function setParameter($key, $value, $type = null)
- {
- if ($type === null) {
- $type = Query\ParameterTypeInferer::inferType($value);
- }
-
- $this->_paramTypes[$key] = $type;
- $this->_params[$key] = $value;
-
- return $this;
- }
-
- /**
- * Sets a collection of query parameters for the query being constructed.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->where('u.id = :user_id1 OR u.id = :user_id2')
- * ->setParameters(array(
- * ':user_id1' => 1,
- * ':user_id2' => 2
- * ));
- * </code>
- *
- * @param array $params The query parameters to set.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function setParameters(array $params, array $types = array())
- {
- foreach ($params as $key => $value) {
- if (isset($types[$key])) {
- $this->setParameter($key, $value, $types[$key]);
- } else {
- $this->setParameter($key, $value);
- }
- }
- return $this;
- }
-
- /**
- * Gets all defined query parameters for the query being constructed.
- *
- * @return array The currently defined query parameters.
- */
- public function getParameters()
- {
- return $this->_params;
- }
-
- /**
- * Gets a (previously set) query parameter of the query being constructed.
- *
- * @param mixed $key The key (index or name) of the bound parameter.
- * @return mixed The value of the bound parameter.
- */
- public function getParameter($key)
- {
- return isset($this->_params[$key]) ? $this->_params[$key] : null;
- }
-
- /**
- * Sets the position of the first result to retrieve (the "offset").
- *
- * @param integer $firstResult The first result to return.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function setFirstResult($firstResult)
- {
- $this->_firstResult = $firstResult;
- return $this;
- }
-
- /**
- * Gets the position of the first result the query object was set to retrieve (the "offset").
- * Returns NULL if {@link setFirstResult} was not applied to this QueryBuilder.
- *
- * @return integer The position of the first result.
- */
- public function getFirstResult()
- {
- return $this->_firstResult;
- }
-
- /**
- * Sets the maximum number of results to retrieve (the "limit").
- *
- * @param integer $maxResults The maximum number of results to retrieve.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function setMaxResults($maxResults)
- {
- $this->_maxResults = $maxResults;
- return $this;
- }
-
- /**
- * Gets the maximum number of results the query object was set to retrieve (the "limit").
- * Returns NULL if {@link setMaxResults} was not applied to this query builder.
- *
- * @return integer Maximum number of results.
- */
- public function getMaxResults()
- {
- return $this->_maxResults;
- }
-
- /**
- * Either appends to or replaces a single, generic query part.
- *
- * The available parts are: 'select', 'from', 'join', 'set', 'where',
- * 'groupBy', 'having' and 'orderBy'.
- *
- * @param string $dqlPartName
- * @param string $dqlPart
- * @param string $append
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function add($dqlPartName, $dqlPart, $append = false)
- {
- $isMultiple = is_array($this->_dqlParts[$dqlPartName]);
-
- // This is introduced for backwards compatibility reasons.
- // TODO: Remove for 3.0
- if ($dqlPartName == 'join') {
- $newDqlPart = array();
- foreach ($dqlPart AS $k => $v) {
- if (is_numeric($k)) {
- $newDqlPart[$this->getRootAlias()] = $v;
- } else {
- $newDqlPart[$k] = $v;
- }
- }
- $dqlPart = $newDqlPart;
- }
-
- if ($append && $isMultiple) {
- if (is_array($dqlPart)) {
- $key = key($dqlPart);
-
- $this->_dqlParts[$dqlPartName][$key][] = $dqlPart[$key];
- } else {
- $this->_dqlParts[$dqlPartName][] = $dqlPart;
- }
- } else {
- $this->_dqlParts[$dqlPartName] = ($isMultiple) ? array($dqlPart) : $dqlPart;
- }
-
- $this->_state = self::STATE_DIRTY;
-
- return $this;
- }
-
- /**
- * Specifies an item that is to be returned in the query result.
- * Replaces any previously specified selections, if any.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u', 'p')
- * ->from('User', 'u')
- * ->leftJoin('u.Phonenumbers', 'p');
- * </code>
- *
- * @param mixed $select The selection expressions.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function select($select = null)
- {
- $this->_type = self::SELECT;
-
- if (empty($select)) {
- return $this;
- }
-
- $selects = is_array($select) ? $select : func_get_args();
-
- return $this->add('select', new Expr\Select($selects), false);
- }
-
- /**
- * Adds an item that is to be returned in the query result.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->addSelect('p')
- * ->from('User', 'u')
- * ->leftJoin('u.Phonenumbers', 'p');
- * </code>
- *
- * @param mixed $select The selection expression.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function addSelect($select = null)
- {
- $this->_type = self::SELECT;
-
- if (empty($select)) {
- return $this;
- }
-
- $selects = is_array($select) ? $select : func_get_args();
-
- return $this->add('select', new Expr\Select($selects), true);
- }
-
- /**
- * Turns the query being built into a bulk delete query that ranges over
- * a certain entity type.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->delete('User', 'u')
- * ->where('u.id = :user_id');
- * ->setParameter(':user_id', 1);
- * </code>
- *
- * @param string $delete The class/type whose instances are subject to the deletion.
- * @param string $alias The class/type alias used in the constructed query.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function delete($delete = null, $alias = null)
- {
- $this->_type = self::DELETE;
-
- if ( ! $delete) {
- return $this;
- }
-
- return $this->add('from', new Expr\From($delete, $alias));
- }
-
- /**
- * Turns the query being built into a bulk update query that ranges over
- * a certain entity type.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->update('User', 'u')
- * ->set('u.password', md5('password'))
- * ->where('u.id = ?');
- * </code>
- *
- * @param string $update The class/type whose instances are subject to the update.
- * @param string $alias The class/type alias used in the constructed query.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function update($update = null, $alias = null)
- {
- $this->_type = self::UPDATE;
-
- if ( ! $update) {
- return $this;
- }
-
- return $this->add('from', new Expr\From($update, $alias));
- }
-
- /**
- * Create and add a query root corresponding to the entity identified by the given alias,
- * forming a cartesian product with any existing query roots.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * </code>
- *
- * @param string $from The class name.
- * @param string $alias The alias of the class.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function from($from, $alias)
- {
- return $this->add('from', new Expr\From($from, $alias), true);
- }
-
- /**
- * Creates and adds a join over an entity association to the query.
- *
- * The entities in the joined association will be fetched as part of the query
- * result if the alias used for the joined association is placed in the select
- * expressions.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->join('u.Phonenumbers', 'p', Expr\Join::WITH, 'p.is_primary = 1');
- * </code>
- *
- * @param string $join The relationship to join
- * @param string $alias The alias of the join
- * @param string $conditionType The condition type constant. Either ON or WITH.
- * @param string $condition The condition for the join
- * @param string $indexBy The index for the join
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function join($join, $alias, $conditionType = null, $condition = null, $indexBy = null)
- {
- return $this->innerJoin($join, $alias, $conditionType, $condition, $indexBy);
- }
-
- /**
- * Creates and adds a join over an entity association to the query.
- *
- * The entities in the joined association will be fetched as part of the query
- * result if the alias used for the joined association is placed in the select
- * expressions.
- *
- * [php]
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->innerJoin('u.Phonenumbers', 'p', Expr\Join::WITH, 'p.is_primary = 1');
- *
- * @param string $join The relationship to join
- * @param string $alias The alias of the join
- * @param string $conditionType The condition type constant. Either ON or WITH.
- * @param string $condition The condition for the join
- * @param string $indexBy The index for the join
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function innerJoin($join, $alias, $conditionType = null, $condition = null, $indexBy = null)
- {
- $rootAlias = substr($join, 0, strpos($join, '.'));
- if (!in_array($rootAlias, $this->getRootAliases())) {
- $rootAlias = $this->getRootAlias();
- }
-
- return $this->add('join', array(
- $rootAlias => new Expr\Join(Expr\Join::INNER_JOIN, $join, $alias, $conditionType, $condition, $indexBy)
- ), true);
- }
-
- /**
- * Creates and adds a left join over an entity association to the query.
- *
- * The entities in the joined association will be fetched as part of the query
- * result if the alias used for the joined association is placed in the select
- * expressions.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->leftJoin('u.Phonenumbers', 'p', Expr\Join::WITH, 'p.is_primary = 1');
- * </code>
- *
- * @param string $join The relationship to join
- * @param string $alias The alias of the join
- * @param string $conditionType The condition type constant. Either ON or WITH.
- * @param string $condition The condition for the join
- * @param string $indexBy The index for the join
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function leftJoin($join, $alias, $conditionType = null, $condition = null, $indexBy = null)
- {
- $rootAlias = substr($join, 0, strpos($join, '.'));
- if (!in_array($rootAlias, $this->getRootAliases())) {
- $rootAlias = $this->getRootAlias();
- }
-
- return $this->add('join', array(
- $rootAlias => new Expr\Join(Expr\Join::LEFT_JOIN, $join, $alias, $conditionType, $condition, $indexBy)
- ), true);
- }
-
- /**
- * Sets a new value for a field in a bulk update query.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->update('User', 'u')
- * ->set('u.password', md5('password'))
- * ->where('u.id = ?');
- * </code>
- *
- * @param string $key The key/field to set.
- * @param string $value The value, expression, placeholder, etc.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function set($key, $value)
- {
- return $this->add('set', new Expr\Comparison($key, Expr\Comparison::EQ, $value), true);
- }
-
- /**
- * Specifies one or more restrictions to the query result.
- * Replaces any previously specified restrictions, if any.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->where('u.id = ?');
- *
- * // You can optionally programatically build and/or expressions
- * $qb = $em->createQueryBuilder();
- *
- * $or = $qb->expr()->orx();
- * $or->add($qb->expr()->eq('u.id', 1));
- * $or->add($qb->expr()->eq('u.id', 2));
- *
- * $qb->update('User', 'u')
- * ->set('u.password', md5('password'))
- * ->where($or);
- * </code>
- *
- * @param mixed $predicates The restriction predicates.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function where($predicates)
- {
- if ( ! (func_num_args() == 1 && $predicates instanceof Expr\Composite)) {
- $predicates = new Expr\Andx(func_get_args());
- }
-
- return $this->add('where', $predicates);
- }
-
- /**
- * Adds one or more restrictions to the query results, forming a logical
- * conjunction with any previously specified restrictions.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->where('u.username LIKE ?')
- * ->andWhere('u.is_active = 1');
- * </code>
- *
- * @param mixed $where The query restrictions.
- * @return QueryBuilder This QueryBuilder instance.
- * @see where()
- */
- public function andWhere($where)
- {
- $where = $this->getDQLPart('where');
- $args = func_get_args();
-
- if ($where instanceof Expr\Andx) {
- $where->addMultiple($args);
- } else {
- array_unshift($args, $where);
- $where = new Expr\Andx($args);
- }
-
- return $this->add('where', $where, true);
- }
-
- /**
- * Adds one or more restrictions to the query results, forming a logical
- * disjunction with any previously specified restrictions.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->where('u.id = 1')
- * ->orWhere('u.id = 2');
- * </code>
- *
- * @param mixed $where The WHERE statement
- * @return QueryBuilder $qb
- * @see where()
- */
- public function orWhere($where)
- {
- $where = $this->getDqlPart('where');
- $args = func_get_args();
-
- if ($where instanceof Expr\Orx) {
- $where->addMultiple($args);
- } else {
- array_unshift($args, $where);
- $where = new Expr\Orx($args);
- }
-
- return $this->add('where', $where, true);
- }
-
- /**
- * Specifies a grouping over the results of the query.
- * Replaces any previously specified groupings, if any.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->groupBy('u.id');
- * </code>
- *
- * @param string $groupBy The grouping expression.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function groupBy($groupBy)
- {
- return $this->add('groupBy', new Expr\GroupBy(func_get_args()));
- }
-
-
- /**
- * Adds a grouping expression to the query.
- *
- * <code>
- * $qb = $em->createQueryBuilder()
- * ->select('u')
- * ->from('User', 'u')
- * ->groupBy('u.lastLogin');
- * ->addGroupBy('u.createdAt')
- * </code>
- *
- * @param string $groupBy The grouping expression.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function addGroupBy($groupBy)
- {
- return $this->add('groupBy', new Expr\GroupBy(func_get_args()), true);
- }
-
- /**
- * Specifies a restriction over the groups of the query.
- * Replaces any previous having restrictions, if any.
- *
- * @param mixed $having The restriction over the groups.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function having($having)
- {
- if ( ! (func_num_args() == 1 && ($having instanceof Expr\Andx || $having instanceof Expr\Orx))) {
- $having = new Expr\Andx(func_get_args());
- }
-
- return $this->add('having', $having);
- }
-
- /**
- * Adds a restriction over the groups of the query, forming a logical
- * conjunction with any existing having restrictions.
- *
- * @param mixed $having The restriction to append.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function andHaving($having)
- {
- $having = $this->getDqlPart('having');
- $args = func_get_args();
-
- if ($having instanceof Expr\Andx) {
- $having->addMultiple($args);
- } else {
- array_unshift($args, $having);
- $having = new Expr\Andx($args);
- }
-
- return $this->add('having', $having);
- }
-
- /**
- * Adds a restriction over the groups of the query, forming a logical
- * disjunction with any existing having restrictions.
- *
- * @param mixed $having The restriction to add.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function orHaving($having)
- {
- $having = $this->getDqlPart('having');
- $args = func_get_args();
-
- if ($having instanceof Expr\Orx) {
- $having->addMultiple($args);
- } else {
- array_unshift($args, $having);
- $having = new Expr\Orx($args);
- }
-
- return $this->add('having', $having);
- }
-
- /**
- * Specifies an ordering for the query results.
- * Replaces any previously specified orderings, if any.
- *
- * @param string $sort The ordering expression.
- * @param string $order The ordering direction.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function orderBy($sort, $order = null)
- {
- return $this->add('orderBy', $sort instanceof Expr\OrderBy ? $sort
- : new Expr\OrderBy($sort, $order));
- }
-
- /**
- * Adds an ordering to the query results.
- *
- * @param string $sort The ordering expression.
- * @param string $order The ordering direction.
- * @return QueryBuilder This QueryBuilder instance.
- */
- public function addOrderBy($sort, $order = null)
- {
- return $this->add('orderBy', new Expr\OrderBy($sort, $order), true);
- }
-
- /**
- * Get a query part by its name.
- *
- * @param string $queryPartName
- * @return mixed $queryPart
- * @todo Rename: getQueryPart (or remove?)
- */
- public function getDQLPart($queryPartName)
- {
- return $this->_dqlParts[$queryPartName];
- }
-
- /**
- * Get all query parts.
- *
- * @return array $dqlParts
- * @todo Rename: getQueryParts (or remove?)
- */
- public function getDQLParts()
- {
- return $this->_dqlParts;
- }
-
- private function _getDQLForDelete()
- {
- return 'DELETE'
- . $this->_getReducedDQLQueryPart('from', array('pre' => ' ', 'separator' => ', '))
- . $this->_getReducedDQLQueryPart('where', array('pre' => ' WHERE '))
- . $this->_getReducedDQLQueryPart('orderBy', array('pre' => ' ORDER BY ', 'separator' => ', '));
- }
-
- private function _getDQLForUpdate()
- {
- return 'UPDATE'
- . $this->_getReducedDQLQueryPart('from', array('pre' => ' ', 'separator' => ', '))
- . $this->_getReducedDQLQueryPart('set', array('pre' => ' SET ', 'separator' => ', '))
- . $this->_getReducedDQLQueryPart('where', array('pre' => ' WHERE '))
- . $this->_getReducedDQLQueryPart('orderBy', array('pre' => ' ORDER BY ', 'separator' => ', '));
- }
-
- private function _getDQLForSelect()
- {
- $dql = 'SELECT' . $this->_getReducedDQLQueryPart('select', array('pre' => ' ', 'separator' => ', '));
-
- $fromParts = $this->getDQLPart('from');
- $joinParts = $this->getDQLPart('join');
- $fromClauses = array();
-
- // Loop through all FROM clauses
- if ( ! empty($fromParts)) {
- $dql .= ' FROM ';
-
- foreach ($fromParts as $from) {
- $fromClause = (string) $from;
-
- if ($from instanceof Expr\From && isset($joinParts[$from->getAlias()])) {
- foreach ($joinParts[$from->getAlias()] as $join) {
- $fromClause .= ' ' . ((string) $join);
- }
- }
-
- $fromClauses[] = $fromClause;
- }
- }
-
- $dql .= implode(', ', $fromClauses)
- . $this->_getReducedDQLQueryPart('where', array('pre' => ' WHERE '))
- . $this->_getReducedDQLQueryPart('groupBy', array('pre' => ' GROUP BY ', 'separator' => ', '))
- . $this->_getReducedDQLQueryPart('having', array('pre' => ' HAVING '))
- . $this->_getReducedDQLQueryPart('orderBy', array('pre' => ' ORDER BY ', 'separator' => ', '));
-
- return $dql;
- }
-
- private function _getReducedDQLQueryPart($queryPartName, $options = array())
- {
- $queryPart = $this->getDQLPart($queryPartName);
-
- if (empty($queryPart)) {
- return (isset($options['empty']) ? $options['empty'] : '');
- }
-
- return (isset($options['pre']) ? $options['pre'] : '')
- . (is_array($queryPart) ? implode($options['separator'], $queryPart) : $queryPart)
- . (isset($options['post']) ? $options['post'] : '');
- }
-
- /**
- * Reset DQL parts
- *
- * @param array $parts
- * @return QueryBuilder
- */
- public function resetDQLParts($parts = null)
- {
- if (is_null($parts)) {
- $parts = array_keys($this->_dqlParts);
- }
- foreach ($parts as $part) {
- $this->resetDQLPart($part);
- }
- return $this;
- }
-
- /**
- * Reset single DQL part
- *
- * @param string $part
- * @return QueryBuilder;
- */
- public function resetDQLPart($part)
- {
- if (is_array($this->_dqlParts[$part])) {
- $this->_dqlParts[$part] = array();
- } else {
- $this->_dqlParts[$part] = null;
- }
- $this->_state = self::STATE_DIRTY;
- return $this;
- }
-
- /**
- * Gets a string representation of this QueryBuilder which corresponds to
- * the final DQL query being constructed.
- *
- * @return string The string representation of this QueryBuilder.
- */
- public function __toString()
- {
- return $this->getDQL();
- }
-
- /**
- * Deep clone of all expression objects in the DQL parts.
- *
- * @return void
- */
- public function __clone()
- {
- foreach ($this->_dqlParts AS $part => $elements) {
- if (is_array($this->_dqlParts[$part])) {
- foreach ($this->_dqlParts[$part] AS $idx => $element) {
- if (is_object($element)) {
- $this->_dqlParts[$part][$idx] = clone $element;
- }
- }
- } else if (\is_object($elements)) {
- $this->_dqlParts[$part] = clone $elements;
- }
- }
- }
- }
|