-
-
Save kabturek/358673 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
<?php | |
/* | |
* Extended Doctrine Query class providing a few additional functions | |
* for wrapping your where clauses more efficiently | |
*/ | |
class Doctrine_Query_Extra extends Doctrine_Query | |
{ | |
protected $_startClause = false; | |
/** | |
* This function begins an AND clause wrapped in parenthesis | |
* Requires a call to endClause() | |
* | |
* @return $this | |
*/ | |
public function andClause() | |
{ | |
if ($this->_hasDqlQueryPart('where')) { | |
$this->_addDqlQueryPart('where', 'AND', true); | |
} | |
$this->_addDqlQueryPart('where', '(', true); | |
$this->_startClause = true; | |
return $this; | |
} | |
/** | |
* This function begins an OR clause wrapped in parenthesis. | |
* Requires a call to endClause() | |
* | |
* @return $this | |
*/ | |
public function orClause() | |
{ | |
if ($this->_hasDqlQueryPart('where')) { | |
$this->_addDqlQueryPart('where', 'OR', true); | |
} | |
$this->_addDqlQueryPart('where', '(', true); | |
$this->_startClause = true; | |
return $this; | |
} | |
/** | |
* This function ends a clause | |
* | |
* @return $this | |
*/ | |
public function endClause() | |
{ | |
if ($this->_startClause) | |
{ | |
$this->_startClause = false; | |
// Remove last two elements (open parenthesis and the "AND or OR" before it) | |
array_pop($this->_dqlParts['where']); | |
array_pop($this->_dqlParts['where']); | |
} | |
else | |
{ | |
$this->_addDqlQueryPart('where', ')', true); | |
} | |
return $this; | |
} | |
protected function _addDqlQueryPart($queryPartName, $queryPart, $append = false) | |
{ | |
if ($queryPartName == 'where' && $this->_startClause && ($queryPart == 'AND' || $queryPart == 'OR')) | |
{ | |
$this->_startClause = false; | |
return $this; | |
} | |
return parent::_addDqlQueryPart($queryPartName, $queryPart, $append); | |
} | |
/** | |
* This function will wrap the current dql where statement in a clause | |
* | |
* @return $this | |
*/ | |
public function whereWrap() | |
{ | |
$where = $this->_dqlParts['where']; | |
if (count($where) > 0) | |
{ | |
array_unshift($where, '('); | |
array_push($where, ')'); | |
$this->_dqlParts['where'] = $where; | |
} | |
return $this; | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment