interface/modules/zend_modules/library/Zend/Paginator/Adapter/DbSelect.php

   1 <?php
   2 /**
   3  * Zend Framework (http://framework.zend.com/)
   4  *
   5  * @link      http://github.com/zendframework/zf2 for the canonical source repository
   6  * @copyright Copyright (c) 2005-2013 Zend Technologies USA Inc. (http://www.zend.com)
   7  * @license   http://framework.zend.com/license/new-bsd New BSD License
   8  */
   9
  10 namespace Zend\Paginator\Adapter;
  11
  12 use Zend\Db\Adapter\Adapter;
  13 use Zend\Db\Sql\Sql;
  14 use Zend\Db\Sql\Expression;
  15 use Zend\Db\Sql\Select;
  16 use Zend\Db\ResultSet\ResultSet;
  17 use Zend\Db\ResultSet\ResultSetInterface;
  18
  19 class DbSelect implements AdapterInterface
  20 {
  21
  22     /**
  23      * @var Sql
  24      */
  25     protected $sql = null;
  26
  27     /**
  28      * Database query
  29      *
  30      * @var Select
  31      */
  32     protected $select = null;
  33
  34     /**
  35      * @var ResultSet
  36      */
  37     protected $resultSetPrototype = null;
  38
  39     /**
  40      * Total item count
  41      *
  42      * @var int
  43      */
  44     protected $rowCount = null;
  45
  46     /**
  47      * Constructor.
  48      *
  49      * @param Select $select The select query
  50      * @param Adapter|Sql $adapterOrSqlObject DB adapter or Sql object
  51      * @param null|ResultSetInterface $resultSetPrototype
  52      * @throws Exception\InvalidArgumentException
  53      */
  54     public function __construct(Select $select, $adapterOrSqlObject, ResultSetInterface $resultSetPrototype = null)
  55     {
  56         $this->select = $select;
  57
  58         if ($adapterOrSqlObject instanceof Adapter) {
  59             $adapterOrSqlObject = new Sql($adapterOrSqlObject);
  60         }
  61
  62         if (!$adapterOrSqlObject instanceof Sql) {
  63             throw new Exception\InvalidArgumentException(
  64                 '$adapterOrSqlObject must be an instance of Zend\Db\Adapter\Adapter or Zend\Db\Sql\Sql'
  65             );
  66         }
  67
  68         $this->sql                = $adapterOrSqlObject;
  69         $this->resultSetPrototype = ($resultSetPrototype) ?: new ResultSet;
  70     }
  71
  72     /**
  73      * Returns an array of items for a page.
  74      *
  75      * @param  int $offset           Page offset
  76      * @param  int $itemCountPerPage Number of items per page
  77      * @return array
  78      */
  79     public function getItems($offset, $itemCountPerPage)
  80     {
  81         $select = clone $this->select;
  82         $select->offset($offset);
  83         $select->limit($itemCountPerPage);
  84
  85         $statement = $this->sql->prepareStatementForSqlObject($select);
  86         $result    = $statement->execute();
  87
  88         $resultSet = clone $this->resultSetPrototype;
  89         $resultSet->initialize($result);
  90
  91         return $resultSet;
  92     }
  93
  94     /**
  95      * Returns the total number of rows in the result set.
  96      *
  97      * @return int
  98      */
  99     public function count()
 100     {
 101         if ($this->rowCount !== null) {
 102             return $this->rowCount;
 103         }
 104
 105         $select = clone $this->select;
 106         $select->reset(Select::LIMIT);
 107         $select->reset(Select::OFFSET);
 108         $select->reset(Select::ORDER);
 109
 110         $countSelect = new Select;
 111         $countSelect->columns(array('c' => new Expression('COUNT(1)')));
 112         $countSelect->from(array('original_select' => $select));
 113
 114         $statement = $this->sql->prepareStatementForSqlObject($countSelect);
 115         $result    = $statement->execute();
 116         $row       = $result->current();
 117
 118         $this->rowCount = $row['c'];
 119
 120         return $this->rowCount;
 121     }
 122 }