Source of file Connection.php
Size: 8,246 Bytes - Last Modified: 2014-03-12T23:21:18+01:00
/home/theseer/Downloads/ZendFramework-2.3.0/library/Zend/Db/Adapter/Driver/Mysqli/Connection.php
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348 | <?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\Mysqli; use Zend\Db\Adapter\Driver\ConnectionInterface; use Zend\Db\Adapter\Exception; use Zend\Db\Adapter\Profiler; class Connection implements ConnectionInterface, Profiler\ProfilerAwareInterface { /** * @var Mysqli */ protected $driver = null; /** * @var Profiler\ProfilerInterface */ protected $profiler = null; /** * Connection parameters * * @var array */ protected $connectionParameters = array(); /** * @var \mysqli */ protected $resource = null; /** * In transaction * * @var bool */ protected $inTransaction = false; /** * Constructor * * @param array|mysqli|null $connectionInfo * @throws \Zend\Db\Adapter\Exception\InvalidArgumentException */ public function __construct($connectionInfo = null) { if (is_array($connectionInfo)) { $this->setConnectionParameters($connectionInfo); } elseif ($connectionInfo instanceof \mysqli) { $this->setResource($connectionInfo); } elseif (null !== $connectionInfo) { throw new Exception\InvalidArgumentException('$connection must be an array of parameters, a mysqli object or null'); } } /** * @param Mysqli $driver * @return Connection */ public function setDriver(Mysqli $driver) { $this->driver = $driver; return $this; } /** * @param Profiler\ProfilerInterface $profiler * @return Connection */ public function setProfiler(Profiler\ProfilerInterface $profiler) { $this->profiler = $profiler; return $this; } /** * @return null|Profiler\ProfilerInterface */ public function getProfiler() { return $this->profiler; } /** * Set connection parameters * * @param array $connectionParameters * @return Connection */ public function setConnectionParameters(array $connectionParameters) { $this->connectionParameters = $connectionParameters; return $this; } /** * Get connection parameters * * @return array */ public function getConnectionParameters() { return $this->connectionParameters; } /** * Get current schema * * @return string */ public function getCurrentSchema() { if (!$this->isConnected()) { $this->connect(); } /** @var $result \mysqli_result */ $result = $this->resource->query('SELECT DATABASE()'); $r = $result->fetch_row(); return $r[0]; } /** * Set resource * * @param \mysqli $resource * @return Connection */ public function setResource(\mysqli $resource) { $this->resource = $resource; return $this; } /** * Get resource * * @return \mysqli */ public function getResource() { $this->connect(); return $this->resource; } /** * Connect * * @throws Exception\RuntimeException * @return Connection */ public function connect() { if ($this->resource instanceof \mysqli) { return $this; } // localize $p = $this->connectionParameters; // given a list of key names, test for existence in $p $findParameterValue = function (array $names) use ($p) { foreach ($names as $name) { if (isset($p[$name])) { return $p[$name]; } } return; }; $hostname = $findParameterValue(array('hostname', 'host')); $username = $findParameterValue(array('username', 'user')); $password = $findParameterValue(array('password', 'passwd', 'pw')); $database = $findParameterValue(array('database', 'dbname', 'db', 'schema')); $port = (isset($p['port'])) ? (int) $p['port'] : null; $socket = (isset($p['socket'])) ? $p['socket'] : null; $this->resource = new \mysqli(); $this->resource->init(); if (!empty($p['driver_options'])) { foreach ($p['driver_options'] as $option => $value) { if (is_string($option)) { $option = strtoupper($option); if (!defined($option)) { continue; } $option = constant($option); } $this->resource->options($option, $value); } } $this->resource->real_connect($hostname, $username, $password, $database, $port, $socket); if ($this->resource->connect_error) { throw new Exception\RuntimeException( 'Connection error', null, new Exception\ErrorException($this->resource->connect_error, $this->resource->connect_errno) ); } if (!empty($p['charset'])) { $this->resource->set_charset($p['charset']); } return $this; } /** * Is connected * * @return bool */ public function isConnected() { return ($this->resource instanceof \mysqli); } /** * Disconnect * * @return void */ public function disconnect() { if ($this->resource instanceof \mysqli) { $this->resource->close(); } $this->resource = null; } /** * Begin transaction * * @return void */ public function beginTransaction() { if (!$this->isConnected()) { $this->connect(); } $this->resource->autocommit(false); $this->inTransaction = true; } /** * In transaction * * @return bool */ public function inTransaction() { return $this->inTransaction; } /** * Commit * * @return void */ public function commit() { if (!$this->resource) { $this->connect(); } $this->resource->commit(); $this->inTransaction = false; $this->resource->autocommit(true); } /** * Rollback * * @throws Exception\RuntimeException * @return Connection */ public function rollback() { if (!$this->resource) { throw new Exception\RuntimeException('Must be connected before you can rollback.'); } if (!$this->inTransaction) { throw new Exception\RuntimeException('Must call beginTransaction() before you can rollback.'); } $this->resource->rollback(); $this->resource->autocommit(true); return $this; } /** * Execute * * @param string $sql * @throws Exception\InvalidQueryException * @return Result */ public function execute($sql) { if (!$this->isConnected()) { $this->connect(); } if ($this->profiler) { $this->profiler->profilerStart($sql); } $resultResource = $this->resource->query($sql); if ($this->profiler) { $this->profiler->profilerFinish($sql); } // if the returnValue is something other than a mysqli_result, bypass wrapping it if ($resultResource === false) { throw new Exception\InvalidQueryException($this->resource->error); } $resultPrototype = $this->driver->createResult(($resultResource === true) ? $this->resource : $resultResource); return $resultPrototype; } /** * Get last generated id * * @param null $name Ignored * @return int */ public function getLastGeneratedValue($name = null) { return $this->resource->insert_id; } } |