452 lines
11 KiB
PHP
452 lines
11 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2006~2019 http://thinkphp.cn All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
// +----------------------------------------------------------------------
|
|
// | Author: liu21st <liu21st@gmail.com>
|
|
// +----------------------------------------------------------------------
|
|
declare (strict_types = 1);
|
|
|
|
namespace think\db;
|
|
|
|
use PDOStatement;
|
|
use think\helper\Str;
|
|
|
|
/**
|
|
* PDO数据查询类
|
|
*/
|
|
class Query extends BaseQuery
|
|
{
|
|
use concern\JoinAndViewQuery;
|
|
use concern\ParamsBind;
|
|
use concern\TableFieldInfo;
|
|
|
|
/**
|
|
* 表达式方式指定Field排序
|
|
* @access public
|
|
* @param string $field 排序字段
|
|
* @param array $bind 参数绑定
|
|
* @return $this
|
|
*/
|
|
public function orderRaw(string $field, array $bind = [])
|
|
{
|
|
$this->options['order'][] = new Raw($field, $bind);
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 表达式方式指定查询字段
|
|
* @access public
|
|
* @param string $field 字段名
|
|
* @return $this
|
|
*/
|
|
public function fieldRaw(string $field)
|
|
{
|
|
$this->options['field'][] = new Raw($field);
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 指定Field排序 orderField('id',[1,2,3],'desc')
|
|
* @access public
|
|
* @param string $field 排序字段
|
|
* @param array $values 排序值
|
|
* @param string $order 排序 desc/asc
|
|
* @return $this
|
|
*/
|
|
public function orderField(string $field, array $values, string $order = '')
|
|
{
|
|
if (!empty($values)) {
|
|
$values['sort'] = $order;
|
|
|
|
$this->options['order'][$field] = $values;
|
|
}
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 随机排序
|
|
* @access public
|
|
* @return $this
|
|
*/
|
|
public function orderRand()
|
|
{
|
|
$this->options['order'][] = '[rand]';
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 使用表达式设置数据
|
|
* @access public
|
|
* @param string $field 字段名
|
|
* @param string $value 字段值
|
|
* @return $this
|
|
*/
|
|
public function exp(string $field, string $value)
|
|
{
|
|
$this->options['data'][$field] = new Raw($value);
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 表达式方式指定当前操作的数据表
|
|
* @access public
|
|
* @param mixed $table 表名
|
|
* @return $this
|
|
*/
|
|
public function tableRaw(string $table)
|
|
{
|
|
$this->options['table'] = new Raw($table);
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 获取执行的SQL语句而不进行实际的查询
|
|
* @access public
|
|
* @param bool $fetch 是否返回sql
|
|
* @return $this|Fetch
|
|
*/
|
|
public function fetchSql(bool $fetch = true)
|
|
{
|
|
$this->options['fetch_sql'] = $fetch;
|
|
|
|
if ($fetch) {
|
|
return new Fetch($this);
|
|
}
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 批处理执行SQL语句
|
|
* 批处理的指令都认为是execute操作
|
|
* @access public
|
|
* @param array $sql SQL批处理指令
|
|
* @return bool
|
|
*/
|
|
public function batchQuery(array $sql = []): bool
|
|
{
|
|
return $this->connection->batchQuery($this, $sql);
|
|
}
|
|
|
|
/**
|
|
* USING支持 用于多表删除
|
|
* @access public
|
|
* @param mixed $using USING
|
|
* @return $this
|
|
*/
|
|
public function using($using)
|
|
{
|
|
$this->options['using'] = $using;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 存储过程调用
|
|
* @access public
|
|
* @param bool $procedure 是否为存储过程查询
|
|
* @return $this
|
|
*/
|
|
public function procedure(bool $procedure = true)
|
|
{
|
|
$this->options['procedure'] = $procedure;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 指定group查询
|
|
* @access public
|
|
* @param string|array $group GROUP
|
|
* @return $this
|
|
*/
|
|
public function group($group)
|
|
{
|
|
$this->options['group'] = $group;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 指定having查询
|
|
* @access public
|
|
* @param string $having having
|
|
* @return $this
|
|
*/
|
|
public function having(string $having)
|
|
{
|
|
$this->options['having'] = $having;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 指定distinct查询
|
|
* @access public
|
|
* @param bool $distinct 是否唯一
|
|
* @return $this
|
|
*/
|
|
public function distinct(bool $distinct = true)
|
|
{
|
|
$this->options['distinct'] = $distinct;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 指定强制索引
|
|
* @access public
|
|
* @param string $force 索引名称
|
|
* @return $this
|
|
*/
|
|
public function force(string $force)
|
|
{
|
|
$this->options['force'] = $force;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 查询注释
|
|
* @access public
|
|
* @param string $comment 注释
|
|
* @return $this
|
|
*/
|
|
public function comment(string $comment)
|
|
{
|
|
$this->options['comment'] = $comment;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置是否REPLACE
|
|
* @access public
|
|
* @param bool $replace 是否使用REPLACE写入数据
|
|
* @return $this
|
|
*/
|
|
public function replace(bool $replace = true)
|
|
{
|
|
$this->options['replace'] = $replace;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置当前查询所在的分区
|
|
* @access public
|
|
* @param string|array $partition 分区名称
|
|
* @return $this
|
|
*/
|
|
public function partition($partition)
|
|
{
|
|
$this->options['partition'] = $partition;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置DUPLICATE
|
|
* @access public
|
|
* @param array|string|Raw $duplicate DUPLICATE信息
|
|
* @return $this
|
|
*/
|
|
public function duplicate($duplicate)
|
|
{
|
|
$this->options['duplicate'] = $duplicate;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置查询的额外参数
|
|
* @access public
|
|
* @param string $extra 额外信息
|
|
* @return $this
|
|
*/
|
|
public function extra(string $extra)
|
|
{
|
|
$this->options['extra'] = $extra;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 创建子查询SQL
|
|
* @access public
|
|
* @param bool $sub 是否添加括号
|
|
* @return string
|
|
* @throws Exception
|
|
*/
|
|
public function buildSql(bool $sub = true): string
|
|
{
|
|
return $sub ? '( ' . $this->fetchSql()->select() . ' )' : $this->fetchSql()->select();
|
|
}
|
|
|
|
/**
|
|
* 获取当前数据表的主键
|
|
* @access public
|
|
* @return string|array
|
|
*/
|
|
public function getPk()
|
|
{
|
|
if (empty($this->pk)) {
|
|
$this->pk = $this->connection->getPk($this->getTable());
|
|
}
|
|
|
|
return $this->pk;
|
|
}
|
|
|
|
/**
|
|
* 指定数据表自增主键
|
|
* @access public
|
|
* @param string $autoinc 自增键
|
|
* @return $this
|
|
*/
|
|
public function autoinc(string $autoinc)
|
|
{
|
|
$this->autoinc = $autoinc;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 获取当前数据表的自增主键
|
|
* @access public
|
|
* @return string|null
|
|
*/
|
|
public function getAutoInc()
|
|
{
|
|
$tableName = $this->getTable();
|
|
|
|
if (empty($this->autoinc) && $tableName) {
|
|
$this->autoinc = $this->connection->getAutoInc($tableName);
|
|
}
|
|
|
|
return $this->autoinc;
|
|
}
|
|
|
|
/**
|
|
* 字段值增长
|
|
* @access public
|
|
* @param string $field 字段名
|
|
* @param float $step 增长值
|
|
* @return $this
|
|
*/
|
|
public function inc(string $field, float $step = 1)
|
|
{
|
|
$this->options['data'][$field] = ['INC', $step];
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 字段值减少
|
|
* @access public
|
|
* @param string $field 字段名
|
|
* @param float $step 增长值
|
|
* @return $this
|
|
*/
|
|
public function dec(string $field, float $step = 1)
|
|
{
|
|
$this->options['data'][$field] = ['DEC', $step];
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 获取当前的查询标识
|
|
* @access public
|
|
* @param mixed $data 要序列化的数据
|
|
* @return string
|
|
*/
|
|
public function getQueryGuid($data = null): string
|
|
{
|
|
return md5($this->getConfig('database') . serialize(var_export($data ?: $this->options, true)) . serialize($this->getBind(false)));
|
|
}
|
|
|
|
/**
|
|
* 执行查询但只返回PDOStatement对象
|
|
* @access public
|
|
* @return PDOStatement
|
|
*/
|
|
public function getPdo(): PDOStatement
|
|
{
|
|
return $this->connection->pdo($this);
|
|
}
|
|
|
|
/**
|
|
* 使用游标查找记录
|
|
* @access public
|
|
* @param mixed $data 数据
|
|
* @return \Generator
|
|
*/
|
|
public function cursor($data = null)
|
|
{
|
|
if (!is_null($data)) {
|
|
// 主键条件分析
|
|
$this->parsePkWhere($data);
|
|
}
|
|
|
|
$this->options['data'] = $data;
|
|
|
|
$connection = clone $this->connection;
|
|
|
|
return $connection->cursor($this);
|
|
}
|
|
|
|
/**
|
|
* 分批数据返回处理
|
|
* @access public
|
|
* @param integer $count 每次处理的数据数量
|
|
* @param callable $callback 处理回调方法
|
|
* @param string|array $column 分批处理的字段名
|
|
* @param string $order 字段排序
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public function chunk(int $count, callable $callback, $column = null, string $order = 'asc'): bool
|
|
{
|
|
$options = $this->getOptions();
|
|
$column = $column ?: $this->getPk();
|
|
|
|
if (isset($options['order'])) {
|
|
unset($options['order']);
|
|
}
|
|
|
|
$bind = $this->bind;
|
|
|
|
if (is_array($column)) {
|
|
$times = 1;
|
|
$query = $this->options($options)->page($times, $count);
|
|
} else {
|
|
$query = $this->options($options)->limit($count);
|
|
|
|
if (strpos($column, '.')) {
|
|
[$alias, $key] = explode('.', $column);
|
|
} else {
|
|
$key = $column;
|
|
}
|
|
}
|
|
|
|
$resultSet = $query->order($column, $order)->select();
|
|
|
|
while (count($resultSet) > 0) {
|
|
if (false === call_user_func($callback, $resultSet)) {
|
|
return false;
|
|
}
|
|
|
|
if (isset($times)) {
|
|
$times++;
|
|
$query = $this->options($options)->page($times, $count);
|
|
} else {
|
|
$end = $resultSet->pop();
|
|
$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;
|
|
}
|
|
}
|