This repository has been archived on 2024-10-25. You can view files and clone it, but cannot push or open issues or pull requests.
Sequenzia/lib/Rails/ActiveRecord/Connection.php
2013-10-02 11:14:53 -05:00

265 lines
7.8 KiB
PHP
Executable File

<?php
namespace Rails\ActiveRecord;
use PDO;
use Rails;
use Rails\ActiveRecord\ActiveRecord;
class Connection
{
protected $pdo;
protected $adapter_name;
protected $name;
protected $config;
public function __construct($config, $name)
{
if (is_array($config)) {
$config = array_merge($this->default_connection_config(), $config);
if (empty($config['dsn'])) {
if ($dsn = $this->build_dsn($config))
$config['dsn'] = $dsn;
else
throw new Exception\ErrorException(sprintf("Not enough info to create DSN string for connection '%s'", $name));
}
$this->pdo = $this->create_connection($config);
} elseif ($config instanceof PDO) {
$this->pdo = $config;
} else {
throw new Exception\InvalidArgumentException('Connection must be either array or instance of PDO.');
}
$this->adapter_name = ActiveRecord::proper_adapter_name($this->pdo->getAttribute(PDO::ATTR_DRIVER_NAME));
$this->name = $name;
}
protected function build_dsn($config)
{
if (!isset($config['adapter']))
return;
switch ($config['adapter']) {
case 'mysql':
$str = $config['adapter'];
$params = [];
if (isset($config['host']))
$params['host'] = $config['host'];
if (isset($config['database']))
$params['dbname'] = $config['database'];
if (isset($config['port']))
$params['port'] = $config['port'];
if (isset($config['charset']))
$params['charset'] = $config['charset'];
if (!isset($config['dsn_params'])) {
$config['dsn_params'] = [];
}
$params = array_merge($params, $config['dsn_params']);
if ($params) {
$str .= ':';
foreach ($params as $key => $value)
$str .= $key . '=' . $value . ';';
}
return $str;
case 'sqlite':
$str = $config['adapter'];
if (isset($config['database'])) {
$str .= ':' . Rails::root() . '/' . $config['database'] . '';
}
return $str;
}
}
protected function default_connection_config()
{
return [
'username' => null,
'password' => null,
'driver_options' => [],
'pdo_attributes' => []
];
}
/**
* Executes the sql and returns the statement.
*/
public function executeSql()
{
$params = func_get_args();
$sql = array_shift($params);
if (!$sql)
throw new Exception\LogicException("Can't execute SQL without SQL");
elseif (is_array($sql)) {
$params = $sql;
$sql = array_shift($params);
}
$this->_parse_query_multimark($sql, $params);
if (!$stmt = $this->pdo->prepare($sql)) {
list($code, $drvrcode, $msg) = $this->pdo->errorInfo();
$e = new Exception\QueryException(
sprintf("[PDOStatement error] [SQLSTATE %s] (%s) %s", $code, $drvrcode, $msg)
);
throw $e;
} elseif (!$stmt->execute($params) && Rails::env() == 'development') {
list($code, $drvrcode, $msg) = $stmt->errorInfo();
$e = new Exception\QueryException(
sprintf("[PDOStatement error] [SQLSTATE %s] (%s) %s", $code, $drvrcode, $msg)
);
$e->setStatement($stmt, $params);
throw $e;
}
$err = $stmt->errorInfo();
if ($err[2]) {
ActiveRecord::setLastError($err, $this->name);
$e = new Exception\QueryException(
sprintf("[SQLSTATE %s] (%s) %s", $err[0], (string)$err[1], $err[2])
);
$e->setStatement($stmt, $params);
$msg = "Error on database query execution\n";
$msg .= $e->getMessage();
Rails::log()->message($msg);
}
return $stmt;
}
/**
* Executes the sql and returns the results.
*/
public function query()
{
$stmt = call_user_func_array([$this, 'executeSql'], func_get_args());
if (ActiveRecord::lastError())
return false;
return $stmt->fetchAll();
}
public function select()
{
$stmt = call_user_func_array([$this, 'executeSql'], func_get_args());
if (ActiveRecord::lastError())
return false;
return $stmt->fetchAll(PDO::FETCH_ASSOC);
}
public function selectRow()
{
$stmt = call_user_func_array([$this, 'executeSql'], func_get_args());
if (ActiveRecord::lastError())
return false;
return $stmt->fetch(PDO::FETCH_ASSOC);
}
# Use only when retrieving ONE column from multiple rows.
public function selectValues()
{
$stmt = call_user_func_array([$this, 'executeSql'], func_get_args());
if (ActiveRecord::lastError())
return false;
$cols = array();
if ($data = $stmt->fetchAll(PDO::FETCH_ASSOC)) {
foreach ($data as $d)
$cols[] = current($d);
}
return $cols;
}
public function selectValue()
{
$stmt = call_user_func_array([$this, 'executeSql'], func_get_args());
if (ActiveRecord::lastError())
return false;
if ($data = $stmt->fetch()) {
$data = array_shift($data);
}
return $data;
}
public function lastInsertId()
{
return $this->pdo->lastInsertId();
}
public function transaction($params = [], \Closure $block = null)
{
if (!$block) {
$block = $params;
}
$this->resource()->beginTransaction();
$block();
$this->resource()->commit();
}
public function resource()
{
return $this->pdo;
}
public function pdo()
{
return $this->pdo;
}
public function adapterName()
{
return $this->adapter_name;
}
public function name()
{
return $this->name;
}
protected function _parse_query_multimark(&$query, array &$params)
{
# If the number of tokens isn't equal to parameters, ignore
# it and return. PDOStatement will trigger a Warning.
if (is_bool(strpos($query, '?')) || substr_count($query, '?') != count($params))
return;
$parts = explode('?', $query);
$parsed_params = array();
foreach ($params as $k => $v) {
if (is_array($v)) {
$k++;
$count = count($v);
$parts[$k] = ($count > 1 ? ', ' . implode(', ', array_fill(0, $count - 1, '?')) : '') . $parts[$k];
$parsed_params = array_merge($parsed_params, $v);
} else {
$parsed_params[] = $v;
}
}
$params = $parsed_params;
$query = implode('?', $parts);
}
protected function create_connection($data)
{
$pdo = new PDO($data['dsn'], $data['username'], $data['password'], $data['driver_options']);
if ($data['pdo_attributes']) {
foreach ($data['pdo_attributes'] as $attr => $val)
$pdo->setAttribute($attr, $val);
}
return $pdo;
}
}