mirror of
https://gitee.com/fastadminnet/framework.git
synced 2026-07-01 12:42:48 +08:00
896 lines
26 KiB
PHP
896 lines
26 KiB
PHP
<?php
|
||
// +----------------------------------------------------------------------
|
||
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
||
// +----------------------------------------------------------------------
|
||
// | Copyright (c) 2006~2016 http://thinkphp.cn All rights reserved.
|
||
// +----------------------------------------------------------------------
|
||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||
// +----------------------------------------------------------------------
|
||
// | Author: liu21st <liu21st@gmail.com>
|
||
// +----------------------------------------------------------------------
|
||
|
||
namespace think\db;
|
||
|
||
use PDO;
|
||
use PDOStatement;
|
||
use think\Collection;
|
||
use think\Db;
|
||
use think\db\Query;
|
||
use think\Debug;
|
||
use think\Exception;
|
||
use think\exception\DbBindParamException;
|
||
use think\exception\PDOException;
|
||
use think\Log;
|
||
|
||
abstract class Connection
|
||
{
|
||
|
||
/** @var PDOStatement PDO操作实例 */
|
||
protected $PDOStatement;
|
||
|
||
// 当前操作的数据表名
|
||
protected $table = '';
|
||
// 当前操作的数据对象名
|
||
protected $name = '';
|
||
|
||
/** @var string 当前SQL指令 */
|
||
protected $queryStr = '';
|
||
// 最后插入ID
|
||
protected $lastInsID;
|
||
// 返回或者影响记录数
|
||
protected $numRows = 0;
|
||
// 事务指令数
|
||
protected $transTimes = 0;
|
||
// 事务标识
|
||
protected $transLabel = '';
|
||
// 错误信息
|
||
protected $error = '';
|
||
|
||
/** @var PDO[] 数据库连接ID 支持多个连接 */
|
||
protected $links = [];
|
||
|
||
/** @var PDO 当前连接ID */
|
||
protected $linkID;
|
||
protected $linkRead;
|
||
protected $linkWrite;
|
||
|
||
// 查询结果类型
|
||
protected $resultSetType = Db::RESULTSET_ARRAY;
|
||
// 查询结果类型
|
||
protected $fetchType = PDO::FETCH_ASSOC;
|
||
// 字段属性大小写
|
||
protected $attrCase = PDO::CASE_LOWER;
|
||
// 监听回调
|
||
protected static $event = [];
|
||
// 查询对象
|
||
protected $query = [];
|
||
// 数据库连接参数配置
|
||
protected $config = [
|
||
// 数据库类型
|
||
'type' => '',
|
||
// 服务器地址
|
||
'hostname' => '',
|
||
// 数据库名
|
||
'database' => '',
|
||
// 用户名
|
||
'username' => '',
|
||
// 密码
|
||
'password' => '',
|
||
// 端口
|
||
'hostport' => '',
|
||
// 连接dsn
|
||
'dsn' => '',
|
||
// 数据库连接参数
|
||
'params' => [],
|
||
// 数据库编码默认采用utf8
|
||
'charset' => 'utf8',
|
||
// 数据库表前缀
|
||
'prefix' => '',
|
||
// 数据库调试模式
|
||
'debug' => false,
|
||
// 数据库部署方式:0 集中式(单一服务器),1 分布式(主从服务器)
|
||
'deploy' => 0,
|
||
// 数据库读写是否分离 主从式有效
|
||
'rw_separate' => false,
|
||
// 读写分离后 主服务器数量
|
||
'master_num' => 1,
|
||
// 指定从服务器序号
|
||
'slave_no' => '',
|
||
// 是否严格检查字段是否存在
|
||
'fields_strict' => true,
|
||
// 数据集返回类型
|
||
'resultset_type' => Db::RESULTSET_ARRAY,
|
||
// 自动写入时间戳字段
|
||
'auto_timestamp' => false,
|
||
];
|
||
|
||
// PDO连接参数
|
||
protected $params = [
|
||
PDO::ATTR_CASE => PDO::CASE_LOWER,
|
||
PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
|
||
PDO::ATTR_ORACLE_NULLS => PDO::NULL_NATURAL,
|
||
PDO::ATTR_STRINGIFY_FETCHES => false,
|
||
PDO::ATTR_EMULATE_PREPARES => false,
|
||
];
|
||
|
||
/**
|
||
* 架构函数 读取数据库配置信息
|
||
* @access public
|
||
* @param array $config 数据库配置数组
|
||
*/
|
||
public function __construct(array $config = [])
|
||
{
|
||
if (!empty($config)) {
|
||
$this->config = array_merge($this->config, $config);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 创建指定模型的查询对象
|
||
* @access public
|
||
* @param string $model 模型类名称
|
||
* @return Query
|
||
*/
|
||
public function model($model)
|
||
{
|
||
if (!isset($this->query[$model])) {
|
||
$this->query[$model] = new Query($this, $model);
|
||
}
|
||
return $this->query[$model];
|
||
}
|
||
|
||
/**
|
||
* 调用Query类的查询方法
|
||
* @access public
|
||
* @param string $method 方法名称
|
||
* @param array $args 调用参数
|
||
* @return mixed
|
||
*/
|
||
public function __call($method, $args)
|
||
{
|
||
if (!isset($this->query['database'])) {
|
||
$this->query['database'] = new Query($this);
|
||
}
|
||
return call_user_func_array([$this->query['database'], $method], $args);
|
||
}
|
||
|
||
/**
|
||
* 解析pdo连接的dsn信息
|
||
* @access protected
|
||
* @param array $config 连接信息
|
||
* @return string
|
||
*/
|
||
abstract protected function parseDsn($config);
|
||
|
||
/**
|
||
* 取得数据表的字段信息
|
||
* @access public
|
||
* @param string $tableName
|
||
* @return array
|
||
*/
|
||
abstract public function getFields($tableName);
|
||
|
||
/**
|
||
* 取得数据库的表信息
|
||
* @access public
|
||
* @param string $dbName
|
||
* @return array
|
||
*/
|
||
abstract public function getTables($dbName);
|
||
|
||
/**
|
||
* SQL性能分析
|
||
* @access protected
|
||
* @param string $sql
|
||
* @return array
|
||
*/
|
||
abstract protected function getExplain($sql);
|
||
|
||
/**
|
||
* 对返数据表字段信息进行大小写转换出来
|
||
* @access public
|
||
* @param array $info 字段信息
|
||
* @return array
|
||
*/
|
||
protected function fieldCase($info)
|
||
{
|
||
// 字段大小写转换
|
||
switch ($this->attrCase) {
|
||
case PDO::CASE_LOWER:
|
||
$info = array_change_key_case($info);
|
||
break;
|
||
case PDO::CASE_UPPER:
|
||
$info = array_change_key_case($info, CASE_UPPER);
|
||
break;
|
||
case PDO::CASE_NATURAL:
|
||
default:
|
||
// 不做转换
|
||
}
|
||
return $info;
|
||
}
|
||
|
||
/**
|
||
* 获取数据库的配置参数
|
||
* @access public
|
||
* @param string $config 配置名称
|
||
* @return mixed
|
||
*/
|
||
public function getConfig($config = '')
|
||
{
|
||
return $config ? $this->config[$config] : $this->config;
|
||
}
|
||
|
||
/**
|
||
* 设置数据库的配置参数
|
||
* @access public
|
||
* @param string $config 配置名称
|
||
* @param mixed $value 配置值
|
||
* @return void
|
||
*/
|
||
public function setConfig($config, $value)
|
||
{
|
||
$this->config[$config] = $value;
|
||
}
|
||
|
||
/**
|
||
* 连接数据库方法
|
||
* @access public
|
||
* @param array $config 连接参数
|
||
* @param integer $linkNum 连接序号
|
||
* @param array|bool $autoConnection 是否自动连接主数据库（用于分布式）
|
||
* @return PDO
|
||
* @throws Exception
|
||
*/
|
||
public function connect(array $config = [], $linkNum = 0, $autoConnection = false)
|
||
{
|
||
if (!isset($this->links[$linkNum])) {
|
||
if (empty($config)) {
|
||
$config = $this->config;
|
||
}
|
||
// 连接参数
|
||
if (isset($config['params']) && is_array($config['params'])) {
|
||
$params = $config['params'] + $this->params;
|
||
} else {
|
||
$params = $this->params;
|
||
}
|
||
// 记录当前字段属性大小写设置
|
||
$this->attrCase = $params[PDO::ATTR_CASE];
|
||
// 记录数据集返回类型
|
||
if (isset($config['resultset_type'])) {
|
||
$this->resultSetType = $config['resultset_type'];
|
||
}
|
||
try {
|
||
if (empty($config['dsn'])) {
|
||
$config['dsn'] = $this->parseDsn($config);
|
||
}
|
||
$this->links[$linkNum] = new PDO($config['dsn'], $config['username'], $config['password'], $params);
|
||
// 记录数据库连接信息
|
||
APP_DEBUG && Log::record('[ DB ] CONNECT: ' . $config['dsn'], 'info');
|
||
} catch (\PDOException $e) {
|
||
if ($autoConnection) {
|
||
Log::record($e->getMessage(), 'error');
|
||
return $this->connect($autoConnection, $linkNum);
|
||
} else {
|
||
throw new Exception($e->getMessage());
|
||
}
|
||
}
|
||
}
|
||
return $this->links[$linkNum];
|
||
}
|
||
|
||
/**
|
||
* 获取当前数据库的驱动类型
|
||
* @access public
|
||
* @return string
|
||
*/
|
||
public function getDriverName()
|
||
{
|
||
if ($this->linkID) {
|
||
return $this->linkID->getAttribute(PDO::ATTR_DRIVER_NAME);
|
||
} else {
|
||
return $this->config['type'];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 释放查询结果
|
||
* @access public
|
||
*/
|
||
public function free()
|
||
{
|
||
$this->PDOStatement = null;
|
||
}
|
||
|
||
/**
|
||
* 获取PDO对象
|
||
* @access public
|
||
* @return \PDO|false
|
||
*/
|
||
public function getPdo()
|
||
{
|
||
if (!$this->linkID) {
|
||
return false;
|
||
} else {
|
||
return $this->linkID;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 执行查询 返回数据集
|
||
* @access public
|
||
* @param string $sql sql指令
|
||
* @param array $bind 参数绑定
|
||
* @param boolean $fetch 不执行只是获取SQL
|
||
* @param boolean $master 是否在主服务器读操作
|
||
* @param bool|string $class 指定返回的数据集对象
|
||
* @return mixed
|
||
* @throws DbBindParamException
|
||
* @throws PDOException
|
||
*/
|
||
public function query($sql, $bind = [], $fetch = false, $master = false, $class = false)
|
||
{
|
||
$this->initConnect($master);
|
||
if (!$this->linkID) {
|
||
return false;
|
||
}
|
||
|
||
// 根据参数绑定组装最终的SQL语句
|
||
$this->queryStr = $this->getBindSql($sql, $bind);
|
||
|
||
if ($fetch) {
|
||
return $this->queryStr;
|
||
}
|
||
//释放前次的查询结果
|
||
if (!empty($this->PDOStatement)) {
|
||
$this->free();
|
||
}
|
||
|
||
Db::$queryTimes++;
|
||
try {
|
||
// 调试开始
|
||
$this->debug(true);
|
||
// 预处理
|
||
$this->PDOStatement = $this->linkID->prepare($sql);
|
||
// 参数绑定
|
||
$this->bindValue($bind);
|
||
// 执行查询
|
||
$result = $this->PDOStatement->execute();
|
||
// 调试结束
|
||
$this->debug(false);
|
||
$procedure = 0 === strpos(strtolower(substr(trim($sql), 0, 4)), 'call');
|
||
return $this->getResult($class, $procedure);
|
||
} catch (\PDOException $e) {
|
||
throw new PDOException($e, $this->config, $this->queryStr);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 执行语句
|
||
* @access public
|
||
* @param string $sql sql指令
|
||
* @param array $bind 参数绑定
|
||
* @param boolean $fetch 不执行只是获取SQL
|
||
* @param boolean $getLastInsID 是否获取自增ID
|
||
* @param string $sequence 自增序列名
|
||
* @return int
|
||
* @throws DbBindParamException
|
||
* @throws PDOException
|
||
*/
|
||
public function execute($sql, $bind = [], $fetch = false, $getLastInsID = false, $sequence = null)
|
||
{
|
||
$this->initConnect(true);
|
||
if (!$this->linkID) {
|
||
return false;
|
||
}
|
||
// 根据参数绑定组装最终的SQL语句
|
||
$this->queryStr = $this->getBindSql($sql, $bind);
|
||
|
||
if ($fetch) {
|
||
return $this->queryStr;
|
||
}
|
||
//释放前次的查询结果
|
||
if (!empty($this->PDOStatement)) {
|
||
$this->free();
|
||
}
|
||
|
||
Db::$executeTimes++;
|
||
try {
|
||
// 调试开始
|
||
$this->debug(true);
|
||
// 预处理
|
||
$this->PDOStatement = $this->linkID->prepare($sql);
|
||
// 参数绑定操作
|
||
$this->bindValue($bind);
|
||
// 执行语句
|
||
$result = $this->PDOStatement->execute();
|
||
// 调试结束
|
||
$this->debug(false);
|
||
|
||
$this->numRows = $this->PDOStatement->rowCount();
|
||
if (preg_match("/^\s*(INSERT\s+INTO|REPLACE\s+INTO)\s+/i", $sql)) {
|
||
$this->lastInsID = $this->linkID->lastInsertId($sequence);
|
||
if ($getLastInsID) {
|
||
return $this->lastInsID;
|
||
}
|
||
}
|
||
return $this->numRows;
|
||
} catch (\PDOException $e) {
|
||
throw new PDOException($e, $this->config, $this->queryStr);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 根据参数绑定组装最终的SQL语句 便于调试
|
||
* @access public
|
||
* @param string $sql 带参数绑定的sql语句
|
||
* @param array $bind 参数绑定列表
|
||
* @return string
|
||
*/
|
||
protected function getBindSql($sql, array $bind = [])
|
||
{
|
||
if ($bind) {
|
||
foreach ($bind as $key => $val) {
|
||
$val = $this->quote(is_array($val) ? $val[0] : $val);
|
||
// 判断占位符
|
||
$sql = is_numeric($key) ?
|
||
substr_replace($sql, $val, strpos($sql, '?'), 1) :
|
||
str_replace(
|
||
[':' . $key . ')', ':' . $key . ',', ':' . $key . ' '],
|
||
[$val . ')', $val . ',', $val . ' '],
|
||
$sql . ' ');
|
||
}
|
||
}
|
||
return $sql;
|
||
}
|
||
|
||
/**
|
||
* 参数绑定
|
||
* 支持 ['name'=>'value','id'=>123] 对应命名占位符
|
||
* 或者 ['value',123] 对应问号占位符
|
||
* @access public
|
||
* @param array $bind 要绑定的参数列表
|
||
* @return void
|
||
* @throws \think\Exception
|
||
*/
|
||
protected function bindValue(array $bind = [])
|
||
{
|
||
foreach ($bind as $key => $val) {
|
||
// 占位符
|
||
$param = is_numeric($key) ? $key + 1 : ':' . $key;
|
||
if (is_array($val)) {
|
||
$result = $this->PDOStatement->bindValue($param, $val[0], $val[1]);
|
||
} else {
|
||
$result = $this->PDOStatement->bindValue($param, $val);
|
||
}
|
||
if (!$result) {
|
||
throw new DbBindParamException(
|
||
"Error occurred when binding parameters '{$param}'",
|
||
$this->config,
|
||
$this->queryStr,
|
||
$bind
|
||
);
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获得数据集
|
||
* @access protected
|
||
* @param bool|string $class true 返回PDOStatement 字符串用于指定返回的类名
|
||
* @param bool $procedure 是否存储过程
|
||
* @return mixed
|
||
*/
|
||
protected function getResult($class = '', $procedure = false)
|
||
{
|
||
if (true === $class) {
|
||
// 返回PDOStatement对象处理
|
||
return $this->PDOStatement;
|
||
}
|
||
if ($procedure) {
|
||
return $this->procedure($class);
|
||
}
|
||
$result = $this->PDOStatement->fetchAll($this->fetchType);
|
||
$this->numRows = count($result);
|
||
|
||
if (!empty($class)) {
|
||
// 返回指定数据集对象类
|
||
return new $class($result);
|
||
}
|
||
switch ($this->resultSetType) {
|
||
case Db::RESULTSET_COLLECTION:
|
||
// 返回数据集Collection对象
|
||
$result = new Collection($result);
|
||
break;
|
||
case Db::RESULTSET_CLASS:
|
||
break;
|
||
case Db::RESULTSET_ARRAY:
|
||
default:
|
||
// 返回二维数组
|
||
}
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 获得存储过程数据集
|
||
* @access protected
|
||
* @param bool|string $class true 返回PDOStatement 字符串用于指定返回的类名
|
||
* @return array
|
||
*/
|
||
protected function procedure($class)
|
||
{
|
||
$item = [];
|
||
do {
|
||
$result = $this->getResult($class);
|
||
if ($result) {
|
||
$item[] = $result;
|
||
}
|
||
} while ($this->PDOStatement->nextRowset());
|
||
$this->numRows = count($item);
|
||
return $item;
|
||
}
|
||
|
||
/**
|
||
* 执行数据库事务
|
||
* @access public
|
||
* @param callable $callback 数据操作方法回调
|
||
* @return mixed
|
||
*/
|
||
public function transaction($callback)
|
||
{
|
||
$label = microtime(true);
|
||
$this->startTrans($label);
|
||
try {
|
||
$result = null;
|
||
if (is_callable($callback)) {
|
||
$result = call_user_func_array($callback, []);
|
||
}
|
||
$this->commit($label);
|
||
return $result;
|
||
} catch (\Exception $e) {
|
||
$this->rollback();
|
||
throw $e;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 启动事务
|
||
* @access public
|
||
* @param string $label 事务标识
|
||
* @return bool|null
|
||
*/
|
||
public function startTrans($label = '')
|
||
{
|
||
$this->initConnect(true);
|
||
if (!$this->linkID) {
|
||
return false;
|
||
}
|
||
|
||
//数据rollback 支持
|
||
if (0 == $this->transTimes) {
|
||
$this->transLabel = $label;
|
||
$this->linkID->beginTransaction();
|
||
}
|
||
$this->transTimes++;
|
||
return;
|
||
}
|
||
|
||
/**
|
||
* 用于非自动提交状态下面的查询提交
|
||
* @access public
|
||
* @param string $label 事务标识
|
||
* @return boolean
|
||
* @throws PDOException
|
||
*/
|
||
public function commit($label = '')
|
||
{
|
||
$this->initConnect(true);
|
||
if ($this->transTimes > 0 && $label == $this->transLabel) {
|
||
try {
|
||
$this->linkID->commit();
|
||
$this->transTimes = 0;
|
||
} catch (\PDOException $e) {
|
||
throw new PDOException($e, $this->config, $this->queryStr);
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 事务回滚
|
||
* @access public
|
||
* @return boolean
|
||
* @throws PDOException
|
||
*/
|
||
public function rollback()
|
||
{
|
||
$this->initConnect(true);
|
||
if ($this->transTimes > 0) {
|
||
try {
|
||
$this->linkID->rollback();
|
||
$this->transTimes = 0;
|
||
} catch (\PDOException $e) {
|
||
throw new PDOException($e, $this->config, $this->queryStr);
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 批处理执行SQL语句
|
||
* 批处理的指令都认为是execute操作
|
||
* @access public
|
||
* @param array $sqlArray SQL批处理指令
|
||
* @return boolean
|
||
*/
|
||
public function batchQuery($sqlArray = [])
|
||
{
|
||
if (!is_array($sqlArray)) {
|
||
return false;
|
||
}
|
||
// 自动启动事务支持
|
||
$label = microtime(true);
|
||
$this->startTrans($label);
|
||
try {
|
||
foreach ($sqlArray as $sql) {
|
||
$result = $this->execute($sql);
|
||
}
|
||
// 提交事务
|
||
$this->commit($label);
|
||
} catch (\PDOException $e) {
|
||
$this->rollback();
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 将SQL语句中的__TABLE_NAME__字符串替换成带前缀的表名（小写）
|
||
* @access public
|
||
* @param string $sql sql语句
|
||
* @return string
|
||
*/
|
||
public function parseSqlTable($sql)
|
||
{
|
||
if (false !== strpos($sql, '__')) {
|
||
$prefix = $this->config['prefix'];
|
||
$sql = preg_replace_callback("/__([A-Z0-9_-]+)__/sU", function ($match) use ($prefix) {
|
||
return $prefix . strtolower($match[1]);
|
||
}, $sql);
|
||
}
|
||
return $sql;
|
||
}
|
||
|
||
/**
|
||
* 获得查询次数
|
||
* @access public
|
||
* @param boolean $execute 是否包含所有查询
|
||
* @return integer
|
||
*/
|
||
public function getQueryTimes($execute = false)
|
||
{
|
||
return $execute ? Db::$queryTimes + Db::$executeTimes : Db::$queryTimes;
|
||
}
|
||
|
||
/**
|
||
* 获得执行次数
|
||
* @access public
|
||
* @return integer
|
||
*/
|
||
public function getExecuteTimes()
|
||
{
|
||
return Db::$executeTimes;
|
||
}
|
||
|
||
/**
|
||
* 关闭数据库
|
||
* @access public
|
||
*/
|
||
public function close()
|
||
{
|
||
$this->linkID = null;
|
||
}
|
||
|
||
/**
|
||
* 获取最近一次查询的sql语句
|
||
* @access public
|
||
* @return string
|
||
*/
|
||
public function getLastSql()
|
||
{
|
||
return $this->queryStr;
|
||
}
|
||
|
||
/**
|
||
* 获取最近插入的ID
|
||
* @access public
|
||
* @return string
|
||
*/
|
||
public function getLastInsID()
|
||
{
|
||
return $this->lastInsID;
|
||
}
|
||
|
||
/**
|
||
* 获取最近的错误信息
|
||
* @access public
|
||
* @return string
|
||
*/
|
||
public function getError()
|
||
{
|
||
if ($this->PDOStatement) {
|
||
$error = $this->PDOStatement->errorInfo();
|
||
$error = $error[1] . ':' . $error[2];
|
||
} else {
|
||
$error = '';
|
||
}
|
||
if ('' != $this->queryStr) {
|
||
$error .= "\n [ SQL语句 ] : " . $this->queryStr;
|
||
}
|
||
return $error;
|
||
}
|
||
|
||
/**
|
||
* SQL指令安全过滤
|
||
* @access public
|
||
* @param string $str SQL字符串
|
||
* @return string
|
||
*/
|
||
public function quote($str)
|
||
{
|
||
$this->initConnect();
|
||
return $this->linkID ? $this->linkID->quote($str) : $str;
|
||
}
|
||
|
||
/**
|
||
* 数据库调试 记录当前SQL及分析性能
|
||
* @access protected
|
||
* @param boolean $start 调试开始标记 true 开始 false 结束
|
||
* @return void
|
||
*/
|
||
protected function debug($start)
|
||
{
|
||
if (!empty($this->config['debug'])) {
|
||
// 开启数据库调试模式
|
||
if ($start) {
|
||
Debug::remark('queryStartTime', 'time');
|
||
} else {
|
||
// 记录操作结束时间
|
||
Debug::remark('queryEndTime', 'time');
|
||
$runtime = Debug::getRangeTime('queryStartTime', 'queryEndTime');
|
||
$log = $this->queryStr . ' [ RunTime:' . $runtime . 's ]';
|
||
$result = [];
|
||
// SQL性能分析
|
||
if (0 === stripos(trim($this->queryStr), 'select')) {
|
||
$result = $this->getExplain($this->queryStr);
|
||
}
|
||
// SQL监听
|
||
$this->trigger($this->queryStr, $runtime, $result);
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 监听SQL执行
|
||
* @access public
|
||
* @param callable $callback 回调方法
|
||
* @return void
|
||
*/
|
||
public function listen($callback)
|
||
{
|
||
self::$event[] = $callback;
|
||
}
|
||
|
||
/**
|
||
* 触发SQL事件
|
||
* @access protected
|
||
* @param string $sql SQL语句
|
||
* @param float $runtime SQL运行时间
|
||
* @param mixed $explain SQL分析
|
||
* @return bool
|
||
*/
|
||
protected function trigger($sql, $runtime, $explain = [])
|
||
{
|
||
if (!empty(self::$event)) {
|
||
foreach (self::$event as $callback) {
|
||
if (is_callable($callback)) {
|
||
call_user_func_array($callback, [$sql, $runtime, $explain]);
|
||
}
|
||
}
|
||
} else {
|
||
// 未注册监听则记录到日志中
|
||
Log::record('[ SQL ] ' . $this->queryStr . ' [ RunTime:' . $runtime . 's ]', 'sql');
|
||
if (!empty($explain)) {
|
||
Log::record('[ EXPLAIN : ' . var_export($explain, true) . ' ]', 'sql');
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 初始化数据库连接
|
||
* @access protected
|
||
* @param boolean $master 是否主服务器
|
||
* @return void
|
||
*/
|
||
protected function initConnect($master = true)
|
||
{
|
||
if (!empty($this->config['deploy'])) {
|
||
// 采用分布式数据库
|
||
if ($master) {
|
||
if (!$this->linkWrite) {
|
||
$this->linkWrite = $this->multiConnect(true);
|
||
}
|
||
$this->linkID = $this->linkWrite;
|
||
} else {
|
||
if (!$this->linkRead) {
|
||
$this->linkRead = $this->multiConnect(false);
|
||
}
|
||
$this->linkID = $this->linkRead;
|
||
}
|
||
} elseif (!$this->linkID) {
|
||
// 默认单数据库
|
||
$this->linkID = $this->connect();
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 连接分布式服务器
|
||
* @access protected
|
||
* @param boolean $master 主服务器
|
||
* @return PDO
|
||
*/
|
||
protected function multiConnect($master = false)
|
||
{
|
||
$_config = [];
|
||
// 分布式数据库配置解析
|
||
foreach (['username', 'password', 'hostname', 'hostport', 'database', 'dsn', 'charset'] as $name) {
|
||
$_config[$name] = explode(',', $this->config[$name]);
|
||
}
|
||
|
||
// 主服务器序号
|
||
$m = floor(mt_rand(0, $this->config['master_num'] - 1));
|
||
|
||
if ($this->config['rw_separate']) {
|
||
// 主从式采用读写分离
|
||
if ($master) // 主服务器写入
|
||
{
|
||
$r = $m;
|
||
} elseif (is_numeric($this->config['slave_no'])) {
|
||
// 指定服务器读
|
||
$r = $this->config['slave_no'];
|
||
} else {
|
||
// 读操作连接从服务器 每次随机连接的数据库
|
||
$r = floor(mt_rand($this->config['master_num'], count($_config['hostname']) - 1));
|
||
}
|
||
} else {
|
||
// 读写操作不区分服务器 每次随机连接的数据库
|
||
$r = floor(mt_rand(0, count($_config['hostname']) - 1));
|
||
}
|
||
$dbMaster = false;
|
||
if ($m != $r) {
|
||
$dbMaster = [];
|
||
foreach (['username', 'password', 'hostname', 'hostport', 'database', 'dsn', 'charset'] as $name) {
|
||
$dbMaster[$name] = isset($_config[$name][$m]) ? $_config[$name][$m] : $_config[$name][0];
|
||
}
|
||
}
|
||
$dbConfig = [];
|
||
foreach (['username', 'password', 'hostname', 'hostport', 'database', 'dsn', 'charset'] as $name) {
|
||
$dbConfig[$name] = isset($_config[$name][$r]) ? $_config[$name][$r] : $_config[$name][0];
|
||
}
|
||
return $this->connect($dbConfig, $r, $r == $m ? false : $dbMaster);
|
||
}
|
||
|
||
/**
|
||
* 析构方法
|
||
* @access public
|
||
*/
|
||
public function __destruct()
|
||
{
|
||
// 释放查询
|
||
if ($this->PDOStatement) {
|
||
$this->free();
|
||
}
|
||
// 关闭连接
|
||
$this->close();
|
||
}
|
||
}
|