interface/modules/zend_modules/library/Zend/Db/Adapter/Driver/Pgsql/Statement.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\Pgsql;
  11
  12 use Zend\Db\Adapter\Driver\StatementInterface;
  13 use Zend\Db\Adapter\Exception;
  14 use Zend\Db\Adapter\ParameterContainer;
  15 use Zend\Db\Adapter\Profiler;
  16
  17 class Statement implements StatementInterface, Profiler\ProfilerAwareInterface
  18 {
  19     /**
  20      * @var int
  21      */
  22     protected static $statementIndex = 0;
  23
  24     /**
  25      * @var string
  26      */
  27     protected $statementName = '';
  28
  29     /**
  30      * @var Pgsql
  31      */
  32     protected $driver = null;
  33
  34     /**
  35      * @var Profiler\ProfilerInterface
  36      */
  37     protected $profiler = null;
  38
  39     /**
  40      * @var resource
  41      */
  42     protected $pgsql = null;
  43
  44     /**
  45      * @var resource
  46      */
  47     protected $resource = null;
  48
  49     /**
  50      * @var string
  51      */
  52     protected $sql;
  53
  54     /**
  55      * @var ParameterContainer
  56      */
  57     protected $parameterContainer;
  58
  59     /**
  60      * @param  Pgsql $driver
  61      * @return Statement
  62      */
  63     public function setDriver(Pgsql $driver)
  64     {
  65         $this->driver = $driver;
  66         return $this;
  67     }
  68
  69     /**
  70      * @param Profiler\ProfilerInterface $profiler
  71      * @return Statement
  72      */
  73     public function setProfiler(Profiler\ProfilerInterface $profiler)
  74     {
  75         $this->profiler = $profiler;
  76         return $this;
  77     }
  78
  79     /**
  80      * @return null|Profiler\ProfilerInterface
  81      */
  82     public function getProfiler()
  83     {
  84         return $this->profiler;
  85     }
  86
  87     /**
  88      * Initialize
  89      *
  90      * @param  resource $pgsql
  91      * @return void
  92      * @throws Exception\RuntimeException for invalid or missing postgresql connection
  93      */
  94     public function initialize($pgsql)
  95     {
  96         if (!is_resource($pgsql) || get_resource_type($pgsql) !== 'pgsql link') {
  97             throw new Exception\RuntimeException(sprintf(
  98                 '%s: Invalid or missing postgresql connection; received "%s"',
  99                 __METHOD__,
 100                 get_resource_type($pgsql)
 101             ));
 102         }
 103         $this->pgsql = $pgsql;
 104     }
 105
 106     /**
 107      * Get resource
 108      *
 109      * @return resource
 110      */
 111     public function getResource()
 112     {
 113         // TODO: Implement getResource() method.
 114     }
 115
 116     /**
 117      * Set sql
 118      *
 119      * @param string $sql
 120      * @return Statement
 121      */
 122     public function setSql($sql)
 123     {
 124         $this->sql = $sql;
 125         return $this;
 126     }
 127
 128     /**
 129      * Get sql
 130      *
 131      * @return string
 132      */
 133     public function getSql()
 134     {
 135         return $this->sql;
 136     }
 137
 138     /**
 139      * Set parameter container
 140      *
 141      * @param ParameterContainer $parameterContainer
 142      * @return Statement
 143      */
 144     public function setParameterContainer(ParameterContainer $parameterContainer)
 145     {
 146         $this->parameterContainer = $parameterContainer;
 147         return $this;
 148     }
 149
 150     /**
 151      * Get parameter container
 152      *
 153      * @return ParameterContainer
 154      */
 155     public function getParameterContainer()
 156     {
 157         return $this->parameterContainer;
 158     }
 159
 160     /**
 161      * Prepare
 162      *
 163      * @param string $sql
 164      */
 165     public function prepare($sql = null)
 166     {
 167         $sql = ($sql) ?: $this->sql;
 168
 169         $pCount = 1;
 170         $sql = preg_replace_callback(
 171             '#\$\##', function ($foo) use (&$pCount) {
 172                 return '$' . $pCount++;
 173             },
 174             $sql
 175         );
 176
 177         $this->sql = $sql;
 178         $this->statementName = 'statement' . ++static::$statementIndex;
 179         $this->resource = pg_prepare($this->pgsql, $this->statementName, $sql);
 180     }
 181
 182     /**
 183      * Is prepared
 184      *
 185      * @return bool
 186      */
 187     public function isPrepared()
 188     {
 189         return isset($this->resource);
 190     }
 191
 192     /**
 193      * Execute
 194      *
 195      * @param  ParameterContainer|null $parameters
 196      * @throws Exception\InvalidQueryException
 197      * @return Result
 198      */
 199     public function execute($parameters = null)
 200     {
 201         if (!$this->isPrepared()) {
 202             $this->prepare();
 203         }
 204
 205         /** START Standard ParameterContainer Merging Block */
 206         if (!$this->parameterContainer instanceof ParameterContainer) {
 207             if ($parameters instanceof ParameterContainer) {
 208                 $this->parameterContainer = $parameters;
 209                 $parameters = null;
 210             } else {
 211                 $this->parameterContainer = new ParameterContainer();
 212             }
 213         }
 214
 215         if (is_array($parameters)) {
 216             $this->parameterContainer->setFromArray($parameters);
 217         }
 218
 219         if ($this->parameterContainer->count() > 0) {
 220             $parameters = $this->parameterContainer->getPositionalArray();
 221         }
 222         /** END Standard ParameterContainer Merging Block */
 223
 224         if ($this->profiler) {
 225             $this->profiler->profilerStart($this);
 226         }
 227
 228         $resultResource = pg_execute($this->pgsql, $this->statementName, (array) $parameters);
 229
 230         if ($this->profiler) {
 231             $this->profiler->profilerFinish();
 232         }
 233
 234         if ($resultResource === false) {
 235             throw new Exception\InvalidQueryException(pg_last_error());
 236         }
 237
 238         $result = $this->driver->createResult($resultResource);
 239         return $result;
 240     }
 241 }