Source of file Pdo.php
Size: 8,627 Bytes - Last Modified: 2014-03-12T23:21:18+01:00
/home/theseer/Downloads/ZendFramework-2.3.0/library/Zend/Db/Adapter/Driver/Pdo/Pdo.php
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315 | <?php /** * Zend Framework (http://framework.zend.com/) * * @link http://github.com/zendframework/zf2 for the canonical source repository * @copyright Copyright (c) 2005-2014 Zend Technologies USA Inc. (http://www.zend.com) * @license http://framework.zend.com/license/new-bsd New BSD License */ namespace Zend\Db\Adapter\Driver\Pdo; use PDOStatement; use Zend\Db\Adapter\Driver\DriverInterface; use Zend\Db\Adapter\Driver\Feature\AbstractFeature; use Zend\Db\Adapter\Driver\Feature\DriverFeatureInterface; use Zend\Db\Adapter\Exception; use Zend\Db\Adapter\Profiler; class Pdo implements DriverInterface, DriverFeatureInterface, Profiler\ProfilerAwareInterface { /** * @const */ const FEATURES_DEFAULT = 'default'; /** * @var Connection */ protected $connection = null; /** * @var Statement */ protected $statementPrototype = null; /** * @var Result */ protected $resultPrototype = null; /** * @var array */ protected $features = array(); /** * @param array|Connection|\PDO $connection * @param null|Statement $statementPrototype * @param null|Result $resultPrototype * @param string $features */ public function __construct($connection, Statement $statementPrototype = null, Result $resultPrototype = null, $features = self::FEATURES_DEFAULT) { if (!$connection instanceof Connection) { $connection = new Connection($connection); } $this->registerConnection($connection); $this->registerStatementPrototype(($statementPrototype) ?: new Statement()); $this->registerResultPrototype(($resultPrototype) ?: new Result()); if (is_array($features)) { foreach ($features as $name => $feature) { $this->addFeature($name, $feature); } } elseif ($features instanceof AbstractFeature) { $this->addFeature($features->getName(), $features); } elseif ($features === self::FEATURES_DEFAULT) { $this->setupDefaultFeatures(); } } /** * @param Profiler\ProfilerInterface $profiler * @return Pdo */ public function setProfiler(Profiler\ProfilerInterface $profiler) { $this->profiler = $profiler; if ($this->connection instanceof Profiler\ProfilerAwareInterface) { $this->connection->setProfiler($profiler); } if ($this->statementPrototype instanceof Profiler\ProfilerAwareInterface) { $this->statementPrototype->setProfiler($profiler); } return $this; } /** * @return null|Profiler\ProfilerInterface */ public function getProfiler() { return $this->profiler; } /** * Register connection * * @param Connection $connection * @return Pdo */ public function registerConnection(Connection $connection) { $this->connection = $connection; $this->connection->setDriver($this); return $this; } /** * Register statement prototype * * @param Statement $statementPrototype */ public function registerStatementPrototype(Statement $statementPrototype) { $this->statementPrototype = $statementPrototype; $this->statementPrototype->setDriver($this); } /** * Register result prototype * * @param Result $resultPrototype */ public function registerResultPrototype(Result $resultPrototype) { $this->resultPrototype = $resultPrototype; } /** * Add feature * * @param string $name * @param AbstractFeature $feature * @return Pdo */ public function addFeature($name, $feature) { if ($feature instanceof AbstractFeature) { $name = $feature->getName(); // overwrite the name, just in case $feature->setDriver($this); } $this->features[$name] = $feature; return $this; } /** * Setup the default features for Pdo * * @return Pdo */ public function setupDefaultFeatures() { $driverName = $this->connection->getDriverName(); if ($driverName == 'sqlite') { $this->addFeature(null, new Feature\SqliteRowCounter); } elseif ($driverName == 'oci') { $this->addFeature(null, new Feature\OracleRowCounter); } return $this; } /** * Get feature * * @param $name * @return AbstractFeature|false */ public function getFeature($name) { if (isset($this->features[$name])) { return $this->features[$name]; } return false; } /** * Get database platform name * * @param string $nameFormat * @return string */ public function getDatabasePlatformName($nameFormat = self::NAME_FORMAT_CAMELCASE) { $name = $this->getConnection()->getDriverName(); if ($nameFormat == self::NAME_FORMAT_CAMELCASE) { switch ($name) { case 'pgsql': return 'Postgresql'; case 'oci': return 'Oracle'; case 'dblib': case 'sqlsrv': return 'SqlServer'; default: return ucfirst($name); } } else { switch ($name) { case 'sqlite': return 'SQLite'; case 'mysql': return 'MySQL'; case 'pgsql': return 'PostgreSQL'; case 'oci': return 'Oracle'; case 'dblib': case 'sqlsrv': return 'SQLServer'; default: return ucfirst($name); } } } /** * Check environment */ public function checkEnvironment() { if (!extension_loaded('PDO')) { throw new Exception\RuntimeException('The PDO extension is required for this adapter but the extension is not loaded'); } } /** * @return Connection */ public function getConnection() { return $this->connection; } /** * @param string|PDOStatement $sqlOrResource * @return Statement */ public function createStatement($sqlOrResource = null) { $statement = clone $this->statementPrototype; if ($sqlOrResource instanceof PDOStatement) { $statement->setResource($sqlOrResource); } else { if (is_string($sqlOrResource)) { $statement->setSql($sqlOrResource); } if (!$this->connection->isConnected()) { $this->connection->connect(); } $statement->initialize($this->connection->getResource()); } return $statement; } /** * @param resource $resource * @param mixed $context * @return Result */ public function createResult($resource, $context = null) { $result = clone $this->resultPrototype; $rowCount = null; // special feature, sqlite PDO counter if ($this->connection->getDriverName() == 'sqlite' && ($sqliteRowCounter = $this->getFeature('SqliteRowCounter')) && $resource->columnCount() > 0) { $rowCount = $sqliteRowCounter->getRowCountClosure($context); } // special feature, oracle PDO counter if ($this->connection->getDriverName() == 'oci' && ($oracleRowCounter = $this->getFeature('OracleRowCounter')) && $resource->columnCount() > 0) { $rowCount = $oracleRowCounter->getRowCountClosure($context); } $result->initialize($resource, $this->connection->getLastGeneratedValue(), $rowCount); return $result; } /** * @return array */ public function getPrepareType() { return self::PARAMETERIZATION_NAMED; } /** * @param string $name * @param string|null $type * @return string */ public function formatParameterName($name, $type = null) { if ($type == null && !is_numeric($name) || $type == self::PARAMETERIZATION_NAMED) { return ':' . $name; } return '?'; } /** * @return mixed */ public function getLastGeneratedValue($name = null) { return $this->connection->getLastGeneratedValue($name); } } |