12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091 |
- <?php
-
-
- namespace Doctrine\DBAL\Query;
-
- use Doctrine\DBAL\Query\Expression\CompositeExpression,
- Doctrine\DBAL\Connection;
-
-
- class QueryBuilder
- {
-
- const SELECT = 0;
- const DELETE = 1;
- const UPDATE = 2;
-
-
- const STATE_DIRTY = 0;
- const STATE_CLEAN = 1;
-
-
-
- private $connection = null;
-
-
-
- private $sqlParts = array(
- 'select' => array(),
- 'from' => array(),
- 'join' => array(),
- 'set' => array(),
- 'where' => null,
- 'groupBy' => array(),
- 'having' => null,
- 'orderBy' => array()
- );
-
-
-
- private $sql;
-
-
-
- private $params = array();
-
-
-
- private $paramTypes = array();
-
-
-
- private $type = self::SELECT;
-
-
-
- private $state = self::STATE_CLEAN;
-
-
-
- private $firstResult = null;
-
-
-
- private $maxResults = null;
-
-
-
- private $boundCounter = 0;
-
-
-
- public function __construct(Connection $connection)
- {
- $this->connection = $connection;
- }
-
-
-
- public function expr()
- {
- return $this->connection->getExpressionBuilder();
- }
-
-
-
- public function getType()
- {
- return $this->type;
- }
-
-
-
- public function getConnection()
- {
- return $this->connection;
- }
-
-
-
- public function getState()
- {
- return $this->state;
- }
-
-
-
- public function execute()
- {
- if ($this->type == self::SELECT) {
- return $this->connection->executeQuery($this->getSQL(), $this->params, $this->paramTypes);
- } else {
- return $this->connection->executeUpdate($this->getSQL(), $this->params, $this->paramTypes);
- }
- }
-
-
-
- public function getSQL()
- {
- if ($this->sql !== null && $this->state === self::STATE_CLEAN) {
- return $this->sql;
- }
-
- $sql = '';
-
- switch ($this->type) {
- case self::DELETE:
- $sql = $this->getSQLForDelete();
- break;
-
- case self::UPDATE:
- $sql = $this->getSQLForUpdate();
- break;
-
- case self::SELECT:
- default:
- $sql = $this->getSQLForSelect();
- break;
- }
-
- $this->state = self::STATE_CLEAN;
- $this->sql = $sql;
-
- return $sql;
- }
-
-
-
- public function setParameter($key, $value, $type = null)
- {
- if ($type !== null) {
- $this->paramTypes[$key] = $type;
- }
-
- $this->params[$key] = $value;
-
- return $this;
- }
-
-
-
- public function setParameters(array $params, array $types = array())
- {
- $this->paramTypes = $types;
- $this->params = $params;
-
- 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->state = self::STATE_DIRTY;
- $this->firstResult = $firstResult;
- return $this;
- }
-
-
-
- public function getFirstResult()
- {
- return $this->firstResult;
- }
-
-
-
- public function setMaxResults($maxResults)
- {
- $this->state = self::STATE_DIRTY;
- $this->maxResults = $maxResults;
- return $this;
- }
-
-
-
- public function getMaxResults()
- {
- return $this->maxResults;
- }
-
-
-
- public function add($sqlPartName, $sqlPart, $append = false)
- {
- $isArray = is_array($sqlPart);
- $isMultiple = is_array($this->sqlParts[$sqlPartName]);
-
- if ($isMultiple && !$isArray) {
- $sqlPart = array($sqlPart);
- }
-
- $this->state = self::STATE_DIRTY;
-
- if ($append) {
- if ($sqlPartName == "orderBy" || $sqlPartName == "groupBy" || $sqlPartName == "select" || $sqlPartName == "set") {
- foreach ($sqlPart AS $part) {
- $this->sqlParts[$sqlPartName][] = $part;
- }
- } else if ($isArray && is_array($sqlPart[key($sqlPart)])) {
- $key = key($sqlPart);
- $this->sqlParts[$sqlPartName][$key][] = $sqlPart[$key];
- } else if ($isMultiple) {
- $this->sqlParts[$sqlPartName][] = $sqlPart;
- } else {
- $this->sqlParts[$sqlPartName] = $sqlPart;
- }
-
- return $this;
- }
-
- $this->sqlParts[$sqlPartName] = $sqlPart;
-
- 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', $selects, false);
- }
-
-
-
- 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', $selects, true);
- }
-
-
-
- public function delete($delete = null, $alias = null)
- {
- $this->type = self::DELETE;
-
- if ( ! $delete) {
- return $this;
- }
-
- return $this->add('from', array(
- 'table' => $delete,
- 'alias' => $alias
- ));
- }
-
-
-
- public function update($update = null, $alias = null)
- {
- $this->type = self::UPDATE;
-
- if ( ! $update) {
- return $this;
- }
-
- return $this->add('from', array(
- 'table' => $update,
- 'alias' => $alias
- ));
- }
-
-
-
- public function from($from, $alias)
- {
- return $this->add('from', array(
- 'table' => $from,
- 'alias' => $alias
- ), true);
- }
-
-
-
- public function join($fromAlias, $join, $alias, $condition = null)
- {
- return $this->innerJoin($fromAlias, $join, $alias, $condition);
- }
-
-
-
- public function innerJoin($fromAlias, $join, $alias, $condition = null)
- {
- return $this->add('join', array(
- $fromAlias => array(
- 'joinType' => 'inner',
- 'joinTable' => $join,
- 'joinAlias' => $alias,
- 'joinCondition' => $condition
- )
- ), true);
- }
-
-
-
- public function leftJoin($fromAlias, $join, $alias, $condition = null)
- {
- return $this->add('join', array(
- $fromAlias => array(
- 'joinType' => 'left',
- 'joinTable' => $join,
- 'joinAlias' => $alias,
- 'joinCondition' => $condition
- )
- ), true);
- }
-
-
-
- public function rightJoin($fromAlias, $join, $alias, $condition = null)
- {
- return $this->add('join', array(
- $fromAlias => array(
- 'joinType' => 'right',
- 'joinTable' => $join,
- 'joinAlias' => $alias,
- 'joinCondition' => $condition
- )
- ), true);
- }
-
-
-
- public function set($key, $value)
- {
- return $this->add('set', $key .' = ' . $value, true);
- }
-
-
-
- public function where($predicates)
- {
- if ( ! (func_num_args() == 1 && $predicates instanceof CompositeExpression) ) {
- $predicates = new CompositeExpression(CompositeExpression::TYPE_AND, func_get_args());
- }
-
- return $this->add('where', $predicates);
- }
-
-
-
- public function andWhere($where)
- {
- $where = $this->getQueryPart('where');
- $args = func_get_args();
-
- if ($where instanceof CompositeExpression && $where->getType() === CompositeExpression::TYPE_AND) {
- $where->addMultiple($args);
- } else {
- array_unshift($args, $where);
- $where = new CompositeExpression(CompositeExpression::TYPE_AND, $args);
- }
-
- return $this->add('where', $where, true);
- }
-
-
-
- public function orWhere($where)
- {
- $where = $this->getQueryPart('where');
- $args = func_get_args();
-
- if ($where instanceof CompositeExpression && $where->getType() === CompositeExpression::TYPE_OR) {
- $where->addMultiple($args);
- } else {
- array_unshift($args, $where);
- $where = new CompositeExpression(CompositeExpression::TYPE_OR, $args);
- }
-
- return $this->add('where', $where, true);
- }
-
-
-
- public function groupBy($groupBy)
- {
- if (empty($groupBy)) {
- return $this;
- }
-
- $groupBy = is_array($groupBy) ? $groupBy : func_get_args();
-
- return $this->add('groupBy', $groupBy, false);
- }
-
-
-
-
- public function addGroupBy($groupBy)
- {
- if (empty($groupBy)) {
- return $this;
- }
-
- $groupBy = is_array($groupBy) ? $groupBy : func_get_args();
-
- return $this->add('groupBy', $groupBy, true);
- }
-
-
-
- public function having($having)
- {
- if ( ! (func_num_args() == 1 && $having instanceof CompositeExpression)) {
- $having = new CompositeExpression(CompositeExpression::TYPE_AND, func_get_args());
- }
-
- return $this->add('having', $having);
- }
-
-
-
- public function andHaving($having)
- {
- $having = $this->getQueryPart('having');
- $args = func_get_args();
-
- if ($having instanceof CompositeExpression && $having->getType() === CompositeExpression::TYPE_AND) {
- $having->addMultiple($args);
- } else {
- array_unshift($args, $having);
- $having = new CompositeExpression(CompositeExpression::TYPE_AND, $args);
- }
-
- return $this->add('having', $having);
- }
-
-
-
- public function orHaving($having)
- {
- $having = $this->getQueryPart('having');
- $args = func_get_args();
-
- if ($having instanceof CompositeExpression && $having->getType() === CompositeExpression::TYPE_OR) {
- $having->addMultiple($args);
- } else {
- array_unshift($args, $having);
- $having = new CompositeExpression(CompositeExpression::TYPE_OR, $args);
- }
-
- return $this->add('having', $having);
- }
-
-
-
- public function orderBy($sort, $order = null)
- {
- return $this->add('orderBy', $sort . ' ' . (! $order ? 'ASC' : $order), false);
- }
-
-
-
- public function addOrderBy($sort, $order = null)
- {
- return $this->add('orderBy', $sort . ' ' . (! $order ? 'ASC' : $order), true);
- }
-
-
-
- public function getQueryPart($queryPartName)
- {
- return $this->sqlParts[$queryPartName];
- }
-
-
-
- public function getQueryParts()
- {
- return $this->sqlParts;
- }
-
-
-
- public function resetQueryParts($queryPartNames = null)
- {
- if (is_null($queryPartNames)) {
- $queryPartNames = array_keys($this->sqlParts);
- }
-
- foreach ($queryPartNames as $queryPartName) {
- $this->resetQueryPart($queryPartName);
- }
-
- return $this;
- }
-
-
-
- public function resetQueryPart($queryPartName)
- {
- $this->sqlParts[$queryPartName] = is_array($this->sqlParts[$queryPartName])
- ? array() : null;
-
- $this->state = self::STATE_DIRTY;
-
- return $this;
- }
-
-
-
- private function getSQLForSelect()
- {
- $query = 'SELECT ' . implode(', ', $this->sqlParts['select']) . ' FROM ';
-
- $fromClauses = array();
-
-
- foreach ($this->sqlParts['from'] as $from) {
- $fromClause = $from['table'] . ' ' . $from['alias'];
-
- if (isset($this->sqlParts['join'][$from['alias']])) {
- foreach ($this->sqlParts['join'][$from['alias']] as $join) {
- $fromClause .= ' ' . strtoupper($join['joinType'])
- . ' JOIN ' . $join['joinTable'] . ' ' . $join['joinAlias']
- . ' ON ' . ((string) $join['joinCondition']);
- }
- }
-
- $fromClauses[$from['alias']] = $fromClause;
- }
-
-
- foreach ($this->sqlParts['join'] as $fromAlias => $joins) {
- if ( ! isset($fromClauses[$fromAlias]) ) {
- throw QueryException::unknownFromAlias($fromAlias, array_keys($fromClauses));
- }
- }
-
- $query .= implode(', ', $fromClauses)
- . ($this->sqlParts['where'] !== null ? ' WHERE ' . ((string) $this->sqlParts['where']) : '')
- . ($this->sqlParts['groupBy'] ? ' GROUP BY ' . implode(', ', $this->sqlParts['groupBy']) : '')
- . ($this->sqlParts['having'] !== null ? ' HAVING ' . ((string) $this->sqlParts['having']) : '')
- . ($this->sqlParts['orderBy'] ? ' ORDER BY ' . implode(', ', $this->sqlParts['orderBy']) : '');
-
- return ($this->maxResults === null && $this->firstResult == null)
- ? $query
- : $this->connection->getDatabasePlatform()->modifyLimitQuery($query, $this->maxResults, $this->firstResult);
- }
-
-
-
- private function getSQLForUpdate()
- {
- $table = $this->sqlParts['from']['table'] . ($this->sqlParts['from']['alias'] ? ' ' . $this->sqlParts['from']['alias'] : '');
- $query = 'UPDATE ' . $table
- . ' SET ' . implode(", ", $this->sqlParts['set'])
- . ($this->sqlParts['where'] !== null ? ' WHERE ' . ((string) $this->sqlParts['where']) : '');
-
- return $query;
- }
-
-
-
- private function getSQLForDelete()
- {
- $table = $this->sqlParts['from']['table'] . ($this->sqlParts['from']['alias'] ? ' ' . $this->sqlParts['from']['alias'] : '');
- $query = 'DELETE FROM ' . $table . ($this->sqlParts['where'] !== null ? ' WHERE ' . ((string) $this->sqlParts['where']) : '');
-
- return $query;
- }
-
-
-
- public function __toString()
- {
- return $this->getSQL();
- }
-
-
-
- public function createNamedParameter( $value, $type = \PDO::PARAM_STR, $placeHolder = null )
- {
- if ( $placeHolder === null ) {
- $this->boundCounter++;
- $placeHolder = ":dcValue" . $this->boundCounter;
- }
- $this->setParameter(substr($placeHolder, 1), $value, $type);
-
- return $placeHolder;
- }
-
-
-
- public function createPositionalParameter($value, $type = \PDO::PARAM_STR)
- {
- $this->boundCounter++;
- $this->setParameter($this->boundCounter, $value, $type);
- return "?";
- }
- }
|