12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115 |
- <?php
- namespace Doctrine\ORM;
- use Doctrine\ORM\Query\Expr;
- class QueryBuilder
- {
-
- const SELECT = 0;
- const DELETE = 1;
- const UPDATE = 2;
-
- const STATE_DIRTY = 0;
- const STATE_CLEAN = 1;
-
- private $_em;
-
- private $_dqlParts = array(
- 'distinct' => false,
- 'select' => array(),
- 'from' => array(),
- 'join' => array(),
- 'set' => array(),
- 'where' => null,
- 'groupBy' => array(),
- 'having' => null,
- 'orderBy' => array()
- );
-
- private $_type = self::SELECT;
-
- private $_state = self::STATE_CLEAN;
-
- private $_dql;
-
- private $_params = array();
-
- private $_paramTypes = array();
-
- private $_firstResult = null;
-
- private $_maxResults = null;
-
- public function __construct(EntityManager $em)
- {
- $this->_em = $em;
- }
-
- public function expr()
- {
- return $this->_em->getExpressionBuilder();
- }
-
- public function getType()
- {
- return $this->_type;
- }
-
- public function getEntityManager()
- {
- return $this->_em;
- }
-
- public function getState()
- {
- return $this->_state;
- }
-
- 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;
- }
-
- public function getQuery()
- {
- return $this->_em->createQuery($this->getDQL())
- ->setParameters($this->_params, $this->_paramTypes)
- ->setFirstResult($this->_firstResult)
- ->setMaxResults($this->_maxResults);
- }
-
- public function getRootAlias()
- {
- $aliases = $this->getRootAliases();
- return $aliases[0];
- }
-
- 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;
- }
-
- 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;
- }
-
- public function setParameter($key, $value, $type = null)
- {
- $key = trim($key, ':');
- if ($type === null) {
- $type = Query\ParameterTypeInferer::inferType($value);
- }
- $this->_paramTypes[$key] = $type;
- $this->_params[$key] = $value;
- return $this;
- }
-
- public function setParameters(array $params, array $types = array())
- {
- foreach ($params as $key => $value) {
- $this->setParameter($key, $value, isset($types[$key]) ? $types[$key] : null);
- }
- return $this;
- }
-
- public function getParameters()
- {
- return $this->_params;
- }
-
- public function getParameter($key)
- {
- return isset($this->_params[$key]) ? $this->_params[$key] : null;
- }
-
- public function setFirstResult($firstResult)
- {
- $this->_firstResult = $firstResult;
- return $this;
- }
-
- public function getFirstResult()
- {
- return $this->_firstResult;
- }
-
- public function setMaxResults($maxResults)
- {
- $this->_maxResults = $maxResults;
- return $this;
- }
-
- public function getMaxResults()
- {
- return $this->_maxResults;
- }
-
- public function add($dqlPartName, $dqlPart, $append = false)
- {
- $isMultiple = is_array($this->_dqlParts[$dqlPartName]);
-
-
- 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;
- }
-
- 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);
- }
-
- public function distinct($flag = true)
- {
- $this->_dqlParts['distinct'] = (bool) $flag;
- return $this;
- }
-
- 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);
- }
-
- public function delete($delete = null, $alias = null)
- {
- $this->_type = self::DELETE;
- if ( ! $delete) {
- return $this;
- }
- return $this->add('from', new Expr\From($delete, $alias));
- }
-
- public function update($update = null, $alias = null)
- {
- $this->_type = self::UPDATE;
- if ( ! $update) {
- return $this;
- }
- return $this->add('from', new Expr\From($update, $alias));
- }
-
- public function from($from, $alias, $indexBy = null)
- {
- return $this->add('from', new Expr\From($from, $alias, $indexBy), true);
- }
-
- public function join($join, $alias, $conditionType = null, $condition = null, $indexBy = null)
- {
- return $this->innerJoin($join, $alias, $conditionType, $condition, $indexBy);
- }
-
- 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();
- }
- $join = new Expr\Join(
- Expr\Join::INNER_JOIN, $join, $alias, $conditionType, $condition, $indexBy
- );
- return $this->add('join', array($rootAlias => $join), true);
- }
-
- 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();
- }
- $join = new Expr\Join(
- Expr\Join::LEFT_JOIN, $join, $alias, $conditionType, $condition, $indexBy
- );
- return $this->add('join', array($rootAlias => $join), true);
- }
-
- public function set($key, $value)
- {
- return $this->add('set', new Expr\Comparison($key, Expr\Comparison::EQ, $value), true);
- }
-
- 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);
- }
-
- 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);
- }
-
- 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);
- }
-
- public function groupBy($groupBy)
- {
- return $this->add('groupBy', new Expr\GroupBy(func_get_args()));
- }
-
- public function addGroupBy($groupBy)
- {
- return $this->add('groupBy', new Expr\GroupBy(func_get_args()), true);
- }
-
- 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);
- }
-
- 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);
- }
-
- 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);
- }
-
- public function orderBy($sort, $order = null)
- {
- return $this->add('orderBy', $sort instanceof Expr\OrderBy ? $sort
- : new Expr\OrderBy($sort, $order));
- }
-
- public function addOrderBy($sort, $order = null)
- {
- return $this->add('orderBy', new Expr\OrderBy($sort, $order), true);
- }
-
- public function getDQLPart($queryPartName)
- {
- return $this->_dqlParts[$queryPartName];
- }
-
- 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->_dqlParts['distinct']===true ? ' DISTINCT' : '')
- . $this->_getReducedDQLQueryPart('select', array('pre' => ' ', 'separator' => ', '));
- $fromParts = $this->getDQLPart('from');
- $joinParts = $this->getDQLPart('join');
- $fromClauses = array();
-
- 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'] : '');
- }
-
- public function resetDQLParts($parts = null)
- {
- if (is_null($parts)) {
- $parts = array_keys($this->_dqlParts);
- }
- foreach ($parts as $part) {
- $this->resetDQLPart($part);
- }
- return $this;
- }
-
- 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;
- }
-
- public function __toString()
- {
- return $this->getDQL();
- }
-
- 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;
- }
- }
- }
- }
|