|
|
@@ -2,7 +2,7 @@
|
|
|
// +----------------------------------------------------------------------
|
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
|
|
// +----------------------------------------------------------------------
|
|
|
-// | Copyright (c) 2006~2017 http://thinkphp.cn All rights reserved.
|
|
|
+// | Copyright (c) 2006~2018 http://thinkphp.cn All rights reserved.
|
|
|
// +----------------------------------------------------------------------
|
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
|
// +----------------------------------------------------------------------
|
|
|
@@ -53,14 +53,16 @@ class Query
|
|
|
protected static $info = [];
|
|
|
// 回调事件
|
|
|
private static $event = [];
|
|
|
+ // 读取主库
|
|
|
+ private static $readMaster = [];
|
|
|
|
|
|
/**
|
|
|
* 构造函数
|
|
|
* @access public
|
|
|
* @param Connection $connection 数据库对象实例
|
|
|
- * @param string $model 模型名
|
|
|
+ * @param Model $model 模型对象
|
|
|
*/
|
|
|
- public function __construct(Connection $connection = null, $model = '')
|
|
|
+ public function __construct(Connection $connection = null, $model = null)
|
|
|
{
|
|
|
$this->connection = $connection ?: Db::connect([], true);
|
|
|
$this->prefix = $this->connection->getConfig('prefix');
|
|
|
@@ -131,15 +133,34 @@ class Query
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
- * 获取当前的模型对象名
|
|
|
+ * 获取当前的模型对象实例
|
|
|
* @access public
|
|
|
- * @return string
|
|
|
+ * @return Model|null
|
|
|
*/
|
|
|
public function getModel()
|
|
|
{
|
|
|
return $this->model;
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 设置后续从主库读取数据
|
|
|
+ * @access public
|
|
|
+ * @param bool $allTable
|
|
|
+ * @return void
|
|
|
+ */
|
|
|
+ public function readMaster($allTable = false)
|
|
|
+ {
|
|
|
+ if ($allTable) {
|
|
|
+ $table = '*';
|
|
|
+ } else {
|
|
|
+ $table = isset($this->options['table']) ? $this->options['table'] : $this->getTable();
|
|
|
+ }
|
|
|
+
|
|
|
+ static::$readMaster[$table] = true;
|
|
|
+
|
|
|
+ return $this;
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* 获取当前的builder实例对象
|
|
|
* @access public
|
|
|
@@ -238,7 +259,7 @@ class Query
|
|
|
*/
|
|
|
public function execute($sql, $bind = [])
|
|
|
{
|
|
|
- return $this->connection->execute($sql, $bind);
|
|
|
+ return $this->connection->execute($sql, $bind, $this);
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
@@ -312,9 +333,9 @@ class Query
|
|
|
* @param array $sql SQL批处理指令
|
|
|
* @return boolean
|
|
|
*/
|
|
|
- public function batchQuery($sql = [])
|
|
|
+ public function batchQuery($sql = [], $bind = [])
|
|
|
{
|
|
|
- return $this->connection->batchQuery($sql);
|
|
|
+ return $this->connection->batchQuery($sql, $bind);
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
@@ -403,7 +424,7 @@ class Query
|
|
|
if (empty($this->options['table'])) {
|
|
|
$this->options['table'] = $this->getTable();
|
|
|
}
|
|
|
- $key = is_string($cache['key']) ? $cache['key'] : md5($field . serialize($this->options) . serialize($this->bind));
|
|
|
+ $key = is_string($cache['key']) ? $cache['key'] : md5($this->connection->getConfig('database') . '.' . $field . serialize($this->options) . serialize($this->bind));
|
|
|
$result = Cache::get($key);
|
|
|
}
|
|
|
if (false === $result) {
|
|
|
@@ -420,7 +441,7 @@ class Query
|
|
|
$result += 0;
|
|
|
}
|
|
|
|
|
|
- if (isset($cache)) {
|
|
|
+ if (isset($cache) && false !== $result) {
|
|
|
// 缓存数据
|
|
|
$this->cacheData($key, $result, $cache);
|
|
|
}
|
|
|
@@ -447,7 +468,7 @@ class Query
|
|
|
if (empty($this->options['table'])) {
|
|
|
$this->options['table'] = $this->getTable();
|
|
|
}
|
|
|
- $guid = is_string($cache['key']) ? $cache['key'] : md5($field . serialize($this->options) . serialize($this->bind));
|
|
|
+ $guid = is_string($cache['key']) ? $cache['key'] : md5($this->connection->getConfig('database') . '.' . $field . serialize($this->options) . serialize($this->bind));
|
|
|
$result = Cache::get($guid);
|
|
|
}
|
|
|
if (false === $result) {
|
|
|
@@ -534,22 +555,24 @@ class Query
|
|
|
* MIN查询
|
|
|
* @access public
|
|
|
* @param string $field 字段名
|
|
|
+ * @param bool $force 强制转为数字类型
|
|
|
* @return mixed
|
|
|
*/
|
|
|
- public function min($field)
|
|
|
+ public function min($field, $force = true)
|
|
|
{
|
|
|
- return $this->value('MIN(' . $field . ') AS tp_min', 0, true);
|
|
|
+ return $this->value('MIN(' . $field . ') AS tp_min', 0, $force);
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* MAX查询
|
|
|
* @access public
|
|
|
* @param string $field 字段名
|
|
|
+ * @param bool $force 强制转为数字类型
|
|
|
* @return mixed
|
|
|
*/
|
|
|
- public function max($field)
|
|
|
+ public function max($field, $force = true)
|
|
|
{
|
|
|
- return $this->value('MAX(' . $field . ') AS tp_max', 0, true);
|
|
|
+ return $this->value('MAX(' . $field . ') AS tp_max', 0, $force);
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
@@ -607,7 +630,7 @@ class Query
|
|
|
return true;
|
|
|
}
|
|
|
}
|
|
|
- return $this->setField($field, ['exp', $field . '+' . $step]);
|
|
|
+ return $this->setField($field, ['inc', $step]);
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
@@ -635,8 +658,9 @@ class Query
|
|
|
$this->options = [];
|
|
|
return true;
|
|
|
}
|
|
|
+ return $this->setField($field, ['inc', $step]);
|
|
|
}
|
|
|
- return $this->setField($field, ['exp', $field . '-' . $step]);
|
|
|
+ return $this->setField($field, ['dec', $step]);
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
@@ -704,7 +728,8 @@ class Query
|
|
|
{
|
|
|
// 传入的表名为数组
|
|
|
if (is_array($join)) {
|
|
|
- list($table, $alias) = each($join);
|
|
|
+ $table = $join;
|
|
|
+ $alias = array_shift($join);
|
|
|
} else {
|
|
|
$join = trim($join);
|
|
|
if (false !== strpos($join, '(')) {
|
|
|
@@ -725,13 +750,9 @@ class Query
|
|
|
$table = $this->getTable($table);
|
|
|
}
|
|
|
}
|
|
|
- }
|
|
|
- if (isset($alias)) {
|
|
|
- if (isset($this->options['alias'][$table])) {
|
|
|
- $table = $table . '@think' . uniqid();
|
|
|
+ if (isset($alias) && $table != $alias) {
|
|
|
+ $table = [$table => $alias];
|
|
|
}
|
|
|
- $table = [$table => $alias];
|
|
|
- $this->alias($table);
|
|
|
}
|
|
|
return $table;
|
|
|
}
|
|
|
@@ -769,8 +790,15 @@ class Query
|
|
|
{
|
|
|
if (empty($field)) {
|
|
|
return $this;
|
|
|
+ } elseif ($field instanceof Expression) {
|
|
|
+ $this->options['field'][] = $field;
|
|
|
+ return $this;
|
|
|
}
|
|
|
+
|
|
|
if (is_string($field)) {
|
|
|
+ if (preg_match('/[\<\'\"\(]/', $field)) {
|
|
|
+ return $this->fieldRaw($field);
|
|
|
+ }
|
|
|
$field = array_map('trim', explode(',', $field));
|
|
|
}
|
|
|
if (true === $field) {
|
|
|
@@ -794,12 +822,30 @@ class Query
|
|
|
}
|
|
|
|
|
|
if (isset($this->options['field'])) {
|
|
|
- $field = array_merge($this->options['field'], $field);
|
|
|
+ $field = array_merge((array) $this->options['field'], $field);
|
|
|
}
|
|
|
$this->options['field'] = array_unique($field);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 表达式方式指定查询字段
|
|
|
+ * @access public
|
|
|
+ * @param string $field 字段名
|
|
|
+ * @param array $bind 参数绑定
|
|
|
+ * @return $this
|
|
|
+ */
|
|
|
+ public function fieldRaw($field, array $bind = [])
|
|
|
+ {
|
|
|
+ $this->options['field'][] = $this->raw($field);
|
|
|
+
|
|
|
+ if ($bind) {
|
|
|
+ $this->bind($bind);
|
|
|
+ }
|
|
|
+
|
|
|
+ return $this;
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* 设置数据
|
|
|
* @access public
|
|
|
@@ -828,7 +874,7 @@ class Query
|
|
|
{
|
|
|
$fields = is_string($field) ? explode(',', $field) : $field;
|
|
|
foreach ($fields as $field) {
|
|
|
- $this->data($field, ['exp', $field . '+' . $step]);
|
|
|
+ $this->data($field, ['inc', $step]);
|
|
|
}
|
|
|
return $this;
|
|
|
}
|
|
|
@@ -844,7 +890,7 @@ class Query
|
|
|
{
|
|
|
$fields = is_string($field) ? explode(',', $field) : $field;
|
|
|
foreach ($fields as $field) {
|
|
|
- $this->data($field, ['exp', $field . '-' . $step]);
|
|
|
+ $this->data($field, ['dec', $step]);
|
|
|
}
|
|
|
return $this;
|
|
|
}
|
|
|
@@ -858,25 +904,36 @@ class Query
|
|
|
*/
|
|
|
public function exp($field, $value)
|
|
|
{
|
|
|
- $this->data($field, ['exp', $value]);
|
|
|
+ $this->data($field, $this->raw($value));
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 使用表达式设置数据
|
|
|
+ * @access public
|
|
|
+ * @param mixed $value 表达式
|
|
|
+ * @return Expression
|
|
|
+ */
|
|
|
+ public function raw($value)
|
|
|
+ {
|
|
|
+ return new Expression($value);
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* 指定JOIN查询字段
|
|
|
* @access public
|
|
|
* @param string|array $table 数据表
|
|
|
* @param string|array $field 查询字段
|
|
|
- * @param string|array $on JOIN条件
|
|
|
+ * @param mixed $on JOIN条件
|
|
|
* @param string $type JOIN类型
|
|
|
* @return $this
|
|
|
*/
|
|
|
public function view($join, $field = true, $on = null, $type = 'INNER')
|
|
|
{
|
|
|
$this->options['view'] = true;
|
|
|
- if (is_array($join) && key($join) !== 0) {
|
|
|
+ if (is_array($join) && key($join) === 0) {
|
|
|
foreach ($join as $key => $val) {
|
|
|
- $this->view($key, $val[0], isset($val[1]) ? $val[1] : null, isset($val[2]) ? $val[2] : 'INNER');
|
|
|
+ $this->view($val[0], $val[1], isset($val[2]) ? $val[2] : null, isset($val[3]) ? $val[3] : 'INNER');
|
|
|
}
|
|
|
} else {
|
|
|
$fields = [];
|
|
|
@@ -975,6 +1032,37 @@ class Query
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 指定表达式查询条件
|
|
|
+ * @access public
|
|
|
+ * @param string $where 查询条件
|
|
|
+ * @param array $bind 参数绑定
|
|
|
+ * @param string $logic 查询逻辑 and or xor
|
|
|
+ * @return $this
|
|
|
+ */
|
|
|
+ public function whereRaw($where, $bind = [], $logic = 'AND')
|
|
|
+ {
|
|
|
+ $this->options['where'][$logic][] = $this->raw($where);
|
|
|
+
|
|
|
+ if ($bind) {
|
|
|
+ $this->bind($bind);
|
|
|
+ }
|
|
|
+
|
|
|
+ return $this;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 指定表达式查询条件 OR
|
|
|
+ * @access public
|
|
|
+ * @param string $where 查询条件
|
|
|
+ * @param array $bind 参数绑定
|
|
|
+ * @return $this
|
|
|
+ */
|
|
|
+ public function whereOrRaw($where, $bind = [])
|
|
|
+ {
|
|
|
+ return $this->whereRaw($where, $bind, 'OR');
|
|
|
+ }
|
|
|
+
|
|
|
/**
|
|
|
* 指定Null查询条件
|
|
|
* @access public
|
|
|
@@ -984,7 +1072,7 @@ class Query
|
|
|
*/
|
|
|
public function whereNull($field, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'null', null);
|
|
|
+ $this->parseWhereExp($logic, $field, 'null', null, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -997,7 +1085,7 @@ class Query
|
|
|
*/
|
|
|
public function whereNotNull($field, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'notnull', null);
|
|
|
+ $this->parseWhereExp($logic, $field, 'notnull', null, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1037,7 +1125,7 @@ class Query
|
|
|
*/
|
|
|
public function whereIn($field, $condition, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'in', $condition);
|
|
|
+ $this->parseWhereExp($logic, $field, 'in', $condition, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1051,7 +1139,7 @@ class Query
|
|
|
*/
|
|
|
public function whereNotIn($field, $condition, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'not in', $condition);
|
|
|
+ $this->parseWhereExp($logic, $field, 'not in', $condition, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1065,7 +1153,7 @@ class Query
|
|
|
*/
|
|
|
public function whereLike($field, $condition, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'like', $condition);
|
|
|
+ $this->parseWhereExp($logic, $field, 'like', $condition, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1079,7 +1167,7 @@ class Query
|
|
|
*/
|
|
|
public function whereNotLike($field, $condition, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'not like', $condition);
|
|
|
+ $this->parseWhereExp($logic, $field, 'not like', $condition, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1093,7 +1181,7 @@ class Query
|
|
|
*/
|
|
|
public function whereBetween($field, $condition, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'between', $condition);
|
|
|
+ $this->parseWhereExp($logic, $field, 'between', $condition, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1107,7 +1195,7 @@ class Query
|
|
|
*/
|
|
|
public function whereNotBetween($field, $condition, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'not between', $condition);
|
|
|
+ $this->parseWhereExp($logic, $field, 'not between', $condition, [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1121,7 +1209,7 @@ class Query
|
|
|
*/
|
|
|
public function whereExp($field, $condition, $logic = 'AND')
|
|
|
{
|
|
|
- $this->parseWhereExp($logic, $field, 'exp', $condition);
|
|
|
+ $this->parseWhereExp($logic, $field, 'exp', $this->raw($condition), [], true);
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1148,9 +1236,10 @@ class Query
|
|
|
* @param mixed $op 查询表达式
|
|
|
* @param mixed $condition 查询条件
|
|
|
* @param array $param 查询参数
|
|
|
+ * @param bool $strict 严格模式
|
|
|
* @return void
|
|
|
*/
|
|
|
- protected function parseWhereExp($logic, $field, $op, $condition, $param = [])
|
|
|
+ protected function parseWhereExp($logic, $field, $op, $condition, $param = [], $strict = false)
|
|
|
{
|
|
|
$logic = strtoupper($logic);
|
|
|
if ($field instanceof \Closure) {
|
|
|
@@ -1161,8 +1250,17 @@ class Query
|
|
|
if (is_string($field) && !empty($this->options['via']) && !strpos($field, '.')) {
|
|
|
$field = $this->options['via'] . '.' . $field;
|
|
|
}
|
|
|
- if (is_string($field) && preg_match('/[,=\>\<\'\"\(\s]/', $field)) {
|
|
|
- $where[] = ['exp', $field];
|
|
|
+
|
|
|
+ if ($field instanceof Expression) {
|
|
|
+ return $this->whereRaw($field, is_array($op) ? $op : []);
|
|
|
+ } elseif ($strict) {
|
|
|
+ // 使用严格模式查询
|
|
|
+ $where[$field] = [$op, $condition];
|
|
|
+
|
|
|
+ // 记录一个字段多次查询条件
|
|
|
+ $this->options['multi'][$logic][$field][] = $where[$field];
|
|
|
+ } elseif (is_string($field) && preg_match('/[,=\>\<\'\"\(\s]/', $field)) {
|
|
|
+ $where[] = ['exp', $this->raw($field)];
|
|
|
if (is_array($op)) {
|
|
|
// 参数绑定
|
|
|
$this->bind($op);
|
|
|
@@ -1183,21 +1281,28 @@ class Query
|
|
|
$where[$field] = $param;
|
|
|
} elseif (in_array(strtolower($op), ['null', 'notnull', 'not null'])) {
|
|
|
// null查询
|
|
|
- $where[$field] = [$op, ''];
|
|
|
+ $where[$field] = [$op, ''];
|
|
|
+
|
|
|
$this->options['multi'][$logic][$field][] = $where[$field];
|
|
|
} elseif (is_null($condition)) {
|
|
|
// 字段相等查询
|
|
|
- $where[$field] = ['eq', $op];
|
|
|
+ $where[$field] = ['eq', $op];
|
|
|
+
|
|
|
$this->options['multi'][$logic][$field][] = $where[$field];
|
|
|
} else {
|
|
|
- $where[$field] = [$op, $condition, isset($param[2]) ? $param[2] : null];
|
|
|
- if ('exp' == strtolower($op) && isset($param[2]) && is_array($param[2])) {
|
|
|
+ if ('exp' == strtolower($op)) {
|
|
|
+ $where[$field] = ['exp', $this->raw($condition)];
|
|
|
// 参数绑定
|
|
|
- $this->bind($param[2]);
|
|
|
+ if (isset($param[2]) && is_array($param[2])) {
|
|
|
+ $this->bind($param[2]);
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ $where[$field] = [$op, $condition];
|
|
|
}
|
|
|
// 记录一个字段多次查询条件
|
|
|
$this->options['multi'][$logic][$field][] = $where[$field];
|
|
|
}
|
|
|
+
|
|
|
if (!empty($where)) {
|
|
|
if (!isset($this->options['where'][$logic])) {
|
|
|
$this->options['where'][$logic] = [];
|
|
|
@@ -1239,6 +1344,7 @@ class Query
|
|
|
$logic = strtoupper($logic);
|
|
|
if (isset($this->options['where'][$logic][$field])) {
|
|
|
unset($this->options['where'][$logic][$field]);
|
|
|
+ unset($this->options['multi'][$logic][$field]);
|
|
|
}
|
|
|
return $this;
|
|
|
}
|
|
|
@@ -1414,31 +1520,59 @@ class Query
|
|
|
*/
|
|
|
public function order($field, $order = null)
|
|
|
{
|
|
|
- if (!empty($field)) {
|
|
|
- if (is_string($field)) {
|
|
|
- if (!empty($this->options['via'])) {
|
|
|
- $field = $this->options['via'] . '.' . $field;
|
|
|
- }
|
|
|
- $field = empty($order) ? $field : [$field => $order];
|
|
|
- } elseif (!empty($this->options['via'])) {
|
|
|
- foreach ($field as $key => $val) {
|
|
|
- if (is_numeric($key)) {
|
|
|
- $field[$key] = $this->options['via'] . '.' . $val;
|
|
|
- } else {
|
|
|
- $field[$this->options['via'] . '.' . $key] = $val;
|
|
|
- unset($field[$key]);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- if (!isset($this->options['order'])) {
|
|
|
- $this->options['order'] = [];
|
|
|
+ if (empty($field)) {
|
|
|
+ return $this;
|
|
|
+ } elseif ($field instanceof Expression) {
|
|
|
+ $this->options['order'][] = $field;
|
|
|
+ return $this;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (is_string($field)) {
|
|
|
+ if (!empty($this->options['via'])) {
|
|
|
+ $field = $this->options['via'] . '.' . $field;
|
|
|
}
|
|
|
- if (is_array($field)) {
|
|
|
- $this->options['order'] = array_merge($this->options['order'], $field);
|
|
|
+ if (strpos($field, ',')) {
|
|
|
+ $field = array_map('trim', explode(',', $field));
|
|
|
} else {
|
|
|
- $this->options['order'][] = $field;
|
|
|
+ $field = empty($order) ? $field : [$field => $order];
|
|
|
+ }
|
|
|
+ } elseif (!empty($this->options['via'])) {
|
|
|
+ foreach ($field as $key => $val) {
|
|
|
+ if (is_numeric($key)) {
|
|
|
+ $field[$key] = $this->options['via'] . '.' . $val;
|
|
|
+ } else {
|
|
|
+ $field[$this->options['via'] . '.' . $key] = $val;
|
|
|
+ unset($field[$key]);
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
+ if (!isset($this->options['order'])) {
|
|
|
+ $this->options['order'] = [];
|
|
|
+ }
|
|
|
+ if (is_array($field)) {
|
|
|
+ $this->options['order'] = array_merge($this->options['order'], $field);
|
|
|
+ } else {
|
|
|
+ $this->options['order'][] = $field;
|
|
|
+ }
|
|
|
+
|
|
|
+ return $this;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 表达式方式指定Field排序
|
|
|
+ * @access public
|
|
|
+ * @param string $field 排序字段
|
|
|
+ * @param array $bind 参数绑定
|
|
|
+ * @return $this
|
|
|
+ */
|
|
|
+ public function orderRaw($field, array $bind = [])
|
|
|
+ {
|
|
|
+ $this->options['order'][] = $this->raw($field);
|
|
|
+
|
|
|
+ if ($bind) {
|
|
|
+ $this->bind($bind);
|
|
|
+ }
|
|
|
+
|
|
|
return $this;
|
|
|
}
|
|
|
|
|
|
@@ -1523,7 +1657,12 @@ class Query
|
|
|
{
|
|
|
if (is_array($alias)) {
|
|
|
foreach ($alias as $key => $val) {
|
|
|
- $this->options['alias'][$key] = $val;
|
|
|
+ if (false !== strpos($key, '__')) {
|
|
|
+ $table = $this->parseSqlTable($key);
|
|
|
+ } else {
|
|
|
+ $table = $key;
|
|
|
+ }
|
|
|
+ $this->options['alias'][$table] = $val;
|
|
|
}
|
|
|
} else {
|
|
|
if (isset($this->options['table'])) {
|
|
|
@@ -1651,46 +1790,49 @@ class Query
|
|
|
* 查询日期或者时间
|
|
|
* @access public
|
|
|
* @param string $field 日期字段名
|
|
|
- * @param string $op 比较运算符或者表达式
|
|
|
+ * @param string|array $op 比较运算符或者表达式
|
|
|
* @param string|array $range 比较范围
|
|
|
* @return $this
|
|
|
*/
|
|
|
public function whereTime($field, $op, $range = null)
|
|
|
{
|
|
|
if (is_null($range)) {
|
|
|
- // 使用日期表达式
|
|
|
- $date = getdate();
|
|
|
- switch (strtolower($op)) {
|
|
|
- case 'today':
|
|
|
- case 'd':
|
|
|
- $range = ['today', 'tomorrow'];
|
|
|
- break;
|
|
|
- case 'week':
|
|
|
- case 'w':
|
|
|
- $range = 'this week 00:00:00';
|
|
|
- break;
|
|
|
- case 'month':
|
|
|
- case 'm':
|
|
|
- $range = mktime(0, 0, 0, $date['mon'], 1, $date['year']);
|
|
|
- break;
|
|
|
- case 'year':
|
|
|
- case 'y':
|
|
|
- $range = mktime(0, 0, 0, 1, 1, $date['year']);
|
|
|
- break;
|
|
|
- case 'yesterday':
|
|
|
- $range = ['yesterday', 'today'];
|
|
|
- break;
|
|
|
- case 'last week':
|
|
|
- $range = ['last week 00:00:00', 'this week 00:00:00'];
|
|
|
- break;
|
|
|
- case 'last month':
|
|
|
- $range = [date('y-m-01', strtotime('-1 month')), mktime(0, 0, 0, $date['mon'], 1, $date['year'])];
|
|
|
- break;
|
|
|
- case 'last year':
|
|
|
- $range = [mktime(0, 0, 0, 1, 1, $date['year'] - 1), mktime(0, 0, 0, 1, 1, $date['year'])];
|
|
|
- break;
|
|
|
- default:
|
|
|
- $range = $op;
|
|
|
+ if (is_array($op)) {
|
|
|
+ $range = $op;
|
|
|
+ } else {
|
|
|
+ // 使用日期表达式
|
|
|
+ switch (strtolower($op)) {
|
|
|
+ case 'today':
|
|
|
+ case 'd':
|
|
|
+ $range = ['today', 'tomorrow'];
|
|
|
+ break;
|
|
|
+ case 'week':
|
|
|
+ case 'w':
|
|
|
+ $range = ['this week 00:00:00', 'next week 00:00:00'];
|
|
|
+ break;
|
|
|
+ case 'month':
|
|
|
+ case 'm':
|
|
|
+ $range = ['first Day of this month 00:00:00', 'first Day of next month 00:00:00'];
|
|
|
+ break;
|
|
|
+ case 'year':
|
|
|
+ case 'y':
|
|
|
+ $range = ['this year 1/1', 'next year 1/1'];
|
|
|
+ break;
|
|
|
+ case 'yesterday':
|
|
|
+ $range = ['yesterday', 'today'];
|
|
|
+ break;
|
|
|
+ case 'last week':
|
|
|
+ $range = ['last week 00:00:00', 'this week 00:00:00'];
|
|
|
+ break;
|
|
|
+ case 'last month':
|
|
|
+ $range = ['first Day of last month 00:00:00', 'first Day of this month 00:00:00'];
|
|
|
+ break;
|
|
|
+ case 'last year':
|
|
|
+ $range = ['last year 1/1', 'this year 1/1'];
|
|
|
+ break;
|
|
|
+ default:
|
|
|
+ $range = $op;
|
|
|
+ }
|
|
|
}
|
|
|
$op = is_array($range) ? 'between' : '>';
|
|
|
}
|
|
|
@@ -1735,7 +1877,7 @@ class Query
|
|
|
$schema = $guid;
|
|
|
}
|
|
|
// 读取缓存
|
|
|
- if (is_file(RUNTIME_PATH . 'schema/' . $schema . '.php')) {
|
|
|
+ if (!App::$debug && is_file(RUNTIME_PATH . 'schema/' . $schema . '.php')) {
|
|
|
$info = include RUNTIME_PATH . 'schema/' . $schema . '.php';
|
|
|
} else {
|
|
|
$info = $this->connection->getFields($guid);
|
|
|
@@ -1810,7 +1952,9 @@ class Query
|
|
|
*/
|
|
|
protected function getFieldBindType($type)
|
|
|
{
|
|
|
- if (preg_match('/(int|double|float|decimal|real|numeric|serial|bit)/is', $type)) {
|
|
|
+ if (0 === strpos($type, 'set') || 0 === strpos($type, 'enum')) {
|
|
|
+ $bind = PDO::PARAM_STR;
|
|
|
+ } elseif (preg_match('/(int|double|float|decimal|real|numeric|serial|bit)/is', $type)) {
|
|
|
$bind = PDO::PARAM_INT;
|
|
|
} elseif (preg_match('/bool/is', $type)) {
|
|
|
$bind = PDO::PARAM_BOOL;
|
|
|
@@ -1892,11 +2036,10 @@ class Query
|
|
|
$with = explode(',', $with);
|
|
|
}
|
|
|
|
|
|
- $first = true;
|
|
|
- $currentModel = $this->model;
|
|
|
+ $first = true;
|
|
|
|
|
|
/** @var Model $class */
|
|
|
- $class = new $currentModel;
|
|
|
+ $class = $this->model;
|
|
|
foreach ($with as $key => $relation) {
|
|
|
$subRelation = '';
|
|
|
$closure = false;
|
|
|
@@ -1955,7 +2098,7 @@ class Query
|
|
|
$relation = $key;
|
|
|
}
|
|
|
$relation = Loader::parseName($relation, 1, false);
|
|
|
- $count = '(' . (new $this->model)->$relation()->getRelationCountQuery($closure) . ')';
|
|
|
+ $count = '(' . $this->model->$relation()->getRelationCountQuery($closure) . ')';
|
|
|
$this->field([$count => Loader::parseName($relation) . '_count']);
|
|
|
}
|
|
|
}
|
|
|
@@ -2082,7 +2225,7 @@ class Query
|
|
|
}
|
|
|
|
|
|
// 执行操作
|
|
|
- $result = 0 === $sql ? 0 : $this->execute($sql, $bind);
|
|
|
+ $result = 0 === $sql ? 0 : $this->execute($sql, $bind, $this);
|
|
|
if ($result) {
|
|
|
$sequence = $sequence ?: (isset($options['sequence']) ? $options['sequence'] : null);
|
|
|
$lastInsId = $this->getLastInsID($sequence);
|
|
|
@@ -2118,27 +2261,40 @@ class Query
|
|
|
/**
|
|
|
* 批量插入记录
|
|
|
* @access public
|
|
|
- * @param mixed $dataSet 数据集
|
|
|
- * @param boolean $replace 是否replace
|
|
|
+ * @param mixed $dataSet 数据集
|
|
|
+ * @param boolean $replace 是否replace
|
|
|
+ * @param integer $limit 每次写入数据限制
|
|
|
* @return integer|string
|
|
|
*/
|
|
|
- public function insertAll(array $dataSet, $replace = false)
|
|
|
+ public function insertAll(array $dataSet, $replace = false, $limit = null)
|
|
|
{
|
|
|
// 分析查询表达式
|
|
|
$options = $this->parseExpress();
|
|
|
if (!is_array(reset($dataSet))) {
|
|
|
return false;
|
|
|
}
|
|
|
+
|
|
|
// 生成SQL语句
|
|
|
- $sql = $this->builder->insertAll($dataSet, $options, $replace);
|
|
|
+ if (is_null($limit)) {
|
|
|
+ $sql = $this->builder->insertAll($dataSet, $options, $replace);
|
|
|
+ } else {
|
|
|
+ $array = array_chunk($dataSet, $limit, true);
|
|
|
+ foreach ($array as $item) {
|
|
|
+ $sql[] = $this->builder->insertAll($item, $options, $replace);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
// 获取参数绑定
|
|
|
$bind = $this->getBind();
|
|
|
if ($options['fetch_sql']) {
|
|
|
// 获取实际执行的SQL语句
|
|
|
return $this->connection->getRealSql($sql, $bind);
|
|
|
+ } elseif (is_array($sql)) {
|
|
|
+ // 执行操作
|
|
|
+ return $this->batchQuery($sql, $bind, $this);
|
|
|
} else {
|
|
|
// 执行操作
|
|
|
- return $this->execute($sql, $bind);
|
|
|
+ return $this->execute($sql, $bind, $this);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -2164,7 +2320,7 @@ class Query
|
|
|
return $this->connection->getRealSql($sql, $bind);
|
|
|
} else {
|
|
|
// 执行操作
|
|
|
- return $this->execute($sql, $bind);
|
|
|
+ return $this->execute($sql, $bind, $this);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -2231,7 +2387,7 @@ class Query
|
|
|
Cache::clear($options['cache']['tag']);
|
|
|
}
|
|
|
// 执行操作
|
|
|
- $result = '' == $sql ? 0 : $this->execute($sql, $bind);
|
|
|
+ $result = '' == $sql ? 0 : $this->execute($sql, $bind, $this);
|
|
|
if ($result) {
|
|
|
if (is_string($pk) && isset($where[$pk])) {
|
|
|
$data[$pk] = $where[$pk];
|
|
|
@@ -2300,7 +2456,7 @@ class Query
|
|
|
// 判断查询缓存
|
|
|
$cache = $options['cache'];
|
|
|
unset($options['cache']);
|
|
|
- $key = is_string($cache['key']) ? $cache['key'] : md5(serialize($options) . serialize($this->bind));
|
|
|
+ $key = is_string($cache['key']) ? $cache['key'] : md5($this->connection->getConfig('database') . '.' . serialize($options) . serialize($this->bind));
|
|
|
$resultSet = Cache::get($key);
|
|
|
}
|
|
|
if (false === $resultSet) {
|
|
|
@@ -2334,11 +2490,10 @@ class Query
|
|
|
// 数据列表读取后的处理
|
|
|
if (!empty($this->model)) {
|
|
|
// 生成模型对象
|
|
|
- $modelName = $this->model;
|
|
|
if (count($resultSet) > 0) {
|
|
|
foreach ($resultSet as $key => $result) {
|
|
|
/** @var Model $model */
|
|
|
- $model = new $modelName($result);
|
|
|
+ $model = $this->model->newInstance($result);
|
|
|
$model->isUpdate(true);
|
|
|
|
|
|
// 关联查询
|
|
|
@@ -2358,7 +2513,7 @@ class Query
|
|
|
// 模型数据集转换
|
|
|
$resultSet = $model->toCollection($resultSet);
|
|
|
} else {
|
|
|
- $resultSet = (new $modelName)->toCollection($resultSet);
|
|
|
+ $resultSet = $this->model->toCollection($resultSet);
|
|
|
}
|
|
|
} elseif ('collection' == $this->connection->getConfig('resultset_type')) {
|
|
|
// 返回Collection对象
|
|
|
@@ -2402,10 +2557,16 @@ class Query
|
|
|
} elseif (is_array($value) && is_string($value[0]) && 'eq' == strtolower($value[0])) {
|
|
|
$data = $value[1];
|
|
|
}
|
|
|
+ $prefix = $this->connection->getConfig('database') . '.';
|
|
|
+
|
|
|
if (isset($data)) {
|
|
|
- return 'think:' . (is_array($options['table']) ? key($options['table']) : $options['table']) . '|' . $data;
|
|
|
- } else {
|
|
|
- return md5(serialize($options) . serialize($bind));
|
|
|
+ return 'think:' . $prefix . (is_array($options['table']) ? key($options['table']) : $options['table']) . '|' . $data;
|
|
|
+ }
|
|
|
+
|
|
|
+ try {
|
|
|
+ return md5($prefix . serialize($options) . serialize($bind));
|
|
|
+ } catch (\Exception $e) {
|
|
|
+ throw new Exception('closure not support cache(true)');
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -2442,11 +2603,11 @@ class Query
|
|
|
// 判断查询缓存
|
|
|
$cache = $options['cache'];
|
|
|
if (true === $cache['key'] && !is_null($data) && !is_array($data)) {
|
|
|
- $key = 'think:' . (is_array($options['table']) ? key($options['table']) : $options['table']) . '|' . $data;
|
|
|
+ $key = 'think:' . $this->connection->getConfig('database') . '.' . (is_array($options['table']) ? key($options['table']) : $options['table']) . '|' . $data;
|
|
|
} elseif (is_string($cache['key'])) {
|
|
|
$key = $cache['key'];
|
|
|
} elseif (!isset($key)) {
|
|
|
- $key = md5(serialize($options) . serialize($this->bind));
|
|
|
+ $key = md5($this->connection->getConfig('database') . '.' . serialize($options) . serialize($this->bind));
|
|
|
}
|
|
|
$result = Cache::get($key);
|
|
|
}
|
|
|
@@ -2484,7 +2645,7 @@ class Query
|
|
|
$result = isset($resultSet[0]) ? $resultSet[0] : null;
|
|
|
}
|
|
|
|
|
|
- if (isset($cache) && false !== $result) {
|
|
|
+ if (isset($cache) && $result) {
|
|
|
// 缓存数据
|
|
|
$this->cacheData($key, $result, $cache);
|
|
|
}
|
|
|
@@ -2494,8 +2655,7 @@ class Query
|
|
|
if (!empty($result)) {
|
|
|
if (!empty($this->model)) {
|
|
|
// 返回模型对象
|
|
|
- $model = $this->model;
|
|
|
- $result = new $model($result);
|
|
|
+ $result = $this->model->newInstance($result);
|
|
|
$result->isUpdate(true, isset($options['where']['AND']) ? $options['where']['AND'] : null);
|
|
|
// 关联查询
|
|
|
if (!empty($options['relation'])) {
|
|
|
@@ -2526,7 +2686,8 @@ class Query
|
|
|
protected function throwNotFound($options = [])
|
|
|
{
|
|
|
if (!empty($this->model)) {
|
|
|
- throw new ModelNotFoundException('model data Not Found:' . $this->model, $this->model, $options);
|
|
|
+ $class = get_class($this->model);
|
|
|
+ throw new ModelNotFoundException('model data Not Found:' . $class, $class, $options);
|
|
|
} else {
|
|
|
$table = is_array($options['table']) ? key($options['table']) : $options['table'];
|
|
|
throw new DataNotFoundException('table data not Found:' . $table, $table, $options);
|
|
|
@@ -2574,48 +2735,54 @@ class Query
|
|
|
public function chunk($count, $callback, $column = null, $order = 'asc')
|
|
|
{
|
|
|
$options = $this->getOptions();
|
|
|
- if (isset($options['table'])) {
|
|
|
- $table = is_array($options['table']) ? key($options['table']) : $options['table'];
|
|
|
- } else {
|
|
|
- $table = '';
|
|
|
- }
|
|
|
- $column = $column ?: $this->getPk($table);
|
|
|
- if (is_array($column)) {
|
|
|
- $column = $column[0];
|
|
|
+ if (empty($options['table'])) {
|
|
|
+ $options['table'] = $this->getTable();
|
|
|
}
|
|
|
+ $column = $column ?: $this->getPk($options);
|
|
|
+
|
|
|
if (isset($options['order'])) {
|
|
|
if (App::$debug) {
|
|
|
throw new \LogicException('chunk not support call order');
|
|
|
}
|
|
|
unset($options['order']);
|
|
|
}
|
|
|
- $bind = $this->bind;
|
|
|
- $resultSet = $this->options($options)->limit($count)->order($column, $order)->select();
|
|
|
- if (strpos($column, '.')) {
|
|
|
- list($alias, $key) = explode('.', $column);
|
|
|
+ $bind = $this->bind;
|
|
|
+ if (is_array($column)) {
|
|
|
+ $times = 1;
|
|
|
+ $query = $this->options($options)->page($times, $count);
|
|
|
} else {
|
|
|
- $key = $column;
|
|
|
- }
|
|
|
- if ($resultSet instanceof Collection) {
|
|
|
- $resultSet = $resultSet->all();
|
|
|
+ if (strpos($column, '.')) {
|
|
|
+ list($alias, $key) = explode('.', $column);
|
|
|
+ } else {
|
|
|
+ $key = $column;
|
|
|
+ }
|
|
|
+ $query = $this->options($options)->limit($count);
|
|
|
}
|
|
|
+ $resultSet = $query->order($column, $order)->select();
|
|
|
+
|
|
|
+ while (count($resultSet) > 0) {
|
|
|
+ if ($resultSet instanceof Collection) {
|
|
|
+ $resultSet = $resultSet->all();
|
|
|
+ }
|
|
|
|
|
|
- while (!empty($resultSet)) {
|
|
|
if (false === call_user_func($callback, $resultSet)) {
|
|
|
return false;
|
|
|
}
|
|
|
- $end = end($resultSet);
|
|
|
- $lastId = is_array($end) ? $end[$key] : $end->$key;
|
|
|
- $resultSet = $this->options($options)
|
|
|
- ->limit($count)
|
|
|
- ->bind($bind)
|
|
|
- ->where($column, 'asc' == strtolower($order) ? '>' : '<', $lastId)
|
|
|
- ->order($column, $order)
|
|
|
- ->select();
|
|
|
- if ($resultSet instanceof Collection) {
|
|
|
- $resultSet = $resultSet->all();
|
|
|
+
|
|
|
+ if (is_array($column)) {
|
|
|
+ $times++;
|
|
|
+ $query = $this->options($options)->page($times, $count);
|
|
|
+ } else {
|
|
|
+ $end = end($resultSet);
|
|
|
+ $lastId = is_array($end) ? $end[$key] : $end->getData($key);
|
|
|
+ $query = $this->options($options)
|
|
|
+ ->limit($count)
|
|
|
+ ->where($column, 'asc' == strtolower($order) ? '>' : '<', $lastId);
|
|
|
}
|
|
|
+
|
|
|
+ $resultSet = $query->bind($bind)->order($column, $order)->select();
|
|
|
}
|
|
|
+
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
@@ -2692,7 +2859,7 @@ class Query
|
|
|
Cache::clear($options['cache']['tag']);
|
|
|
}
|
|
|
// 执行操作
|
|
|
- $result = $this->execute($sql, $bind);
|
|
|
+ $result = $this->execute($sql, $bind, $this);
|
|
|
if ($result) {
|
|
|
if (!is_array($data) && is_string($pk) && isset($key) && strpos($key, '|')) {
|
|
|
list($a, $val) = explode('|', $key);
|
|
|
@@ -2777,6 +2944,10 @@ class Query
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+ if (isset(static::$readMaster['*']) || (is_string($options['table']) && isset(static::$readMaster[$options['table']]))) {
|
|
|
+ $options['master'] = true;
|
|
|
+ }
|
|
|
+
|
|
|
foreach (['join', 'union', 'group', 'having', 'limit', 'order', 'force', 'comment'] as $name) {
|
|
|
if (!isset($options[$name])) {
|
|
|
$options[$name] = '';
|
聆听