/vendor/doctrine/dbal/lib/Doctrine/DBAL/Driver/IBMDB2/DB2Statement.php
PHP | 344 lines | 190 code | 54 blank | 100 comment | 22 complexity | 5ad65cd9a9d16ca997f8eddb27f4a25d MD5 | raw file
- <?php
- /*
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- * This software consists of voluntary contributions made by many individuals
- * and is licensed under the MIT license. For more information, see
- * <http://www.doctrine-project.org>.
- */
- namespace Doctrine\DBAL\Driver\IBMDB2;
- use Doctrine\DBAL\Driver\Statement;
- class DB2Statement implements \IteratorAggregate, Statement
- {
- /**
- * @var resource
- */
- private $_stmt = null;
- /**
- * @var array
- */
- private $_bindParam = array();
- /**
- * @var string Name of the default class to instantiate when fetch mode is \PDO::FETCH_CLASS.
- */
- private $defaultFetchClass = '\stdClass';
- /**
- * @var string Constructor arguments for the default class to instantiate when fetch mode is \PDO::FETCH_CLASS.
- */
- private $defaultFetchClassCtorArgs = array();
- /**
- * @var integer
- */
- private $_defaultFetchMode = \PDO::FETCH_BOTH;
- /**
- * DB2_BINARY, DB2_CHAR, DB2_DOUBLE, or DB2_LONG
- *
- * @var array
- */
- static private $_typeMap = array(
- \PDO::PARAM_INT => DB2_LONG,
- \PDO::PARAM_STR => DB2_CHAR,
- );
- /**
- * @param resource $stmt
- */
- public function __construct($stmt)
- {
- $this->_stmt = $stmt;
- }
- /**
- * {@inheritdoc}
- */
- public function bindValue($param, $value, $type = null)
- {
- return $this->bindParam($param, $value, $type);
- }
- /**
- * {@inheritdoc}
- */
- public function bindParam($column, &$variable, $type = null, $length = null)
- {
- $this->_bindParam[$column] =& $variable;
- if ($type && isset(self::$_typeMap[$type])) {
- $type = self::$_typeMap[$type];
- } else {
- $type = DB2_CHAR;
- }
- if (!db2_bind_param($this->_stmt, $column, "variable", DB2_PARAM_IN, $type)) {
- throw new DB2Exception(db2_stmt_errormsg());
- }
- return true;
- }
- /**
- * {@inheritdoc}
- */
- public function closeCursor()
- {
- if ( ! $this->_stmt) {
- return false;
- }
- $this->_bindParam = array();
- db2_free_result($this->_stmt);
- $ret = db2_free_stmt($this->_stmt);
- $this->_stmt = false;
- return $ret;
- }
- /**
- * {@inheritdoc}
- */
- public function columnCount()
- {
- if ( ! $this->_stmt) {
- return false;
- }
- return db2_num_fields($this->_stmt);
- }
- /**
- * {@inheritdoc}
- */
- public function errorCode()
- {
- return db2_stmt_error();
- }
- /**
- * {@inheritdoc}
- */
- public function errorInfo()
- {
- return array(
- 0 => db2_stmt_errormsg(),
- 1 => db2_stmt_error(),
- );
- }
- /**
- * {@inheritdoc}
- */
- public function execute($params = null)
- {
- if ( ! $this->_stmt) {
- return false;
- }
- /*$retval = true;
- if ($params !== null) {
- $retval = @db2_execute($this->_stmt, $params);
- } else {
- $retval = @db2_execute($this->_stmt);
- }*/
- if ($params === null) {
- ksort($this->_bindParam);
- $params = array_values($this->_bindParam);
- }
- $retval = @db2_execute($this->_stmt, $params);
- if ($retval === false) {
- throw new DB2Exception(db2_stmt_errormsg());
- }
- return $retval;
- }
- /**
- * {@inheritdoc}
- */
- public function setFetchMode($fetchMode, $arg2 = null, $arg3 = null)
- {
- $this->_defaultFetchMode = $fetchMode;
- $this->defaultFetchClass = $arg2 ? $arg2 : $this->defaultFetchClass;
- $this->defaultFetchClassCtorArgs = $arg3 ? (array) $arg3 : $this->defaultFetchClassCtorArgs;
- return true;
- }
- /**
- * {@inheritdoc}
- */
- public function getIterator()
- {
- $data = $this->fetchAll();
- return new \ArrayIterator($data);
- }
- /**
- * {@inheritdoc}
- */
- public function fetch($fetchMode = null)
- {
- $fetchMode = $fetchMode ?: $this->_defaultFetchMode;
- switch ($fetchMode) {
- case \PDO::FETCH_BOTH:
- return db2_fetch_both($this->_stmt);
- case \PDO::FETCH_ASSOC:
- return db2_fetch_assoc($this->_stmt);
- case \PDO::FETCH_CLASS:
- $className = $this->defaultFetchClass;
- $ctorArgs = $this->defaultFetchClassCtorArgs;
- if (func_num_args() >= 2) {
- $args = func_get_args();
- $className = $args[1];
- $ctorArgs = isset($args[2]) ? $args[2] : array();
- }
- $result = db2_fetch_object($this->_stmt);
- if ($result instanceof \stdClass) {
- $result = $this->castObject($result, $className, $ctorArgs);
- }
- return $result;
- case \PDO::FETCH_NUM:
- return db2_fetch_array($this->_stmt);
- case \PDO::FETCH_OBJ:
- return db2_fetch_object($this->_stmt);
- default:
- throw new DB2Exception("Given Fetch-Style " . $fetchMode . " is not supported.");
- }
- }
- /**
- * {@inheritdoc}
- */
- public function fetchAll($fetchMode = null)
- {
- $rows = array();
- switch ($fetchMode) {
- case \PDO::FETCH_CLASS:
- while ($row = call_user_func_array(array($this, 'fetch'), func_get_args())) {
- $rows[] = $row;
- }
- break;
- case \PDO::FETCH_COLUMN:
- while ($row = $this->fetchColumn()) {
- $rows[] = $row;
- }
- break;
- default:
- while ($row = $this->fetch($fetchMode)) {
- $rows[] = $row;
- }
- }
- return $rows;
- }
- /**
- * {@inheritdoc}
- */
- public function fetchColumn($columnIndex = 0)
- {
- $row = $this->fetch(\PDO::FETCH_NUM);
- if (false === $row) {
- return false;
- }
- return isset($row[$columnIndex]) ? $row[$columnIndex] : null;
- }
- /**
- * {@inheritdoc}
- */
- public function rowCount()
- {
- return (@db2_num_rows($this->_stmt))?:0;
- }
- /**
- * Casts a stdClass object to the given class name mapping its' properties.
- *
- * @param \stdClass $sourceObject Object to cast from.
- * @param string|object $destinationClass Name of the class or class instance to cast to.
- * @param array $ctorArgs Arguments to use for constructing the destination class instance.
- *
- * @return object
- *
- * @throws DB2Exception
- */
- private function castObject(\stdClass $sourceObject, $destinationClass, array $ctorArgs = array())
- {
- if ( ! is_string($destinationClass)) {
- if ( ! is_object($destinationClass)) {
- throw new DB2Exception(sprintf(
- 'Destination class has to be of type string or object, %s given.', gettype($destinationClass)
- ));
- }
- } else {
- $destinationClass = new \ReflectionClass($destinationClass);
- $destinationClass = $destinationClass->newInstanceArgs($ctorArgs);
- }
- $sourceReflection = new \ReflectionObject($sourceObject);
- $destinationClassReflection = new \ReflectionObject($destinationClass);
- /** @var \ReflectionProperty[] $destinationProperties */
- $destinationProperties = array_change_key_case($destinationClassReflection->getProperties(), \CASE_LOWER);
- foreach ($sourceReflection->getProperties() as $sourceProperty) {
- $sourceProperty->setAccessible(true);
- $name = $sourceProperty->getName();
- $value = $sourceProperty->getValue($sourceObject);
- // Try to find a case-matching property.
- if ($destinationClassReflection->hasProperty($name)) {
- $destinationProperty = $destinationClassReflection->getProperty($name);
- $destinationProperty->setAccessible(true);
- $destinationProperty->setValue($destinationClass, $value);
- continue;
- }
- $name = strtolower($name);
- // Try to find a property without matching case.
- // Fallback for the driver returning either all uppercase or all lowercase column names.
- if (isset($destinationProperties[$name])) {
- $destinationProperty = $destinationProperties[$name];
- $destinationProperty->setAccessible(true);
- $destinationProperty->setValue($destinationClass, $value);
- continue;
- }
- $destinationClass->$name = $value;
- }
- return $destinationClass;
- }
- }