Custom Zend_Paginator_Adapter_Doctrine

Source Code:

<?php
require_once ('Zend/Paginator/Adapter/Interface.php');
/**
* @author uommo
* @link http://code.google.com/p/smart-framework/source/browse/trunk/library/SmartL/Zend/Paginator/Adapter/Doctrine.php?r=67
*/
class Zend_Paginator_Adapter_Doctrine implements Zend_Paginator_Adapter_Interface
{
/**
* Doctrine's selection for paginator.
*
* @var Doctrine_Query
*/
private $_query;
/**
* Doctrine's selection for count of items.
*
* @var Doctrine_Query
*/
private $_countQuery = null;
/**
* Hydration mode for doctrine select query.
*
* @var integer
*/
private $_hydrationMode = null;

/**
* Constructs SmartL_Zend_Paginator_Doctrine
*
* @param Doctrine_Query|string $query
* @param integer $hydratationMode Use constaints Doctrine::HYDRATE_*.
* @param array[string]=>mixed $options Options may be:
* 'countQuery'-custom query for count counting. Dql or Doctrine_Query instance.
*/
public function __construct($query, $hydrationMode = null, $options = array())
{
if ( is_string($query) ) {
$newQuery = new Doctrine_Query();
$newQuery->parseDqlQuery($query);
$query = $newQuery;
}
else if ( ! ($query instanceof Doctrine_Query) ) {
require_once 'Zend/Paginator/Exception.php';
throw new Zend_Paginator_Exception("Given query is not instance of Doctrine_Query");
}
$this->_query = $query;
$this->_hydrationMode = $hydrationMode;

//options
if ( !empty($options['countQuery']) ) {
if ( is_string($options['countQuery']) ) {
$countQuery = new Doctrine_Query();
$countQuery->parseDqlQuery($options['countQuery']);
$options['countQuery'] = $countQuery;
}
else if ( ! ($options['countQuery'] instanceof Doctrine_Query) ) {
require_once 'Zend/Paginator/Exception.php';
throw new Zend_Paginator_Exception("Given count-query is not instance of Doctrine_Query");
}
$this->_countQuery = $options['countQuery'];
$this->_countQuery->select('count(*) as count');
}
}

/**
* Returns query for count of items.
*
* @return Doctrine_Query
*/
protected function getCountQuery()
{
if ( $this->_countQuery == null ) {
$this->_countQuery = $this->_query->copy();
$this->_countQuery->select('count(*) as count');

$partsToBeRemoved = array('offset','limit','orderby');
foreach( $partsToBeRemoved as $part ) {
$this->_countQuery->removeDqlQueryPart($part);
$this->_countQuery->removeSqlQueryPart($part);
}
}
return $this->_countQuery;
}

/**
* Implementation of method from Zend_Paginator_Adapter_Interface.
*
* @return integer
*/
public function count()
{
$result = $this->getCountQuery()->execute(array(), Doctrine::HYDRATE_ARRAY);
return $result[0]['count'];
}

/**
* Implementation of method from Zend_Paginator_Adapter_Interface.
*
* @param integer $offset
* @param integer $itemsPerPage
* @return array[numeric|whatever]=>array|Doctrine_Record
*/
public function getItems($offset, $itemsPerPage)
{
$this->_query->limit($itemsPerPage);
$this->_query->offset($offset);
$result = $this->_query->execute(array(), $this->_hydrationMode);
return $result;
}
}



PS. Big Thanks to author of this page: http://code.google.com/p/smart-framework/source/browse/trunk/library/SmartL/Zend/Paginator/Adapter/Doctrine.php?r=67