265 lines
7.8 KiB
PHP
Executable File
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;
|
|
}
|
|
} |