143 lines
3.7 KiB
PHP
143 lines
3.7 KiB
PHP
<?php
|
|
class Doctrine_DB extends PDO implements Countable, IteratorAggregate {
|
|
/**
|
|
* default DSN
|
|
*/
|
|
const DSN = "mysql://root:dc34@localhost/test";
|
|
/**
|
|
* executed queries
|
|
*/
|
|
private $queries = array();
|
|
/**
|
|
* execution times of the executed queries
|
|
*/
|
|
private $exectimes = array();
|
|
|
|
/**
|
|
* constructor
|
|
* @param string $dsn
|
|
* @param string $username
|
|
* @param string $password
|
|
*/
|
|
public function __construct($dsn,$username,$password) {
|
|
parent::__construct($dsn,$username,$password);
|
|
$this->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
|
|
$this->setAttribute(PDO::ATTR_STATEMENT_CLASS, array("Doctrine_DBStatement",array($this)));
|
|
}
|
|
|
|
|
|
public static function getConn($dsn,$username = null, $password = null) {
|
|
static $instance;
|
|
|
|
if( ! isset($instance)) {
|
|
$instance = new Doctrine_DB($dsn,$username,$password);
|
|
}
|
|
return $instance;
|
|
}
|
|
|
|
/**
|
|
* @param string $dsn PEAR::DB like DSN
|
|
* format: schema://user:password@address/dbname
|
|
*/
|
|
public static function getConnection($dsn = null) {
|
|
static $instance = array();
|
|
$md5 = md5($dsn);
|
|
|
|
if( ! isset($instance[$md5])) {
|
|
if( ! isset($dsn)) {
|
|
$a = parse_url(self::DSN);
|
|
} else {
|
|
$a = parse_url($dsn);
|
|
}
|
|
$e = array();
|
|
|
|
$e[0] = $a["scheme"].":host=".$a["host"].";dbname=".substr($a["path"],1);
|
|
$e[1] = $a["user"];
|
|
$e[2] = $a["pass"];
|
|
|
|
$instance[$md5] = new Doctrine_DB($e[0],$e[1],$e[2]);
|
|
}
|
|
return $instance[$md5];
|
|
}
|
|
/**
|
|
* @param string $query query to be executed
|
|
*/
|
|
public function query($query) {
|
|
try {
|
|
$this->queries[] = $query;
|
|
$time = microtime();
|
|
|
|
$stmt = parent::query($query);
|
|
|
|
$this->exectimes[] = (microtime() - $time);
|
|
return $stmt;
|
|
} catch(PDOException $e) {
|
|
throw $e;
|
|
}
|
|
}
|
|
/**
|
|
* @param string $query query to be prepared
|
|
*/
|
|
public function prepare($query) {
|
|
$this->queries[] = $query;
|
|
|
|
return parent::prepare($query);
|
|
}
|
|
/**
|
|
* @param string $time exectime of the last executed query
|
|
* @return void
|
|
*/
|
|
public function addExecTime($time) {
|
|
$this->exectimes[] = $time;
|
|
}
|
|
|
|
public function getExecTimes() {
|
|
return $this->exectimes;
|
|
}
|
|
/**
|
|
* @return array an array of executed queries
|
|
*/
|
|
public function getQueries() {
|
|
return $this->queries;
|
|
}
|
|
/**
|
|
* @return ArrayIterator
|
|
*/
|
|
public function getIterator() {
|
|
return new ArrayIterator($this->queries);
|
|
}
|
|
/**
|
|
* returns the number of executed queries
|
|
* @return integer
|
|
*/
|
|
public function count() {
|
|
return count($this->queries);
|
|
}
|
|
|
|
}
|
|
class Doctrine_DBStatement extends PDOStatement {
|
|
/**
|
|
* @param Doctrine_DB $dbh Doctrine Database Handler
|
|
*/
|
|
private $dbh;
|
|
/**
|
|
* @param Doctrine_DB $dbh
|
|
*/
|
|
private function __construct(Doctrine_DB $dbh) {
|
|
$this->dbh = $dbh;
|
|
}
|
|
/**
|
|
* @param array $params
|
|
*/
|
|
public function execute(array $params = null) {
|
|
$time = microtime();
|
|
|
|
$result = parent::execute($params);
|
|
|
|
$exectime = (microtime() - $time);
|
|
$this->dbh->addExecTime($exectime);
|
|
return $result;
|
|
}
|
|
}
|
|
?>
|