Skip to content

Instantly share code, notes, and snippets.

@kabturek
Forked from bshaffer/gist:328103
Created April 7, 2010 08:46
Show Gist options
  • Save kabturek/358673 to your computer and use it in GitHub Desktop.
Save kabturek/358673 to your computer and use it in GitHub Desktop.
<?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