interface/modules/zend_modules/library/Zend/Db/Adapter/Driver/Pdo/Feature/SqliteRowCounter.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\Db\Adapter\Driver\Pdo\Feature;
  11
  12 use Zend\Db\Adapter\Driver\Feature\AbstractFeature;
  13 use Zend\Db\Adapter\Driver\Pdo;
  14
  15 /**
  16  * SqliteRowCounter
  17  */
  18 class SqliteRowCounter extends AbstractFeature
  19 {
  20
  21     /**
  22      * @return string
  23      */
  24     public function getName()
  25     {
  26         return 'SqliteRowCounter';
  27     }
  28
  29     /**
  30      * @param \Zend\Db\Adapter\Driver\Pdo\Statement $statement
  31      * @return int
  32      */
  33     public function getCountForStatement(Pdo\Statement $statement)
  34     {
  35         $countStmt = clone $statement;
  36         $sql = $statement->getSql();
  37         if ($sql == '' || stripos($sql, 'select') === false) {
  38             return null;
  39         }
  40         $countSql = 'SELECT COUNT(*) as "count" FROM (' . $sql . ')';
  41         $countStmt->prepare($countSql);
  42         $result = $countStmt->execute();
  43         $countRow = $result->getResource()->fetch(\PDO::FETCH_ASSOC);
  44         unset($statement, $result);
  45         return $countRow['count'];
  46     }
  47
  48     /**
  49      * @param $sql
  50      * @return null|int
  51      */
  52     public function getCountForSql($sql)
  53     {
  54         if (!stripos($sql, 'select')) {
  55             return null;
  56         }
  57         $countSql = 'SELECT COUNT(*) as count FROM (' . $sql . ')';
  58         /** @var $pdo \PDO */
  59         $pdo = $this->pdoDriver->getConnection()->getResource();
  60         $result = $pdo->query($countSql);
  61         $countRow = $result->fetch(\PDO::FETCH_ASSOC);
  62         return $countRow['count'];
  63     }
  64
  65     /**
  66      * @param $context
  67      * @return closure
  68      */
  69     public function getRowCountClosure($context)
  70     {
  71         $sqliteRowCounter = $this;
  72         return function () use ($sqliteRowCounter, $context) {
  73             /** @var $sqliteRowCounter SqliteRowCounter */
  74             return ($context instanceof Pdo\Statement)
  75                 ? $sqliteRowCounter->getCountForStatement($context)
  76                 : $sqliteRowCounter->getCountForSql($context);
  77         };
  78     }
  79 }