1165 lines
36 KiB
PHP
Executable File
1165 lines
36 KiB
PHP
Executable File
<?php
|
||
// +----------------------------------------------------------------------
|
||
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
||
// +----------------------------------------------------------------------
|
||
// | Copyright (c) 2006~2018 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\Exception;
|
||
|
||
abstract class Builder
|
||
{
|
||
// connection对象实例
|
||
protected $connection;
|
||
|
||
// 查询表达式映射
|
||
protected $exp = ['EQ' => '=', 'NEQ' => '<>', 'GT' => '>', 'EGT' => '>=', 'LT' => '<', 'ELT' => '<=', 'NOTLIKE' => 'NOT LIKE', 'NOTIN' => 'NOT IN', 'NOTBETWEEN' => 'NOT BETWEEN', 'NOTEXISTS' => 'NOT EXISTS', 'NOTNULL' => 'NOT NULL', 'NOTBETWEEN TIME' => 'NOT BETWEEN TIME'];
|
||
|
||
// 查询表达式解析
|
||
protected $parser = [
|
||
'parseCompare' => ['=', '<>', '>', '>=', '<', '<='],
|
||
'parseLike' => ['LIKE', 'NOT LIKE'],
|
||
'parseBetween' => ['NOT BETWEEN', 'BETWEEN'],
|
||
'parseIn' => ['NOT IN', 'IN'],
|
||
'parseExp' => ['EXP'],
|
||
'parseNull' => ['NOT NULL', 'NULL'],
|
||
'parseBetweenTime' => ['BETWEEN TIME', 'NOT BETWEEN TIME'],
|
||
'parseTime' => ['< TIME', '> TIME', '<= TIME', '>= TIME'],
|
||
'parseExists' => ['NOT EXISTS', 'EXISTS'],
|
||
'parseColumn' => ['COLUMN'],
|
||
];
|
||
|
||
// SQL表达式
|
||
protected $selectSql = 'SELECT%DISTINCT% %FIELD% FROM %TABLE%%FORCE%%JOIN%%WHERE%%GROUP%%HAVING%%UNION%%ORDER%%LIMIT% %LOCK%%COMMENT%';
|
||
|
||
protected $insertSql = '%INSERT% INTO %TABLE% (%FIELD%) VALUES (%DATA%) %COMMENT%';
|
||
|
||
protected $insertAllSql = '%INSERT% INTO %TABLE% (%FIELD%) %DATA% %COMMENT%';
|
||
|
||
protected $updateSql = 'UPDATE %TABLE% SET %SET%%JOIN%%WHERE%%ORDER%%LIMIT% %LOCK%%COMMENT%';
|
||
|
||
protected $deleteSql = 'DELETE FROM %TABLE%%USING%%JOIN%%WHERE%%ORDER%%LIMIT% %LOCK%%COMMENT%';
|
||
|
||
/**
|
||
* 架构函数
|
||
* @access public
|
||
* @param Connection $connection 数据库连接对象实例
|
||
*/
|
||
public function __construct(Connection $connection)
|
||
{
|
||
$this->connection = $connection;
|
||
}
|
||
|
||
/**
|
||
* 获取当前的连接对象实例
|
||
* @access public
|
||
* @return Connection
|
||
*/
|
||
public function getConnection()
|
||
{
|
||
return $this->connection;
|
||
}
|
||
|
||
/**
|
||
* 注册查询表达式解析
|
||
* @access public
|
||
* @param string $name 解析方法
|
||
* @param array $parser 匹配表达式数据
|
||
* @return $this
|
||
*/
|
||
public function bindParser($name, $parser)
|
||
{
|
||
$this->parser[$name] = $parser;
|
||
return $this;
|
||
}
|
||
|
||
/**
|
||
* 数据分析
|
||
* @access protected
|
||
* @param Query $query 查询对象
|
||
* @param array $data 数据
|
||
* @param array $fields 字段信息
|
||
* @param array $bind 参数绑定
|
||
* @return array
|
||
*/
|
||
protected function parseData(Query $query, $data = [], $fields = [], $bind = [])
|
||
{
|
||
if (empty($data)) {
|
||
return [];
|
||
}
|
||
|
||
$options = $query->getOptions();
|
||
|
||
// 获取绑定信息
|
||
if (empty($bind)) {
|
||
$bind = $this->connection->getFieldsBind($options['table']);
|
||
}
|
||
|
||
if (empty($fields)) {
|
||
if ('*' == $options['field']) {
|
||
$fields = array_keys($bind);
|
||
} else {
|
||
$fields = $options['field'];
|
||
}
|
||
}
|
||
|
||
$result = [];
|
||
|
||
foreach ($data as $key => $val) {
|
||
if ('*' != $options['field'] && !in_array($key, $fields, true)) {
|
||
continue;
|
||
}
|
||
|
||
$item = $this->parseKey($query, $key, true);
|
||
|
||
if ($val instanceof Expression) {
|
||
$result[$item] = $val->getValue();
|
||
continue;
|
||
} elseif (!is_scalar($val) && (in_array($key, (array) $query->getOptions('json')) || 'json' == $this->connection->getFieldsType($options['table'], $key))) {
|
||
$val = json_encode($val, JSON_UNESCAPED_UNICODE);
|
||
} elseif (is_object($val) && method_exists($val, '__toString')) {
|
||
// 对象数据写入
|
||
$val = $val->__toString();
|
||
}
|
||
|
||
if (false !== strpos($key, '->')) {
|
||
list($key, $name) = explode('->', $key);
|
||
$item = $this->parseKey($query, $key);
|
||
$result[$item] = 'json_set(' . $item . ', \'$.' . $name . '\', ' . $this->parseDataBind($query, $key, $val, $bind) . ')';
|
||
} elseif ('*' == $options['field'] && false === strpos($key, '.') && !in_array($key, $fields, true)) {
|
||
if ($options['strict']) {
|
||
throw new Exception('fields not exists:[' . $key . ']');
|
||
}
|
||
} elseif (is_null($val)) {
|
||
$result[$item] = 'NULL';
|
||
} elseif (is_array($val) && !empty($val)) {
|
||
switch (strtoupper($val[0])) {
|
||
case 'INC':
|
||
$result[$item] = $item . ' + ' . floatval($val[1]);
|
||
break;
|
||
case 'DEC':
|
||
$result[$item] = $item . ' - ' . floatval($val[1]);
|
||
break;
|
||
case 'EXP':
|
||
throw new Exception('not support data:[' . $val[0] . ']');
|
||
}
|
||
} elseif (is_scalar($val)) {
|
||
// 过滤非标量数据
|
||
$result[$item] = $this->parseDataBind($query, $key, $val, $bind);
|
||
}
|
||
}
|
||
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 数据绑定处理
|
||
* @access protected
|
||
* @param Query $query 查询对象
|
||
* @param string $key 字段名
|
||
* @param mixed $data 数据
|
||
* @param array $bind 绑定数据
|
||
* @return string
|
||
*/
|
||
protected function parseDataBind(Query $query, $key, $data, $bind = [])
|
||
{
|
||
if ($data instanceof Expression) {
|
||
return $data->getValue();
|
||
}
|
||
|
||
$name = $query->bind($data, isset($bind[$key]) ? $bind[$key] : PDO::PARAM_STR);
|
||
|
||
return ':' . $name;
|
||
}
|
||
|
||
/**
|
||
* 字段名分析
|
||
* @access public
|
||
* @param Query $query 查询对象
|
||
* @param mixed $key 字段名
|
||
* @param bool $strict 严格检测
|
||
* @return string
|
||
*/
|
||
public function parseKey(Query $query, $key, $strict = false)
|
||
{
|
||
return $key instanceof Expression ? $key->getValue() : $key;
|
||
}
|
||
|
||
/**
|
||
* field分析
|
||
* @access protected
|
||
* @param Query $query 查询对象
|
||
* @param mixed $fields 字段名
|
||
* @return string
|
||
*/
|
||
protected function parseField(Query $query, $fields)
|
||
{
|
||
if ('*' == $fields || empty($fields)) {
|
||
$fieldsStr = '*';
|
||
} elseif (is_array($fields)) {
|
||
// 支持 'field1'=>'field2' 这样的字段别名定义
|
||
$array = [];
|
||
|
||
foreach ($fields as $key => $field) {
|
||
if (!is_numeric($key)) {
|
||
$array[] = $this->parseKey($query, $key) . ' AS ' . $this->parseKey($query, $field, true);
|
||
} else {
|
||
$array[] = $this->parseKey($query, $field);
|
||
}
|
||
}
|
||
|
||
$fieldsStr = implode(',', $array);
|
||
}
|
||
|
||
return $fieldsStr;
|
||
}
|
||
|
||
/**
|
||
* table分析
|
||
* @access protected
|
||
* @param Query $query 查询对象
|
||
* @param mixed $tables 表名
|
||
* @return string
|
||
*/
|
||
protected function parseTable(Query $query, $tables)
|
||
{
|
||
$item = [];
|
||
$options = $query->getOptions();
|
||
|
||
foreach ((array) $tables as $key => $table) {
|
||
if (!is_numeric($key)) {
|
||
$key = $this->connection->parseSqlTable($key);
|
||
$item[] = $this->parseKey($query, $key) . ' ' . $this->parseKey($query, $table</ |