import v1.0.0-RC4 | 2009-05-20
This commit is contained in:
326
libs/Zend/Db/Adapter/Pdo/Abstract.php
Normal file
326
libs/Zend/Db/Adapter/Pdo/Abstract.php
Normal file
@ -0,0 +1,326 @@
|
||||
<?php
|
||||
/**
|
||||
* Zend Framework
|
||||
*
|
||||
* LICENSE
|
||||
*
|
||||
* This source file is subject to the new BSD license that is bundled
|
||||
* with this package in the file LICENSE.txt.
|
||||
* It is also available through the world-wide-web at this URL:
|
||||
* http://framework.zend.com/license/new-bsd
|
||||
* If you did not receive a copy of the license and are unable to
|
||||
* obtain it through the world-wide-web, please send an email
|
||||
* to license@zend.com so we can send you a copy immediately.
|
||||
*
|
||||
* @category Zend
|
||||
* @package Zend_Db
|
||||
* @subpackage Adapter
|
||||
* @copyright Copyright (c) 2005-2008 Zend Technologies USA Inc. (http://www.zend.com)
|
||||
* @license http://framework.zend.com/license/new-bsd New BSD License
|
||||
* @version $Id: Abstract.php 12629 2008-11-13 17:23:13Z alexander $
|
||||
*/
|
||||
|
||||
|
||||
/**
|
||||
* @see Zend_Db_Adapter_Abstract
|
||||
*/
|
||||
require_once 'Zend/Db/Adapter/Abstract.php';
|
||||
|
||||
|
||||
/**
|
||||
* @see Zend_Loader
|
||||
*/
|
||||
require_once 'Zend/Loader.php';
|
||||
|
||||
|
||||
/**
|
||||
* @see Zend_Db_Statement_Pdo
|
||||
*/
|
||||
require_once 'Zend/Db/Statement/Pdo.php';
|
||||
|
||||
|
||||
/**
|
||||
* Class for connecting to SQL databases and performing common operations using PDO.
|
||||
*
|
||||
* @category Zend
|
||||
* @package Zend_Db
|
||||
* @subpackage Adapter
|
||||
* @copyright Copyright (c) 2005-2008 Zend Technologies USA Inc. (http://www.zend.com)
|
||||
* @license http://framework.zend.com/license/new-bsd New BSD License
|
||||
*/
|
||||
abstract class Zend_Db_Adapter_Pdo_Abstract extends Zend_Db_Adapter_Abstract
|
||||
{
|
||||
|
||||
/**
|
||||
* Default class name for a DB statement.
|
||||
*
|
||||
* @var string
|
||||
*/
|
||||
protected $_defaultStmtClass = 'Zend_Db_Statement_Pdo';
|
||||
|
||||
/**
|
||||
* Creates a PDO DSN for the adapter from $this->_config settings.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
protected function _dsn()
|
||||
{
|
||||
// baseline of DSN parts
|
||||
$dsn = $this->_config;
|
||||
|
||||
// don't pass the username, password, and driver_options in the DSN
|
||||
unset($dsn['username']);
|
||||
unset($dsn['password']);
|
||||
unset($dsn['options']);
|
||||
unset($dsn['driver_options']);
|
||||
|
||||
// use all remaining parts in the DSN
|
||||
foreach ($dsn as $key => $val) {
|
||||
$dsn[$key] = "$key=$val";
|
||||
}
|
||||
|
||||
return $this->_pdoType . ':' . implode(';', $dsn);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a PDO object and connects to the database.
|
||||
*
|
||||
* @return void
|
||||
* @throws Zend_Db_Adapter_Exception
|
||||
*/
|
||||
protected function _connect()
|
||||
{
|
||||
// if we already have a PDO object, no need to re-connect.
|
||||
if ($this->_connection) {
|
||||
return;
|
||||
}
|
||||
|
||||
// get the dsn first, because some adapters alter the $_pdoType
|
||||
$dsn = $this->_dsn();
|
||||
|
||||
// check for PDO extension
|
||||
if (!extension_loaded('pdo')) {
|
||||
/**
|
||||
* @see Zend_Db_Adapter_Exception
|
||||
*/
|
||||
require_once 'Zend/Db/Adapter/Exception.php';
|
||||
throw new Zend_Db_Adapter_Exception('The PDO extension is required for this adapter but the extension is not loaded');
|
||||
}
|
||||
|
||||
// check the PDO driver is available
|
||||
if (!in_array($this->_pdoType, PDO::getAvailableDrivers())) {
|
||||
/**
|
||||
* @see Zend_Db_Adapter_Exception
|
||||
*/
|
||||
require_once 'Zend/Db/Adapter/Exception.php';
|
||||
throw new Zend_Db_Adapter_Exception('The ' . $this->_pdoType . ' driver is not currently installed');
|
||||
}
|
||||
|
||||
// create PDO connection
|
||||
$q = $this->_profiler->queryStart('connect', Zend_Db_Profiler::CONNECT);
|
||||
|
||||
try {
|
||||
$this->_connection = new PDO(
|
||||
$dsn,
|
||||
$this->_config['username'],
|
||||
$this->_config['password'],
|
||||
$this->_config['driver_options']
|
||||
);
|
||||
|
||||
$this->_profiler->queryEnd($q);
|
||||
|
||||
// set the PDO connection to perform case-folding on array keys, or not
|
||||
$this->_connection->setAttribute(PDO::ATTR_CASE, $this->_caseFolding);
|
||||
|
||||
// always use exceptions.
|
||||
$this->_connection->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
|
||||
|
||||
} catch (PDOException $e) {
|
||||
/**
|
||||
* @see Zend_Db_Adapter_Exception
|
||||
*/
|
||||
require_once 'Zend/Db/Adapter/Exception.php';
|
||||
throw new Zend_Db_Adapter_Exception($e->getMessage());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Force the connection to close.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
public function closeConnection()
|
||||
{
|
||||
$this->_connection = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Prepares an SQL statement.
|
||||
*
|
||||
* @param string $sql The SQL statement with placeholders.
|
||||
* @param array $bind An array of data to bind to the placeholders.
|
||||
* @return PDOStatement
|
||||
*/
|
||||
public function prepare($sql)
|
||||
{
|
||||
$this->_connect();
|
||||
$stmtClass = $this->_defaultStmtClass;
|
||||
Zend_Loader::loadClass($stmtClass);
|
||||
$stmt = new $stmtClass($this, $sql);
|
||||
$stmt->setFetchMode($this->_fetchMode);
|
||||
return $stmt;
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the last ID generated automatically by an IDENTITY/AUTOINCREMENT column.
|
||||
*
|
||||
* As a convention, on RDBMS brands that support sequences
|
||||
* (e.g. Oracle, PostgreSQL, DB2), this method forms the name of a sequence
|
||||
* from the arguments and returns the last id generated by that sequence.
|
||||
* On RDBMS brands that support IDENTITY/AUTOINCREMENT columns, this method
|
||||
* returns the last value generated for such a column, and the table name
|
||||
* argument is disregarded.
|
||||
*
|
||||
* On RDBMS brands that don't support sequences, $tableName and $primaryKey
|
||||
* are ignored.
|
||||
*
|
||||
* @param string $tableName OPTIONAL Name of table.
|
||||
* @param string $primaryKey OPTIONAL Name of primary key column.
|
||||
* @return string
|
||||
*/
|
||||
public function lastInsertId($tableName = null, $primaryKey = null)
|
||||
{
|
||||
$this->_connect();
|
||||
return $this->_connection->lastInsertId();
|
||||
}
|
||||
|
||||
/**
|
||||
* Special handling for PDO query().
|
||||
* All bind parameter names must begin with ':'
|
||||
*
|
||||
* @param string|Zend_Db_Select $sql The SQL statement with placeholders.
|
||||
* @param array $bind An array of data to bind to the placeholders.
|
||||
* @return Zend_Db_Statement_Pdo
|
||||
* @throws Zend_Db_Adapter_Exception To re-throw PDOException.
|
||||
*/
|
||||
public function query($sql, $bind = array())
|
||||
{
|
||||
if (is_array($bind)) {
|
||||
foreach ($bind as $name => $value) {
|
||||
if (!is_int($name) && !preg_match('/^:/', $name)) {
|
||||
$newName = ":$name";
|
||||
unset($bind[$name]);
|
||||
$bind[$newName] = $value;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
return parent::query($sql, $bind);
|
||||
} catch (PDOException $e) {
|
||||
/**
|
||||
* @see Zend_Db_Statement_Exception
|
||||
*/
|
||||
require_once 'Zend/Db/Statement/Exception.php';
|
||||
throw new Zend_Db_Statement_Exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Quote a raw string.
|
||||
*
|
||||
* @param string $value Raw string
|
||||
* @return string Quoted string
|
||||
*/
|
||||
protected function _quote($value)
|
||||
{
|
||||
if (is_int($value) || is_float($value)) {
|
||||
return $value;
|
||||
}
|
||||
$this->_connect();
|
||||
return $this->_connection->quote($value);
|
||||
}
|
||||
|
||||
/**
|
||||
* Begin a transaction.
|
||||
*/
|
||||
protected function _beginTransaction()
|
||||
{
|
||||
$this->_connect();
|
||||
$this->_connection->beginTransaction();
|
||||
}
|
||||
|
||||
/**
|
||||
* Commit a transaction.
|
||||
*/
|
||||
protected function _commit()
|
||||
{
|
||||
$this->_connect();
|
||||
$this->_connection->commit();
|
||||
}
|
||||
|
||||
/**
|
||||
* Roll-back a transaction.
|
||||
*/
|
||||
protected function _rollBack() {
|
||||
$this->_connect();
|
||||
$this->_connection->rollBack();
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the PDO fetch mode.
|
||||
*
|
||||
* @todo Support FETCH_CLASS and FETCH_INTO.
|
||||
*
|
||||
* @param int $mode A PDO fetch mode.
|
||||
* @return void
|
||||
* @throws Zend_Db_Adapter_Exception
|
||||
*/
|
||||
public function setFetchMode($mode)
|
||||
{
|
||||
//check for PDO extension
|
||||
if (!extension_loaded('pdo')) {
|
||||
/**
|
||||
* @see Zend_Db_Adapter_Exception
|
||||
*/
|
||||
require_once 'Zend/Db/Adapter/Exception.php';
|
||||
throw new Zend_Db_Adapter_Exception('The PDO extension is required for this adapter but the extension is not loaded');
|
||||
}
|
||||
switch ($mode) {
|
||||
case PDO::FETCH_LAZY:
|
||||
case PDO::FETCH_ASSOC:
|
||||
case PDO::FETCH_NUM:
|
||||
case PDO::FETCH_BOTH:
|
||||
case PDO::FETCH_NAMED:
|
||||
case PDO::FETCH_OBJ:
|
||||
$this->_fetchMode = $mode;
|
||||
break;
|
||||
default:
|
||||
/**
|
||||
* @see Zend_Db_Adapter_Exception
|
||||
*/
|
||||
require_once 'Zend/Db/Adapter/Exception.php';
|
||||
throw new Zend_Db_Adapter_Exception("Invalid fetch mode '$mode' specified");
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if the adapter supports real SQL parameters.
|
||||
*
|
||||
* @param string $type 'positional' or 'named'
|
||||
* @return bool
|
||||
*/
|
||||
public function supportsParameters($type)
|
||||
{
|
||||
switch ($type) {
|
||||
case 'positional':
|
||||
case 'named':
|
||||
default:
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user