interface/modules/zend_modules/library/Zend/Db/Adapter/Driver/IbmDb2/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\IbmDb2;
  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 resource
  21      */
  22     protected $db2 = null;
  23
  24     /**
  25      * @var IbmDb2
  26      */
  27     protected $driver = null;
  28
  29     /**
  30      * @var Profiler\ProfilerInterface
  31      */
  32     protected $profiler = null;
  33
  34     /**
  35      * @var string
  36      */
  37     protected $sql = '';
  38
  39     /**
  40      * @var ParameterContainer
  41      */
  42     protected $parameterContainer = null;
  43
  44     /**
  45      * @var bool
  46      */
  47     protected $isPrepared = false;
  48
  49     /**
  50      * @var resource
  51      */
  52     protected $resource = null;
  53
  54     /**
  55      * @param $resource
  56      * @return Statement
  57      */
  58     public function initialize($resource)
  59     {
  60         $this->db2 = $resource;
  61         return $this;
  62     }
  63
  64     /**
  65      * @param IbmDb2 $driver
  66      * @return Statement
  67      */
  68     public function setDriver(IbmDb2 $driver)
  69     {
  70         $this->driver = $driver;
  71         return $this;
  72     }
  73
  74     /**
  75      * @param Profiler\ProfilerInterface $profiler
  76      * @return Statement
  77      */
  78     public function setProfiler(Profiler\ProfilerInterface $profiler)
  79     {
  80         $this->profiler = $profiler;
  81         return $this;
  82     }
  83
  84     /**
  85      * @return null|Profiler\ProfilerInterface
  86      */
  87     public function getProfiler()
  88     {
  89         return $this->profiler;
  90     }
  91
  92     /**
  93      * Set sql
  94      *
  95      * @param $sql
  96      * @return mixed
  97      */
  98     public function setSql($sql)
  99     {
 100         $this->sql = $sql;
 101         return $this;
 102     }
 103
 104     /**
 105      * Get sql
 106      *
 107      * @return mixed
 108      */
 109     public function getSql()
 110     {
 111         return $this->sql;
 112     }
 113
 114     /**
 115      * Set parameter container
 116      *
 117      * @param ParameterContainer $parameterContainer
 118      * @return mixed
 119      */
 120     public function setParameterContainer(ParameterContainer $parameterContainer)
 121     {
 122         $this->parameterContainer = $parameterContainer;
 123         return $this;
 124     }
 125
 126     /**
 127      * Get parameter container
 128      *
 129      * @return mixed
 130      */
 131     public function getParameterContainer()
 132     {
 133         return $this->parameterContainer;
 134     }
 135
 136     /**
 137      * @param $resource
 138      * @throws \Zend\Db\Adapter\Exception\InvalidArgumentException
 139      */
 140     public function setResource($resource)
 141     {
 142         if (get_resource_type($resource) !== 'DB2 Statement') {
 143             throw new Exception\InvalidArgumentException('Resource must be of type DB2 Statement');
 144         }
 145         $this->resource = $resource;
 146     }
 147
 148     /**
 149      * Get resource
 150      *
 151      * @return resource
 152      */
 153     public function getResource()
 154     {
 155         return $this->resource;
 156     }
 157
 158     /**
 159      * Prepare sql
 160      *
 161      * @param string|null $sql
 162      * @return Statement
 163      */
 164     public function prepare($sql = null)
 165     {
 166         if ($this->isPrepared) {
 167             throw new Exception\RuntimeException('This statement has been prepared already');
 168         }
 169
 170         if ($sql == null) {
 171             $sql = $this->sql;
 172         }
 173
 174         $this->resource = db2_prepare($this->db2, $sql);
 175
 176         if ($this->resource === false) {
 177             throw new Exception\RuntimeException(db2_stmt_errormsg(), db2_stmt_error());
 178         }
 179
 180         $this->isPrepared = true;
 181         return $this;
 182     }
 183
 184     /**
 185      * Check if is prepared
 186      *
 187      * @return bool
 188      */
 189     public function isPrepared()
 190     {
 191         return $this->isPrepared;
 192     }
 193
 194     /**
 195      * Execute
 196      *
 197      * @param null $parameters
 198      * @return Result
 199      */
 200     public function execute($parameters = null)
 201     {
 202         if (!$this->isPrepared) {
 203             $this->prepare();
 204         }
 205
 206         /** START Standard ParameterContainer Merging Block */
 207         if (!$this->parameterContainer instanceof ParameterContainer) {
 208             if ($parameters instanceof ParameterContainer) {
 209                 $this->parameterContainer = $parameters;
 210                 $parameters = null;
 211             } else {
 212                 $this->parameterContainer = new ParameterContainer();
 213             }
 214         }
 215
 216         if (is_array($parameters)) {
 217             $this->parameterContainer->setFromArray($parameters);
 218         }
 219         /** END Standard ParameterContainer Merging Block */
 220
 221         if ($this->profiler) {
 222             $this->profiler->profilerStart($this);
 223         }
 224
 225         set_error_handler(function () {}, E_WARNING); // suppress warnings
 226         $response = db2_execute($this->resource, $this->parameterContainer->getPositionalArray());
 227         restore_error_handler();
 228
 229         if ($this->profiler) {
 230             $this->profiler->profilerFinish();
 231         }
 232
 233         if ($response === false) {
 234             throw new Exception\RuntimeException(db2_stmt_errormsg($this->resource));
 235         }
 236
 237         $result = $this->driver->createResult($this->resource);
 238         return $result;
 239     }
 240 }