mirror of
https://gitee.com/fastadminnet/framework.git
synced 2026-07-01 20:52:48 +08:00
1255 lines
42 KiB
PHP
1255 lines
42 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 think\Config;
|
||
use think\Db;
|
||
use think\Debug;
|
||
use think\Exception;
|
||
use think\Log;
|
||
|
||
abstract class Driver
|
||
{
|
||
// PDO操作实例
|
||
protected $PDOStatement = null;
|
||
// 当前操作所属的模型名
|
||
protected $model = '_think_';
|
||
// 当前SQL指令
|
||
protected $queryStr = '';
|
||
protected $modelSql = [];
|
||
// 最后插入ID
|
||
protected $lastInsID = null;
|
||
// 返回或者影响记录数
|
||
protected $numRows = 0;
|
||
// 事务指令数
|
||
protected $transTimes = 0;
|
||
// 错误信息
|
||
protected $error = '';
|
||
// 数据库连接ID 支持多个连接
|
||
protected $linkID = [];
|
||
// 当前连接ID
|
||
protected $_linkID = null;
|
||
// 数据库连接参数配置
|
||
protected $config = [
|
||
// 数据库类型
|
||
'type' => '',
|
||
// 服务器地址
|
||
'hostname' => '127.0.0.1',
|
||
// 数据库名
|
||
'database' => '',
|
||
// 用户名
|
||
'username' => '',
|
||
// 密码
|
||
'password' => '',
|
||
// 端口
|
||
'hostport' => '',
|
||
'dsn' => '',
|
||
// 数据库连接参数
|
||
'params' => [],
|
||
// 数据库编码默认采用utf8
|
||
'charset' => 'utf8',
|
||
// 数据库表前缀
|
||
'prefix' => '',
|
||
// 数据库调试模式
|
||
'debug' => false,
|
||
// 数据库部署方式:0 集中式(单一服务器),1 分布式(主从服务器)
|
||
'deploy' => 0,
|
||
// 数据库读写是否分离 主从式有效
|
||
'rw_separate' => false,
|
||
// 读写分离后 主服务器数量
|
||
'master_num' => 1,
|
||
// 指定从服务器序号
|
||
'slave_no' => '',
|
||
// like字段自动替换为%%包裹
|
||
'db_like_fields' => '',
|
||
// 是否开启数据库调试
|
||
'debug' => false,
|
||
];
|
||
// 数据库表达式
|
||
protected $exp = ['eq' => '=', 'neq' => '<>', 'gt' => '>', 'egt' => '>=', 'lt' => '<', 'elt' => '<=', 'notlike' => 'NOT LIKE', 'like' => 'LIKE', 'in' => 'IN', 'notin' => 'NOT IN', 'not in' => 'NOT IN', 'between' => 'BETWEEN', 'not between' => 'NOT BETWEEN', 'notbetween' => 'NOT BETWEEN'];
|
||
// 查询表达式
|
||
protected $selectSql = 'SELECT%DISTINCT% %FIELD% FROM %TABLE%%FORCE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%%LIMIT% %UNION%%LOCK%%COMMENT%';
|
||
|
||
// PDO连接参数
|
||
protected $options = [
|
||
PDO::ATTR_CASE => PDO::CASE_LOWER,
|
||
PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
|
||
PDO::ATTR_ORACLE_NULLS => PDO::NULL_NATURAL,
|
||
PDO::ATTR_STRINGIFY_FETCHES => false,
|
||
];
|
||
// 参数绑定
|
||
protected $bind = [];
|
||
|
||
/**
|
||
* 架构函数 读取数据库配置信息
|
||
* @access public
|
||
* @param array $config 数据库配置数组
|
||
*/
|
||
public function __construct($config = '')
|
||
{
|
||
if (!empty($config)) {
|
||
$this->config = array_merge($this->config, $config);
|
||
if (is_array($this->config['params'])) {
|
||
$this->options = $this->config['params'] + $this->options;
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 连接数据库方法
|
||
* @access public
|
||
*/
|
||
public function connect($config = '', $linkNum = 0, $autoConnection = false)
|
||
{
|
||
if (!isset($this->linkID[$linkNum])) {
|
||
if (empty($config)) {
|
||
$config = $this->config;
|
||
}
|
||
|
||
try {
|
||
if (empty($config['dsn'])) {
|
||
$config['dsn'] = $this->parseDsn($config);
|
||
}
|
||
$this->linkID[$linkNum] = new PDO($config['dsn'], $config['username'], $config['password'], $this->options);
|
||
} catch (\PDOException $e) {
|
||
if ($autoConnection) {
|
||
Log::record($e->getMessage(), 'error');
|
||
return $this->connect($autoConnection, $linkNum);
|
||
} else {
|
||
throw new Exception($e->getMessage());
|
||
}
|
||
}
|
||
}
|
||
return $this->linkID[$linkNum];
|
||
}
|
||
|
||
/**
|
||
* 解析pdo连接的dsn信息
|
||
* @access public
|
||
* @param array $config 连接信息
|
||
* @return string
|
||
*/
|
||
protected function parseDsn($config)
|
||
{}
|
||
|
||
/**
|
||
* 释放查询结果
|
||
* @access public
|
||
*/
|
||
public function free()
|
||
{
|
||
$this->PDOStatement = null;
|
||
}
|
||
|
||
/**
|
||
* 执行查询 返回数据集
|
||
* @access public
|
||
* @param string $sql sql指令
|
||
* @param array $bind 参数绑定
|
||
* @param boolean $fetch 不执行只是获取SQL
|
||
* @param boolean $master 是否在主服务器读操作
|
||
* @return mixed
|
||
*/
|
||
public function query($sql, $bind = [], $fetch = false, $master = false)
|
||
{
|
||
$this->initConnect($master);
|
||
if (!$this->_linkID) {
|
||
return false;
|
||
}
|
||
|
||
$this->queryStr = $sql;
|
||
|
||
if ($bind) {
|
||
$this->queryStr .= '[ ' . print_r($bind, true) . ' ]';
|
||
}
|
||
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);
|
||
return $this->getResult();
|
||
} catch (\PDOException $e) {
|
||
throw new Exception($e->getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 执行语句
|
||
* @access public
|
||
* @param string $sql sql指令
|
||
* @param array $bind 参数绑定
|
||
* @param boolean $fetch 不执行只是获取SQL
|
||
* @return integer
|
||
*/
|
||
public function execute($sql, $bind = [], $fetch = false)
|
||
{
|
||
$this->initConnect(true);
|
||
if (!$this->_linkID) {
|
||
return false;
|
||
}
|
||
|
||
$this->queryStr = $sql;
|
||
|
||
if ($bind) {
|
||
$this->queryStr .= '[ ' . print_r($bind, true) . ' ]';
|
||
}
|
||
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();
|
||
}
|
||
return $this->numRows;
|
||
} catch (\PDOException $e) {
|
||
throw new Exception($e->getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 参数绑定
|
||
* 支持 ['name'=>'value','id'=>123] 对应命名占位符
|
||
* 或者 ['value',123] 对应问号占位符
|
||
* @access public
|
||
* @param array $bind 要绑定的参数列表
|
||
* @return void
|
||
*/
|
||
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 Exception('bind param error : [ ' . $param . '=>' . $val . ' ]');
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 启动事务
|
||
* @access public
|
||
* @return void
|
||
*/
|
||
public function startTrans()
|
||
{
|
||
$this->initConnect(true);
|
||
if (!$this->_linkID) {
|
||
return false;
|
||
}
|
||
|
||
//数据rollback 支持
|
||
if (0 == $this->transTimes) {
|
||
$this->_linkID->beginTransaction();
|
||
}
|
||
$this->transTimes++;
|
||
return;
|
||
}
|
||
|
||
/**
|
||
* 用于非自动提交状态下面的查询提交
|
||
* @access public
|
||
* @return boolen
|
||
*/
|
||
public function commit()
|
||
{
|
||
if ($this->transTimes > 0) {
|
||
try {
|
||
$result = $this->_linkID->commit();
|
||
$this->transTimes = 0;
|
||
} catch (\PDOException $e) {
|
||
throw new Exception($e->getMessage());
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 事务回滚
|
||
* @access public
|
||
* @return boolen
|
||
*/
|
||
public function rollback()
|
||
{
|
||
if ($this->transTimes > 0) {
|
||
try {
|
||
$result = $this->_linkID->rollback();
|
||
$this->transTimes = 0;
|
||
} catch (\PDOException $e) {
|
||
throw new Exception($e->getMessage());
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 获得所有的查询数据
|
||
* @access private
|
||
* @return array
|
||
*/
|
||
private function getResult()
|
||
{
|
||
//返回数据集
|
||
$result = $this->PDOStatement->fetchAll(PDO::FETCH_ASSOC);
|
||
$this->numRows = count($result);
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 获得查询次数
|
||
* @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;
|
||
}
|
||
|
||
/**
|
||
* 设置锁机制
|
||
* @access protected
|
||
* @return string
|
||
*/
|
||
protected function parseLock($lock = false)
|
||
{
|
||
return $lock ? ' FOR UPDATE ' : '';
|
||
}
|
||
|
||
/**
|
||
* set分析
|
||
* @access protected
|
||
* @param array $data
|
||
* @return string
|
||
*/
|
||
protected function parseSet($data)
|
||
{
|
||
foreach ($data as $key => $val) {
|
||
if (isset($val[0]) && 'exp' == $val[0]) {
|
||
$set[] = $this->parseKey($key) . '=' . $val[1];
|
||
} elseif (is_null($val)) {
|
||
$set[] = $this->parseKey($key) . '=NULL';
|
||
} elseif (is_scalar($val)) {
|
||
// 过滤非标量数据
|
||
if (0 === strpos($val, ':') && isset($this->bind[substr($val, 1)])) {
|
||
$set[] = $this->parseKey($key) . '=' . $val;
|
||
} else {
|
||
$name = count($this->bind);
|
||
$set[] = $this->parseKey($key) . '=:' . $key . $_SERVER['REQUEST_TIME'] . '_' . $name;
|
||
$this->bindParam($key . $_SERVER['REQUEST_TIME'] . '_' . $name, $val);
|
||
}
|
||
}
|
||
}
|
||
return ' SET ' . implode(',', $set);
|
||
}
|
||
|
||
/**
|
||
* 参数绑定
|
||
* @access protected
|
||
* @param string $name 绑定参数名
|
||
* @param mixed $value 绑定值
|
||
* @return void
|
||
*/
|
||
protected function bindParam($name, $value)
|
||
{
|
||
$this->bind[$name] = $value;
|
||
}
|
||
|
||
/**
|
||
* 获取参数绑定信息并清空
|
||
* @access protected
|
||
* @param bool $reset 获取后清空
|
||
* @return array
|
||
*/
|
||
protected function getBindParams($reset = false)
|
||
{
|
||
$bind = $this->bind;
|
||
if ($reset) {
|
||
$this->bind = [];
|
||
}
|
||
return $bind;
|
||
}
|
||
/**
|
||
* 字段名分析
|
||
* @access protected
|
||
* @param string $key
|
||
* @return string
|
||
*/
|
||
protected function parseKey($key)
|
||
{
|
||
return $key;
|
||
}
|
||
|
||
/**
|
||
* value分析
|
||
* @access protected
|
||
* @param mixed $value
|
||
* @return string
|
||
*/
|
||
protected function parseValue($value)
|
||
{
|
||
if (is_string($value)) {
|
||
$value = strpos($value, ':') === 0 && isset($this->bind[substr($value, 1)]) ? $value : $this->quote($value);
|
||
} elseif (isset($value[0]) && is_string($value[0]) && strtolower($value[0]) == 'exp') {
|
||
$value = $this->quote($value[1]);
|
||
} elseif (is_array($value)) {
|
||
$value = array_map([$this, 'parseValue'], $value);
|
||
} elseif (is_bool($value)) {
|
||
$value = $value ? '1' : '0';
|
||
} elseif (is_null($value)) {
|
||
$value = 'null';
|
||
}
|
||
return $value;
|
||
}
|
||
|
||
/**
|
||
* field分析
|
||
* @access protected
|
||
* @param mixed $fields
|
||
* @return string
|
||
*/
|
||
protected function parseField($fields)
|
||
{
|
||
if (is_string($fields) && strpos($fields, ',')) {
|
||
$fields = explode(',', $fields);
|
||
}
|
||
if (is_array($fields)) {
|
||
// 完善数组方式传字段名的支持
|
||
// 支持 'field1'=>'field2' 这样的字段别名定义
|
||
$array = [];
|
||
foreach ($fields as $key => $field) {
|
||
if (!is_numeric($key)) {
|
||
$array[] = $this->parseKey($key) . ' AS ' . $this->parseKey($field);
|
||
} else {
|
||
$array[] = $this->parseKey($field);
|
||
}
|
||
|
||
}
|
||
$fieldsStr = implode(',', $array);
|
||
} elseif (is_string($fields) && !empty($fields)) {
|
||
$fieldsStr = $this->parseKey($fields);
|
||
} else {
|
||
$fieldsStr = '*';
|
||
}
|
||
//TODO 如果是查询全部字段，并且是join的方式，那么就把要查的表加个别名，以免字段被覆盖
|
||
return $fieldsStr;
|
||
}
|
||
|
||
/**
|
||
* table分析
|
||
* @access protected
|
||
* @param mixed $table
|
||
* @return string
|
||
*/
|
||
protected function parseTable($tables)
|
||
{
|
||
if (is_array($tables)) {
|
||
// 支持别名定义
|
||
$array = [];
|
||
foreach ($tables as $table => $alias) {
|
||
if (!is_numeric($table)) {
|
||
$array[] = $this->parseKey($table) . ' ' . $this->parseKey($alias);
|
||
} else {
|
||
$array[] = $this->parseKey($alias);
|
||
}
|
||
}
|
||
$tables = $array;
|
||
} elseif (is_string($tables)) {
|
||
$tables = array_map([$this, 'parseKey'], explode(',', $tables));
|
||
}
|
||
return implode(',', $tables);
|
||
}
|
||
|
||
/**
|
||
* where分析
|
||
* @access protected
|
||
* @param mixed $where
|
||
* @return string
|
||
*/
|
||
protected function parseWhere($where)
|
||
{
|
||
$whereStr = '';
|
||
if (is_string($where)) {
|
||
// 直接使用字符串条件
|
||
$whereStr = $where;
|
||
} else {
|
||
// 使用数组表达式
|
||
$operate = isset($where['_logic']) ? strtoupper($where['_logic']) : '';
|
||
if (in_array($operate, ['AND', 'OR', 'XOR'])) {
|
||
// 定义逻辑运算规则 例如 OR XOR AND NOT
|
||
$operate = ' ' . $operate . ' ';
|
||
unset($where['_logic']);
|
||
} else {
|
||
// 默认进行 AND 运算
|
||
$operate = ' AND ';
|
||
}
|
||
foreach ($where as $key => $val) {
|
||
if (is_numeric($key)) {
|
||
$key = '_complex';
|
||
}
|
||
if (0 === strpos($key, '_')) {
|
||
// 解析特殊条件表达式
|
||
$whereStr .= $this->parseThinkWhere($key, $val);
|
||
} else {
|
||
// 多条件支持
|
||
$multi = is_array($val) && isset($val['_multi']);
|
||
$key = trim($key);
|
||
if (strpos($key, '|')) {
|
||
// 支持 name|title|nickname 方式定义查询字段
|
||
$array = explode('|', $key);
|
||
$str = [];
|
||
foreach ($array as $m => $k) {
|
||
$v = $multi ? $val[$m] : $val;
|
||
$str[] = $this->parseWhereItem($this->parseKey($k), $v);
|
||
}
|
||
$whereStr .= '( ' . implode(' OR ', $str) . ' )';
|
||
} elseif (strpos($key, '&')) {
|
||
$array = explode('&', $key);
|
||
$str = [];
|
||
foreach ($array as $m => $k) {
|
||
$v = $multi ? $val[$m] : $val;
|
||
$str[] = '(' . $this->parseWhereItem($this->parseKey($k), $v) . ')';
|
||
}
|
||
$whereStr .= '( ' . implode(' AND ', $str) . ' )';
|
||
} else {
|
||
$whereStr .= $this->parseWhereItem($this->parseKey($key), $val);
|
||
}
|
||
}
|
||
$whereStr .= $operate;
|
||
}
|
||
$whereStr = substr($whereStr, 0, -strlen($operate));
|
||
}
|
||
return empty($whereStr) ? '' : ' WHERE ' . $whereStr;
|
||
}
|
||
|
||
// where子单元分析
|
||
protected function parseWhereItem($key, $val)
|
||
{
|
||
$whereStr = '';
|
||
if (is_array($val)) {
|
||
if (is_string($val[0])) {
|
||
$exp = strtolower($val[0]);
|
||
if (preg_match('/^(eq|neq|gt|egt|lt|elt)$/', $exp)) {
|
||
// 比较运算
|
||
$whereStr .= $key . ' ' . $this->exp[$exp] . ' ' . $this->parseValue($val[1]);
|
||
} elseif (preg_match('/^(notlike|like)$/', $exp)) {
|
||
// 模糊查找
|
||
if (is_array($val[1])) {
|
||
$likeLogic = isset($val[2]) ? strtoupper($val[2]) : 'OR';
|
||
if (in_array($likeLogic, ['AND', 'OR', 'XOR'])) {
|
||
$like = [];
|
||
foreach ($val[1] as $item) {
|
||
$like[] = $key . ' ' . $this->exp[$exp] . ' ' . $this->parseValue($item);
|
||
}
|
||
$whereStr .= '(' . implode(' ' . $likeLogic . ' ', $like) . ')';
|
||
}
|
||
} else {
|
||
$whereStr .= $key . ' ' . $this->exp[$exp] . ' ' . $this->parseValue($val[1]);
|
||
}
|
||
} elseif ('exp' == $exp) {
|
||
// 使用表达式
|
||
$whereStr .= $key . ' ' . $val[1];
|
||
} elseif (preg_match('/^(notin|not in|in)$/', $exp)) {
|
||
// IN 运算
|
||
if (isset($val[2]) && 'exp' == $val[2]) {
|
||
$whereStr .= $key . ' ' . $this->exp[$exp] . ' ' . $val[1];
|
||
} else {
|
||
if (is_string($val[1])) {
|
||
$val[1] = explode(',', $val[1]);
|
||
}
|
||
$zone = implode(',', $this->parseValue($val[1]));
|
||
$whereStr .= $key . ' ' . $this->exp[$exp] . ' (' . $zone . ')';
|
||
}
|
||
} elseif (preg_match('/^(notbetween|not between|between)$/', $exp)) {
|
||
// BETWEEN运算
|
||
$data = is_string($val[1]) ? explode(',', $val[1]) : $val[1];
|
||
$whereStr .= $key . ' ' . $this->exp[$exp] . ' ' . $this->parseValue($data[0]) . ' AND ' . $this->parseValue($data[1]);
|
||
} else {
|
||
throw new Exception('where express error:' . $val[0]);
|
||
}
|
||
} else {
|
||
$count = count($val);
|
||
$rule = isset($val[$count - 1]) ? (is_array($val[$count - 1]) ? strtoupper($val[$count - 1][0]) : strtoupper($val[$count - 1])) : '';
|
||
if (in_array($rule, ['AND', 'OR', 'XOR'])) {
|
||
$count = $count - 1;
|
||
} else {
|
||
$rule = 'AND';
|
||
}
|
||
for ($i = 0; $i < $count; $i++) {
|
||
$data = is_array($val[$i]) ? $val[$i][1] : $val[$i];
|
||
if ('exp' == strtolower($val[$i][0])) {
|
||
$whereStr .= $key . ' ' . $data . ' ' . $rule . ' ';
|
||
} else {
|
||
$whereStr .= $this->parseWhereItem($key, $val[$i]) . ' ' . $rule . ' ';
|
||
}
|
||
}
|
||
$whereStr = '( ' . substr($whereStr, 0, -4) . ' )';
|
||
}
|
||
} else {
|
||
//对字符串类型字段采用模糊匹配
|
||
$likeFields = $this->config['db_like_fields'];
|
||
if ($likeFields && preg_match('/^(' . $likeFields . ')$/i', $key)) {
|
||
$whereStr .= $key . ' LIKE ' . $this->parseValue('%' . $val . '%');
|
||
} else {
|
||
$whereStr .= $key . ' = ' . $this->parseValue($val);
|
||
}
|
||
}
|
||
return $whereStr;
|
||
}
|
||
|
||
/**
|
||
* 特殊条件分析
|
||
* @access protected
|
||
* @param string $key
|
||
* @param mixed $val
|
||
* @return string
|
||
*/
|
||
protected function parseThinkWhere($key, $val)
|
||
{
|
||
$whereStr = '';
|
||
switch ($key) {
|
||
case '_string':
|
||
// 字符串模式查询条件
|
||
$whereStr = $val;
|
||
break;
|
||
case '_complex':
|
||
// 复合查询条件
|
||
$whereStr = substr($this->parseWhere($val), 6);
|
||
break;
|
||
case '_query':
|
||
// 字符串模式查询条件
|
||
parse_str($val, $where);
|
||
if (isset($where['_logic'])) {
|
||
$op = ' ' . strtoupper($where['_logic']) . ' ';
|
||
unset($where['_logic']);
|
||
} else {
|
||
$op = ' AND ';
|
||
}
|
||
$array = [];
|
||
foreach ($where as $field => $data) {
|
||
$array[] = $this->parseKey($field) . ' = ' . $this->parseValue($data);
|
||
}
|
||
|
||
$whereStr = implode($op, $array);
|
||
break;
|
||
}
|
||
return $whereStr;
|
||
}
|
||
|
||
/**
|
||
* limit分析
|
||
* @access protected
|
||
* @param mixed $lmit
|
||
* @return string
|
||
*/
|
||
protected function parseLimit($limit)
|
||
{
|
||
return (!empty($limit) && false === strpos($limit, '(')) ? ' LIMIT ' . $limit . ' ' : '';
|
||
}
|
||
|
||
/**
|
||
* join分析
|
||
* @access protected
|
||
* @param mixed $join
|
||
* @return string
|
||
*/
|
||
protected function parseJoin($join)
|
||
{
|
||
$joinStr = '';
|
||
if (!empty($join)) {
|
||
$joinStr = ' ' . implode(' ', $join) . ' ';
|
||
}
|
||
return $joinStr;
|
||
}
|
||
|
||
/**
|
||
* order分析
|
||
* @access protected
|
||
* @param mixed $order
|
||
* @return string
|
||
*/
|
||
protected function parseOrder($order)
|
||
{
|
||
if (is_array($order)) {
|
||
$array = [];
|
||
foreach ($order as $key => $val) {
|
||
if (is_numeric($key)) {
|
||
if (false === strpos($val, '(')) {
|
||
$array[] = $this->parseKey($val);
|
||
} elseif ('[rand]' == $val) {
|
||
$array[] = $this->parseRand();
|
||
}
|
||
} else {
|
||
$sort = in_array(strtolower(trim($val)), ['asc', 'desc']) ? ' ' . $val : '';
|
||
$array[] = $this->parseKey($key) . ' ' . $sort;
|
||
}
|
||
}
|
||
$order = implode(',', $array);
|
||
}
|
||
return !empty($order) ? ' ORDER BY ' . $order : '';
|
||
}
|
||
|
||
/**
|
||
* group分析
|
||
* @access protected
|
||
* @param mixed $group
|
||
* @return string
|
||
*/
|
||
protected function parseGroup($group)
|
||
{
|
||
return !empty($group) ? ' GROUP BY ' . $group : '';
|
||
}
|
||
|
||
/**
|
||
* having分析
|
||
* @access protected
|
||
* @param string $having
|
||
* @return string
|
||
*/
|
||
protected function parseHaving($having)
|
||
{
|
||
return !empty($having) ? ' HAVING ' . $having : '';
|
||
}
|
||
|
||
/**
|
||
* comment分析
|
||
* @access protected
|
||
* @param string $comment
|
||
* @return string
|
||
*/
|
||
protected function parseComment($comment)
|
||
{
|
||
return !empty($comment) ? ' /* ' . $comment . ' */' : '';
|
||
}
|
||
|
||
/**
|
||
* distinct分析
|
||
* @access protected
|
||
* @param mixed $distinct
|
||
* @return string
|
||
*/
|
||
protected function parseDistinct($distinct)
|
||
{
|
||
return !empty($distinct) ? ' DISTINCT ' : '';
|
||
}
|
||
|
||
/**
|
||
* union分析
|
||
* @access protected
|
||
* @param mixed $union
|
||
* @return string
|
||
*/
|
||
protected function parseUnion($union)
|
||
{
|
||
if (empty($union)) {
|
||
return '';
|
||
}
|
||
|
||
if (isset($union['_all'])) {
|
||
$str = 'UNION ALL ';
|
||
unset($union['_all']);
|
||
} else {
|
||
$str = 'UNION ';
|
||
}
|
||
foreach ($union as $u) {
|
||
$sql[] = $str . (is_array($u) ? $this->buildSelectSql($u) : $u);
|
||
}
|
||
return implode(' ', $sql);
|
||
}
|
||
|
||
/**
|
||
* index分析，可在操作链中指定需要强制使用的索引
|
||
* @access protected
|
||
* @param mixed $index
|
||
* @return string
|
||
*/
|
||
protected function parseForce($index)
|
||
{
|
||
if (empty($index)) {
|
||
return '';
|
||
}
|
||
|
||
if (is_array($index)) {
|
||
$index = join(",", $index);
|
||
}
|
||
|
||
return sprintf(" FORCE INDEX ( %s ) ", $index);
|
||
}
|
||
|
||
/**
|
||
* ON DUPLICATE KEY UPDATE 分析
|
||
* @access protected
|
||
* @param mixed $duplicate
|
||
* @return string
|
||
*/
|
||
protected function parseDuplicate($duplicate)
|
||
{
|
||
return '';
|
||
}
|
||
|
||
/**
|
||
* 插入记录
|
||
* @access public
|
||
* @param mixed $data 数据
|
||
* @param array $options 参数表达式
|
||
* @param boolean $replace 是否replace
|
||
* @return false | integer
|
||
*/
|
||
public function insert($data, $options = [], $replace = false)
|
||
{
|
||
$values = $fields = [];
|
||
$this->model = $options['model'];
|
||
$this->bind = array_merge($this->bind, !empty($options['bind']) ? $options['bind'] : []);
|
||
|
||
foreach ($data as $key => $val) {
|
||
if (isset($val[0]) && 'exp' == $val[0]) {
|
||
$fields[] = $this->parseKey($key);
|
||
$values[] = $val[1];
|
||
} elseif (is_null($val)) {
|
||
$fields[] = $this->parseKey($key);
|
||
$values[] = 'NULL';
|
||
} elseif (is_scalar($val)) {
|
||
// 过滤非标量数据
|
||
$fields[] = $this->parseKey($key);
|
||
if (0 === strpos($val, ':') && isset($this->bind[substr($val, 1)])) {
|
||
$values[] = $val;
|
||
} else {
|
||
$name = count($this->bind);
|
||
$values[] = ':' . $key . $_SERVER['REQUEST_TIME'] . '_' . $name;
|
||
$this->bindParam($key . $_SERVER['REQUEST_TIME'] . '_' . $name, $val);
|
||
}
|
||
}
|
||
}
|
||
// 兼容数字传入方式
|
||
$replace = (is_numeric($replace) && $replace > 0) ? true : $replace;
|
||
$sql = (true === $replace ? 'REPLACE' : 'INSERT') . ' INTO ' . $this->parseTable($options['table']) . ' (' . implode(',', $fields) . ') VALUES (' . implode(',', $values) . ')' . $this->parseDuplicate($replace);
|
||
$sql .= $this->parseComment(!empty($options['comment']) ? $options['comment'] : '');
|
||
return $this->execute($sql, $this->getBindParams(true), !empty($options['fetch_sql']) ? true : false);
|
||
}
|
||
|
||
/**
|
||
* 批量插入记录
|
||
* @access public
|
||
* @param mixed $dataSet 数据集
|
||
* @param array $options 参数表达式
|
||
* @param boolean $replace 是否replace
|
||
* @return false | integer
|
||
*/
|
||
public function insertAll($dataSet, $options = [], $replace = false)
|
||
{
|
||
$values = [];
|
||
$this->model = $options['model'];
|
||
if (!is_array($dataSet[0])) {
|
||
return false;
|
||
}
|
||
$this->bind = array_merge($this->bind, !empty($options['bind']) ? $options['bind'] : []);
|
||
$fields = array_map([$this, 'parseKey'], array_keys($dataSet[0]));
|
||
foreach ($dataSet as $data) {
|
||
$value = [];
|
||
foreach ($data as $key => $val) {
|
||
if (is_array($val) && 'exp' == $val[0]) {
|
||
$value[] = $val[1];
|
||
} elseif (is_null($val)) {
|
||
$value[] = 'NULL';
|
||
} elseif (is_scalar($val)) {
|
||
if (0 === strpos($val, ':') && isset($this->bind[substr($val, 1)])) {
|
||
$value[] = $val;
|
||
} else {
|
||
$name = count($this->bind);
|
||
$value[] = ':' . $key . $_SERVER['REQUEST_TIME'] . '_' . $name;
|
||
$this->bindParam($key . $_SERVER['REQUEST_TIME'] . '_' . $name, $val);
|
||
}
|
||
}
|
||
}
|
||
$values[] = 'SELECT ' . implode(',', $value);
|
||
}
|
||
$sql = 'INSERT INTO ' . $this->parseTable($options['table']) . ' (' . implode(',', $fields) . ') ' . implode(' UNION ALL ', $values);
|
||
$sql .= $this->parseComment(!empty($options['comment']) ? $options['comment'] : '');
|
||
return $this->execute($sql, $this->getBindParams(true), !empty($options['fetch_sql']) ? true : false);
|
||
}
|
||
|
||
/**
|
||
* 通过Select方式插入记录
|
||
* @access public
|
||
* @param string $fields 要插入的数据表字段名
|
||
* @param string $table 要插入的数据表名
|
||
* @param array $option 查询数据参数
|
||
* @return false | integer
|
||
*/
|
||
public function selectInsert($fields, $table, $options = [])
|
||
{
|
||
$this->model = $options['model'];
|
||
$this->bind = array_merge($this->bind, !empty($options['bind']) ? $options['bind'] : []);
|
||
if (is_string($fields)) {
|
||
$fields = explode(',', $fields);
|
||
}
|
||
|
||
$fields = array_map([$this, 'parseKey'], $fields);
|
||
$sql = 'INSERT INTO ' . $this->parseTable($table) . ' (' . implode(',', $fields) . ') ';
|
||
$sql .= $this->buildSelectSql($options);
|
||
return $this->execute($sql, $this->getBindParams(true), !empty($options['fetch_sql']) ? true : false);
|
||
}
|
||
|
||
/**
|
||
* 更新记录
|
||
* @access public
|
||
* @param mixed $data 数据
|
||
* @param array $options 表达式
|
||
* @return false | integer
|
||
*/
|
||
public function update($data, $options)
|
||
{
|
||
$this->model = $options['model'];
|
||
$this->bind = array_merge($this->bind, !empty($options['bind']) ? $options['bind'] : []);
|
||
$table = $this->parseTable($options['table']);
|
||
$sql = 'UPDATE ' . $table . $this->parseSet($data);
|
||
if (strpos($table, ',')) {
|
||
// 多表更新支持JOIN操作
|
||
$sql .= $this->parseJoin(!empty($options['join']) ? $options['join'] : '');
|
||
}
|
||
$sql .= $this->parseWhere(!empty($options['where']) ? $options['where'] : '');
|
||
if (!strpos($table, ',')) {
|
||
// 单表更新支持order和lmit
|
||
$sql .= $this->parseOrder(!empty($options['order']) ? $options['order'] : '')
|
||
. $this->parseLimit(!empty($options['limit']) ? $options['limit'] : '');
|
||
}
|
||
$sql .= $this->parseComment(!empty($options['comment']) ? $options['comment'] : '');
|
||
return $this->execute($sql, $this->getBindParams(true), !empty($options['fetch_sql']) ? true : false);
|
||
}
|
||
|
||
/**
|
||
* 删除记录
|
||
* @access public
|
||
* @param array $options 表达式
|
||
* @return false | integer
|
||
*/
|
||
public function delete($options = [])
|
||
{
|
||
$this->model = $options['model'];
|
||
$this->bind = array_merge($this->bind, !empty($options['bind']) ? $options['bind'] : []);
|
||
$table = $this->parseTable($options['table']);
|
||
$sql = 'DELETE FROM ' . $table;
|
||
if (strpos($table, ',')) {
|
||
// 多表删除支持USING和JOIN操作
|
||
if (!empty($options['using'])) {
|
||
$sql .= ' USING ' . $this->parseTable($options['using']) . ' ';
|
||
}
|
||
$sql .= $this->parseJoin(!empty($options['join']) ? $options['join'] : '');
|
||
}
|
||
$sql .= $this->parseWhere(!empty($options['where']) ? $options['where'] : '');
|
||
if (!strpos($table, ',')) {
|
||
// 单表删除支持order和limit
|
||
$sql .= $this->parseOrder(!empty($options['order']) ? $options['order'] : '')
|
||
. $this->parseLimit(!empty($options['limit']) ? $options['limit'] : '');
|
||
}
|
||
$sql .= $this->parseComment(!empty($options['comment']) ? $options['comment'] : '');
|
||
return $this->execute($sql, $this->getBindParams(true), !empty($options['fetch_sql']) ? true : false);
|
||
}
|
||
|
||
/**
|
||
* 查找记录
|
||
* @access public
|
||
* @param array $options 表达式
|
||
* @return mixed
|
||
*/
|
||
public function select($options = [])
|
||
{
|
||
$this->model = $options['model'];
|
||
$this->bind = array_merge($this->bind, !empty($options['bind']) ? $options['bind'] : []);
|
||
$sql = $this->buildSelectSql($options);
|
||
$result = $this->query($sql, $this->getBindParams(true), !empty($options['fetch_sql']) ? true : false, !empty($options['master']) ? true : false);
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 生成查询SQL
|
||
* @access public
|
||
* @param array $options 表达式
|
||
* @return string
|
||
*/
|
||
public function buildSelectSql($options = [])
|
||
{
|
||
if (isset($options['page'])) {
|
||
// 根据页数计算limit
|
||
list($page, $listRows) = $options['page'];
|
||
$page = $page > 0 ? $page : 1;
|
||
$listRows = $listRows > 0 ? $listRows : (is_numeric($options['limit']) ? $options['limit'] : 20);
|
||
$offset = $listRows * ($page - 1);
|
||
$options['limit'] = $offset . ',' . $listRows;
|
||
}
|
||
$sql = $this->parseSql($this->selectSql, $options);
|
||
return $sql;
|
||
}
|
||
|
||
/**
|
||
* 替换SQL语句中表达式
|
||
* @access public
|
||
* @param array $options 表达式
|
||
* @return string
|
||
*/
|
||
public function parseSql($sql, $options = [])
|
||
{
|
||
$sql = str_replace(
|
||
['%TABLE%', '%DISTINCT%', '%FIELD%', '%JOIN%', '%WHERE%', '%GROUP%', '%HAVING%', '%ORDER%', '%LIMIT%', '%UNION%', '%LOCK%', '%COMMENT%', '%FORCE%'],
|
||
[
|
||
$this->parseTable($options['table']),
|
||
$this->parseDistinct(isset($options['distinct']) ? $options['distinct'] : false),
|
||
$this->parseField(!empty($options['field']) ? $options['field'] : '*'),
|
||
$this->parseJoin(!empty($options['join']) ? $options['join'] : ''),
|
||
$this->parseWhere(!empty($options['where']) ? $options['where'] : ''),
|
||
$this->parseGroup(!empty($options['group']) ? $options['group'] : ''),
|
||
$this->parseHaving(!empty($options['having']) ? $options['having'] : ''),
|
||
$this->parseOrder(!empty($options['order']) ? $options['order'] : ''),
|
||
$this->parseLimit(!empty($options['limit']) ? $options['limit'] : ''),
|
||
$this->parseUnion(!empty($options['union']) ? $options['union'] : ''),
|
||
$this->parseLock(isset($options['lock']) ? $options['lock'] : false),
|
||
$this->parseComment(!empty($options['comment']) ? $options['comment'] : ''),
|
||
$this->parseForce(!empty($options['force']) ? $options['force'] : ''),
|
||
], $sql);
|
||
return $sql;
|
||
}
|
||
|
||
/**
|
||
* 获取最近一次查询的sql语句
|
||
* @param string $model 模型名
|
||
* @access public
|
||
* @return string
|
||
*/
|
||
public function getLastSql($model = '')
|
||
{
|
||
return $model ? $this->modelSql[$model] : $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;
|
||
}
|
||
|
||
/**
|
||
* 设置当前操作模型
|
||
* @access public
|
||
* @param string $model 模型名
|
||
* @return void
|
||
*/
|
||
public function setModel($model)
|
||
{
|
||
$this->model = $model;
|
||
}
|
||
|
||
/**
|
||
* 数据库调试 记录当前SQL
|
||
* @access protected
|
||
* @param boolean $start 调试开始标记 true 开始 false 结束
|
||
*/
|
||
protected function debug($start)
|
||
{
|
||
if (!empty($this->config['debug'])) {
|
||
// 开启数据库调试模式
|
||
if ($start) {
|
||
Debug::remark('queryStartTime', 'time');
|
||
} else {
|
||
$this->modelSql[$this->model] = $this->queryStr;
|
||
// 记录操作结束时间
|
||
Debug::remark('queryEndTime', 'time');
|
||
Log::record($this->queryStr . ' [ RunTime:' . Debug::getRangeTime('queryStartTime', 'queryEndTime') . 's ]', 'sql');
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 初始化数据库连接
|
||
* @access protected
|
||
* @param boolean $master 主服务器
|
||
* @return void
|
||
*/
|
||
protected function initConnect($master = true)
|
||
{
|
||
if (!empty($this->config['deploy'])) {
|
||
// 采用分布式数据库
|
||
$this->_linkID = $this->multiConnect($master);
|
||
} elseif (!$this->_linkID) {
|
||
// 默认单数据库
|
||
$this->_linkID = $this->connect();
|
||
}
|
||
|
||
}
|
||
|
||
/**
|
||
* 连接分布式服务器
|
||
* @access protected
|
||
* @param boolean $master 主服务器
|
||
* @return void
|
||
*/
|
||
protected function multiConnect($master = false)
|
||
{
|
||
// 分布式数据库配置解析
|
||
$_config['username'] = explode(',', $this->config['username']);
|
||
$_config['password'] = explode(',', $this->config['password']);
|
||
$_config['hostname'] = explode(',', $this->config['hostname']);
|
||
$_config['hostport'] = explode(',', $this->config['hostport']);
|
||
$_config['database'] = explode(',', $this->config['database']);
|
||
$_config['dsn'] = explode(',', $this->config['dsn']);
|
||
$_config['charset'] = explode(',', $this->config['charset']);
|
||
|
||
$m = floor(mt_rand(0, $this->config['master_num'] - 1));
|
||
// 数据库读写是否分离
|
||
if ($this->config['rw_separate']) {
|
||
// 主从式采用读写分离
|
||
if ($master)
|
||
// 主服务器写入
|
||
{
|
||
$r = $m;
|
||
} else {
|
||
if (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)); // 每次随机连接的数据库
|
||
}
|
||
|
||
if ($m != $r) {
|
||
$db_master = [
|
||
'username' => isset($_config['username'][$m]) ? $_config['username'][$m] : $_config['username'][0],
|
||
'password' => isset($_config['password'][$m]) ? $_config['password'][$m] : $_config['password'][0],
|
||
'hostname' => isset($_config['hostname'][$m]) ? $_config['hostname'][$m] : $_config['hostname'][0],
|
||
'hostport' => isset($_config['hostport'][$m]) ? $_config['hostport'][$m] : $_config['hostport'][0],
|
||
'database' => isset($_config['database'][$m]) ? $_config['database'][$m] : $_config['database'][0],
|
||
'dsn' => isset($_config['dsn'][$m]) ? $_config['dsn'][$m] : $_config['dsn'][0],
|
||
'charset' => isset($_config['charset'][$m]) ? $_config['charset'][$m] : $_config['charset'][0],
|
||
];
|
||
}
|
||
$db_config = [
|
||
'username' => isset($_config['username'][$r]) ? $_config['username'][$r] : $_config['username'][0],
|
||
'password' => isset($_config['password'][$r]) ? $_config['password'][$r] : $_config['password'][0],
|
||
'hostname' => isset($_config['hostname'][$r]) ? $_config['hostname'][$r] : $_config['hostname'][0],
|
||
'hostport' => isset($_config['hostport'][$r]) ? $_config['hostport'][$r] : $_config['hostport'][0],
|
||
'database' => isset($_config['database'][$r]) ? $_config['database'][$r] : $_config['database'][0],
|
||
'dsn' => isset($_config['dsn'][$r]) ? $_config['dsn'][$r] : $_config['dsn'][0],
|
||
'charset' => isset($_config['charset'][$r]) ? $_config['charset'][$r] : $_config['charset'][0],
|
||
];
|
||
return $this->connect($db_config, $r, $r == $m ? false : $db_master);
|
||
}
|
||
|
||
/**
|
||
* 析构方法
|
||
* @access public
|
||
*/
|
||
public function __destruct()
|
||
{
|
||
// 释放查询
|
||
if ($this->PDOStatement) {
|
||
$this->free();
|
||
}
|
||
// 关闭连接
|
||
$this->close();
|
||
}
|
||
}
|