123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173 |
- <?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);
- } else {
- $table = $this->connection->parseSqlTable($table);
- if (isset($options['alias'][$table])) {
- $item[] = $this->parseKey($query, $table) . ' ' . $this->parseKey($query, $options['alias'][$table]);
- } else {
- $item[] = $this->parseKey($query, $table);
- }
- }
- }
- return implode(',', $item);
- }
- /**
- * where分析
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $where 查询条件
- * @return string
- */
- protected function parseWhere(Query $query, $where)
- {
- $options = $query->getOptions();
- $whereStr = $this->buildWhere($query, $where);
- if (!empty($options['soft_delete'])) {
- // 附加软删除条件
- list($field, $condition) = $options['soft_delete'];
- $binds = $this->connection->getFieldsBind($options['table']);
- $whereStr = $whereStr ? '( ' . $whereStr . ' ) AND ' : '';
- $whereStr = $whereStr . $this->parseWhereItem($query, $field, $condition, '', $binds);
- }
- return empty($whereStr) ? '' : ' WHERE ' . $whereStr;
- }
- /**
- * 生成查询条件SQL
- * @access public
- * @param Query $query 查询对象
- * @param mixed $where 查询条件
- * @return string
- */
- public function buildWhere(Query $query, $where)
- {
- if (empty($where)) {
- $where = [];
- }
- $whereStr = '';
- $binds = $this->connection->getFieldsBind($query->getOptions('table'));
- foreach ($where as $logic => $val) {
- $str = [];
- foreach ($val as $value) {
- if ($value instanceof Expression) {
- $str[] = ' ' . $logic . ' ( ' . $value->getValue() . ' )';
- continue;
- }
- if (is_array($value)) {
- if (key($value) !== 0) {
- throw new Exception('where express error:' . var_export($value, true));
- }
- $field = array_shift($value);
- } elseif (!($value instanceof \Closure)) {
- throw new Exception('where express error:' . var_export($value, true));
- }
- if ($value instanceof \Closure) {
- // 使用闭包查询
- $newQuery = $query->newQuery()->setConnection($this->connection);
- $value($newQuery);
- $whereClause = $this->buildWhere($newQuery, $newQuery->getOptions('where'));
- if (!empty($whereClause)) {
- $query->bind($newQuery->getBind(false));
- $str[] = ' ' . $logic . ' ( ' . $whereClause . ' )';
- }
- } elseif (is_array($field)) {
- array_unshift($value, $field);
- $str2 = [];
- foreach ($value as $item) {
- $str2[] = $this->parseWhereItem($query, array_shift($item), $item, $logic, $binds);
- }
- $str[] = ' ' . $logic . ' ( ' . implode(' AND ', $str2) . ' )';
- } elseif (strpos($field, '|')) {
- // 不同字段使用相同查询条件(OR)
- $array = explode('|', $field);
- $item = [];
- foreach ($array as $k) {
- $item[] = $this->parseWhereItem($query, $k, $value, '', $binds);
- }
- $str[] = ' ' . $logic . ' ( ' . implode(' OR ', $item) . ' )';
- } elseif (strpos($field, '&')) {
- // 不同字段使用相同查询条件(AND)
- $array = explode('&', $field);
- $item = [];
- foreach ($array as $k) {
- $item[] = $this->parseWhereItem($query, $k, $value, '', $binds);
- }
- $str[] = ' ' . $logic . ' ( ' . implode(' AND ', $item) . ' )';
- } else {
- // 对字段使用表达式查询
- $field = is_string($field) ? $field : '';
- $str[] = ' ' . $logic . ' ' . $this->parseWhereItem($query, $field, $value, $logic, $binds);
- }
- }
- $whereStr .= empty($whereStr) ? substr(implode(' ', $str), strlen($logic) + 1) : implode(' ', $str);
- }
- return $whereStr;
- }
- // where子单元分析
- protected function parseWhereItem(Query $query, $field, $val, $rule = '', $binds = [])
- {
- // 字段分析
- $key = $field ? $this->parseKey($query, $field, true) : '';
- // 查询规则和条件
- if (!is_array($val)) {
- $val = is_null($val) ? ['NULL', ''] : ['=', $val];
- }
- list($exp, $value) = $val;
- // 对一个字段使用多个查询条件
- if (is_array($exp)) {
- $item = array_pop($val);
- // 传入 or 或者 and
- if (is_string($item) && in_array($item, ['AND', 'and', 'OR', 'or'])) {
- $rule = $item;
- } else {
- array_push($val, $item);
- }
- foreach ($val as $k => $item) {
- $str[] = $this->parseWhereItem($query, $field, $item, $rule, $binds);
- }
- return '( ' . implode(' ' . $rule . ' ', $str) . ' )';
- }
- // 检测操作符
- $exp = strtoupper($exp);
- if (isset($this->exp[$exp])) {
- $exp = $this->exp[$exp];
- }
- if ($value instanceof Expression) {
- } elseif (is_object($value) && method_exists($value, '__toString')) {
- // 对象数据写入
- $value = $value->__toString();
- }
- if (strpos($field, '->')) {
- $jsonType = $query->getJsonFieldType($field);
- $bindType = $this->connection->getFieldBindType($jsonType);
- } else {
- $bindType = isset($binds[$field]) && 'LIKE' != $exp ? $binds[$field] : PDO::PARAM_STR;
- }
- if (is_scalar($value) && !in_array($exp, ['EXP', 'NOT NULL', 'NULL', 'IN', 'NOT IN', 'BETWEEN', 'NOT BETWEEN']) && strpos($exp, 'TIME') === false) {
- if (0 === strpos($value, ':') && $query->isBind(substr($value, 1))) {
- } else {
- $name = $query->bind($value, $bindType);
- $value = ':' . $name;
- }
- }
- // 解析查询表达式
- foreach ($this->parser as $fun => $parse) {
- if (in_array($exp, $parse)) {
- $whereStr = $this->$fun($query, $key, $exp, $value, $field, $bindType, isset($val[2]) ? $val[2] : 'AND');
- break;
- }
- }
- if (!isset($whereStr)) {
- throw new Exception('where express error:' . $exp);
- }
- return $whereStr;
- }
- /**
- * 模糊查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @param string $logic
- * @return string
- */
- protected function parseLike(Query $query, $key, $exp, $value, $field, $bindType, $logic)
- {
- // 模糊匹配
- if (is_array($value)) {
- foreach ($value as $item) {
- $name = $query->bind($item, PDO::PARAM_STR);
- $array[] = $key . ' ' . $exp . ' :' . $name;
- }
- $whereStr = '(' . implode(' ' . strtoupper($logic) . ' ', $array) . ')';
- } else {
- $whereStr = $key . ' ' . $exp . ' ' . $value;
- }
- return $whereStr;
- }
- /**
- * 表达式查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param array $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseColumn(Query $query, $key, $exp, array $value, $field, $bindType)
- {
- // 字段比较查询
- list($op, $field2) = $value;
- if (!in_array($op, ['=', '<>', '>', '>=', '<', '<='])) {
- throw new Exception('where express error:' . var_export($value, true));
- }
- return '( ' . $key . ' ' . $op . ' ' . $this->parseKey($query, $field2, true) . ' )';
- }
- /**
- * 表达式查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param Expression $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseExp(Query $query, $key, $exp, Expression $value, $field, $bindType)
- {
- // 表达式查询
- return '( ' . $key . ' ' . $value->getValue() . ' )';
- }
- /**
- * Null查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseNull(Query $query, $key, $exp, $value, $field, $bindType)
- {
- // NULL 查询
- return $key . ' IS ' . $exp;
- }
- /**
- * 范围查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseBetween(Query $query, $key, $exp, $value, $field, $bindType)
- {
- // BETWEEN 查询
- $data = is_array($value) ? $value : explode(',', $value);
- $min = $query->bind($data[0], $bindType);
- $max = $query->bind($data[1], $bindType);
- return $key . ' ' . $exp . ' :' . $min . ' AND :' . $max . ' ';
- }
- /**
- * Exists查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseExists(Query $query, $key, $exp, $value, $field, $bindType)
- {
- // EXISTS 查询
- if ($value instanceof \Closure) {
- $value = $this->parseClosure($query, $value, false);
- } elseif ($value instanceof Expression) {
- $value = $value->getValue();
- } else {
- throw new Exception('where express error:' . $value);
- }
- return $exp . ' (' . $value . ')';
- }
- /**
- * 时间比较查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseTime(Query $query, $key, $exp, $value, $field, $bindType)
- {
- return $key . ' ' . substr($exp, 0, 2) . ' ' . $this->parseDateTime($query, $value, $field, $bindType);
- }
- /**
- * 大小比较查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseCompare(Query $query, $key, $exp, $value, $field, $bindType)
- {
- if (is_array($value)) {
- throw new Exception('where express error:' . $exp . var_export($value, true));
- }
- // 比较运算
- if ($value instanceof \Closure) {
- $value = $this->parseClosure($query, $value);
- }
- if ('=' == $exp && is_null($value)) {
- return $key . ' IS NULL';
- }
- return $key . ' ' . $exp . ' ' . $value;
- }
- /**
- * 时间范围查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseBetweenTime(Query $query, $key, $exp, $value, $field, $bindType)
- {
- if (is_string($value)) {
- $value = explode(',', $value);
- }
- return $key . ' ' . substr($exp, 0, -4)
- . $this->parseDateTime($query, $value[0], $field, $bindType)
- . ' AND '
- . $this->parseDateTime($query, $value[1], $field, $bindType);
- }
- /**
- * IN查询
- * @access protected
- * @param Query $query 查询对象
- * @param string $key
- * @param string $exp
- * @param mixed $value
- * @param string $field
- * @param integer $bindType
- * @return string
- */
- protected function parseIn(Query $query, $key, $exp, $value, $field, $bindType)
- {
- // IN 查询
- if ($value instanceof \Closure) {
- $value = $this->parseClosure($query, $value, false);
- } elseif ($value instanceof Expression) {
- $value = $value->getValue();
- } else {
- $value = array_unique(is_array($value) ? $value : explode(',', $value));
- $array = [];
- foreach ($value as $k => $v) {
- $name = $query->bind($v, $bindType);
- $array[] = ':' . $name;
- }
- if (count($array) == 1) {
- return $key . ('IN' == $exp ? ' = ' : ' <> ') . $array[0];
- } else {
- $zone = implode(',', $array);
- $value = empty($zone) ? "''" : $zone;
- }
- }
- return $key . ' ' . $exp . ' (' . $value . ')';
- }
- /**
- * 闭包子查询
- * @access protected
- * @param Query $query 查询对象
- * @param \Closure $call
- * @param bool $show
- * @return string
- */
- protected function parseClosure(Query $query, $call, $show = true)
- {
- $newQuery = $query->newQuery()->setConnection($this->connection);
- $call($newQuery);
- return $newQuery->buildSql($show);
- }
- /**
- * 日期时间条件解析
- * @access protected
- * @param Query $query 查询对象
- * @param string $value
- * @param string $key
- * @param integer $bindType
- * @return string
- */
- protected function parseDateTime(Query $query, $value, $key, $bindType = null)
- {
- $options = $query->getOptions();
- // 获取时间字段类型
- if (strpos($key, '.')) {
- list($table, $key) = explode('.', $key);
- if (isset($options['alias']) && $pos = array_search($table, $options['alias'])) {
- $table = $pos;
- }
- } else {
- $table = $options['table'];
- }
- $type = $this->connection->getTableInfo($table, 'type');
- if (isset($type[$key])) {
- $info = $type[$key];
- }
- if (isset($info)) {
- if (is_string($value)) {
- $value = strtotime($value) ?: $value;
- }
- if (preg_match('/(datetime|timestamp)/is', $info)) {
- // 日期及时间戳类型
- $value = date('Y-m-d H:i:s', $value);
- } elseif (preg_match('/(date)/is', $info)) {
- // 日期及时间戳类型
- $value = date('Y-m-d', $value);
- }
- }
- $name = $query->bind($value, $bindType);
- return ':' . $name;
- }
- /**
- * limit分析
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $limit
- * @return string
- */
- protected function parseLimit(Query $query, $limit)
- {
- return (!empty($limit) && false === strpos($limit, '(')) ? ' LIMIT ' . $limit . ' ' : '';
- }
- /**
- * join分析
- * @access protected
- * @param Query $query 查询对象
- * @param array $join
- * @return string
- */
- protected function parseJoin(Query $query, $join)
- {
- $joinStr = '';
- if (!empty($join)) {
- foreach ($join as $item) {
- list($table, $type, $on) = $item;
- $condition = [];
- foreach ((array) $on as $val) {
- if ($val instanceof Expression) {
- $condition[] = $val->getValue();
- } elseif (strpos($val, '=')) {
- list($val1, $val2) = explode('=', $val, 2);
- $condition[] = $this->parseKey($query, $val1) . '=' . $this->parseKey($query, $val2);
- } else {
- $condition[] = $val;
- }
- }
- $table = $this->parseTable($query, $table);
- $joinStr .= ' ' . $type . ' JOIN ' . $table . ' ON ' . implode(' AND ', $condition);
- }
- }
- return $joinStr;
- }
- /**
- * order分析
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $order
- * @return string
- */
- protected function parseOrder(Query $query, $order)
- {
- foreach ($order as $key => $val) {
- if ($val instanceof Expression) {
- $array[] = $val->getValue();
- } elseif (is_array($val) && preg_match('/^[\w\.]+$/', $key)) {
- $array[] = $this->parseOrderField($query, $key, $val);
- } elseif ('[rand]' == $val) {
- $array[] = $this->parseRand($query);
- } elseif (is_string($val)) {
- if (is_numeric($key)) {
- list($key, $sort) = explode(' ', strpos($val, ' ') ? $val : $val . ' ');
- } else {
- $sort = $val;
- }
- if (preg_match('/^[\w\.]+$/', $key)) {
- $sort = strtoupper($sort);
- $sort = in_array($sort, ['ASC', 'DESC'], true) ? ' ' . $sort : '';
- $array[] = $this->parseKey($query, $key, true) . $sort;
- } else {
- throw new Exception('order express error:' . $key);
- }
- }
- }
- return empty($array) ? '' : ' ORDER BY ' . implode(',', $array);
- }
- /**
- * orderField分析
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $key
- * @param array $val
- * @return string
- */
- protected function parseOrderField($query, $key, $val)
- {
- if (isset($val['sort'])) {
- $sort = $val['sort'];
- unset($val['sort']);
- } else {
- $sort = '';
- }
- $sort = strtoupper($sort);
- $sort = in_array($sort, ['ASC', 'DESC'], true) ? ' ' . $sort : '';
- $options = $query->getOptions();
- $bind = $this->connection->getFieldsBind($options['table']);
- foreach ($val as $k => $item) {
- $val[$k] = $this->parseDataBind($query, $key, $item, $bind);
- }
- return 'field(' . $this->parseKey($query, $key, true) . ',' . implode(',', $val) . ')' . $sort;
- }
- /**
- * group分析
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $group
- * @return string
- */
- protected function parseGroup(Query $query, $group)
- {
- if (empty($group)) {
- return '';
- }
- if (is_string($group)) {
- $group = explode(',', $group);
- }
- foreach ($group as $key) {
- $val[] = $this->parseKey($query, $key);
- }
- return ' GROUP BY ' . implode(',', $val);
- }
- /**
- * having分析
- * @access protected
- * @param Query $query 查询对象
- * @param string $having
- * @return string
- */
- protected function parseHaving(Query $query, $having)
- {
- return !empty($having) ? ' HAVING ' . $having : '';
- }
- /**
- * comment分析
- * @access protected
- * @param Query $query 查询对象
- * @param string $comment
- * @return string
- */
- protected function parseComment(Query $query, $comment)
- {
- if (false !== strpos($comment, '*/')) {
- $comment = strstr($comment, '*/', true);
- }
- return !empty($comment) ? ' /* ' . $comment . ' */' : '';
- }
- /**
- * distinct分析
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $distinct
- * @return string
- */
- protected function parseDistinct(Query $query, $distinct)
- {
- return !empty($distinct) ? ' DISTINCT ' : '';
- }
- /**
- * union分析
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $union
- * @return string
- */
- protected function parseUnion(Query $query, $union)
- {
- if (empty($union)) {
- return '';
- }
- $type = $union['type'];
- unset($union['type']);
- foreach ($union as $u) {
- if ($u instanceof \Closure) {
- $sql[] = $type . ' ' . $this->parseClosure($query, $u);
- } elseif (is_string($u)) {
- $sql[] = $type . ' ( ' . $this->connection->parseSqlTable($u) . ' )';
- }
- }
- return ' ' . implode(' ', $sql);
- }
- /**
- * index分析,可在操作链中指定需要强制使用的索引
- * @access protected
- * @param Query $query 查询对象
- * @param mixed $index
- * @return string
- */
- protected function parseForce(Query $query, $index)
- {
- if (empty($index)) {
- return '';
- }
- return sprintf(" FORCE INDEX ( %s ) ", is_array($index) ? implode(',', $index) : $index);
- }
- /**
- * 设置锁机制
- * @access protected
- * @param Query $query 查询对象
- * @param bool|string $lock
- * @return string
- */
- protected function parseLock(Query $query, $lock = false)
- {
- if (is_bool($lock)) {
- return $lock ? ' FOR UPDATE ' : '';
- } elseif (is_string($lock) && !empty($lock)) {
- return ' ' . trim($lock) . ' ';
- }
- }
- /**
- * 生成查询SQL
- * @access public
- * @param Query $query 查询对象
- * @return string
- */
- public function select(Query $query)
- {
- $options = $query->getOptions();
- return str_replace(
- ['%TABLE%', '%DISTINCT%', '%FIELD%', '%JOIN%', '%WHERE%', '%GROUP%', '%HAVING%', '%ORDER%', '%LIMIT%', '%UNION%', '%LOCK%', '%COMMENT%', '%FORCE%'],
- [
- $this->parseTable($query, $options['table']),
- $this->parseDistinct($query, $options['distinct']),
- $this->parseField($query, $options['field']),
- $this->parseJoin($query, $options['join']),
- $this->parseWhere($query, $options['where']),
- $this->parseGroup($query, $options['group']),
- $this->parseHaving($query, $options['having']),
- $this->parseOrder($query, $options['order']),
- $this->parseLimit($query, $options['limit']),
- $this->parseUnion($query, $options['union']),
- $this->parseLock($query, $options['lock']),
- $this->parseComment($query, $options['comment']),
- $this->parseForce($query, $options['force']),
- ],
- $this->selectSql);
- }
- /**
- * 生成Insert SQL
- * @access public
- * @param Query $query 查询对象
- * @param bool $replace 是否replace
- * @return string
- */
- public function insert(Query $query, $replace = false)
- {
- $options = $query->getOptions();
- // 分析并处理数据
- $data = $this->parseData($query, $options['data']);
- if (empty($data)) {
- return '';
- }
- $fields = array_keys($data);
- $values = array_values($data);
- return str_replace(
- ['%INSERT%', '%TABLE%', '%FIELD%', '%DATA%', '%COMMENT%'],
- [
- $replace ? 'REPLACE' : 'INSERT',
- $this->parseTable($query, $options['table']),
- implode(' , ', $fields),
- implode(' , ', $values),
- $this->parseComment($query, $options['comment']),
- ],
- $this->insertSql);
- }
- /**
- * 生成insertall SQL
- * @access public
- * @param Query $query 查询对象
- * @param array $dataSet 数据集
- * @param bool $replace 是否replace
- * @return string
- */
- public function insertAll(Query $query, $dataSet, $replace = false)
- {
- $options = $query->getOptions();
- // 获取合法的字段
- if ('*' == $options['field']) {
- $allowFields = $this->connection->getTableFields($options['table']);
- } else {
- $allowFields = $options['field'];
- }
- // 获取绑定信息
- $bind = $this->connection->getFieldsBind($options['table']);
- foreach ($dataSet as $data) {
- $data = $this->parseData($query, $data, $allowFields, $bind);
- $values[] = 'SELECT ' . implode(',', array_values($data));
- if (!isset($insertFields)) {
- $insertFields = array_keys($data);
- }
- }
- $fields = [];
- foreach ($insertFields as $field) {
- $fields[] = $this->parseKey($query, $field);
- }
- return str_replace(
- ['%INSERT%', '%TABLE%', '%FIELD%', '%DATA%', '%COMMENT%'],
- [
- $replace ? 'REPLACE' : 'INSERT',
- $this->parseTable($query, $options['table']),
- implode(' , ', $fields),
- implode(' UNION ALL ', $values),
- $this->parseComment($query, $options['comment']),
- ],
- $this->insertAllSql);
- }
- /**
- * 生成slect insert SQL
- * @access public
- * @param Query $query 查询对象
- * @param array $fields 数据
- * @param string $table 数据表
- * @return string
- */
- public function selectInsert(Query $query, $fields, $table)
- {
- if (is_string($fields)) {
- $fields = explode(',', $fields);
- }
- foreach ($fields as &$field) {
- $field = $this->parseKey($query, $field, true);
- }
- return 'INSERT INTO ' . $this->parseTable($query, $table) . ' (' . implode(',', $fields) . ') ' . $this->select($query);
- }
- /**
- * 生成update SQL
- * @access public
- * @param Query $query 查询对象
- * @return string
- */
- public function update(Query $query)
- {
- $options = $query->getOptions();
- $data = $this->parseData($query, $options['data']);
- if (empty($data)) {
- return '';
- }
- foreach ($data as $key => $val) {
- $set[] = $key . ' = ' . $val;
- }
- return str_replace(
- ['%TABLE%', '%SET%', '%JOIN%', '%WHERE%', '%ORDER%', '%LIMIT%', '%LOCK%', '%COMMENT%'],
- [
- $this->parseTable($query, $options['table']),
- implode(' , ', $set),
- $this->parseJoin($query, $options['join']),
- $this->parseWhere($query, $options['where']),
- $this->parseOrder($query, $options['order']),
- $this->parseLimit($query, $options['limit']),
- $this->parseLock($query, $options['lock']),
- $this->parseComment($query, $options['comment']),
- ],
- $this->updateSql);
- }
- /**
- * 生成delete SQL
- * @access public
- * @param Query $query 查询对象
- * @return string
- */
- public function delete(Query $query)
- {
- $options = $query->getOptions();
- return str_replace(
- ['%TABLE%', '%USING%', '%JOIN%', '%WHERE%', '%ORDER%', '%LIMIT%', '%LOCK%', '%COMMENT%'],
- [
- $this->parseTable($query, $options['table']),
- !empty($options['using']) ? ' USING ' . $this->parseTable($query, $options['using']) . ' ' : '',
- $this->parseJoin($query, $options['join']),
- $this->parseWhere($query, $options['where']),
- $this->parseOrder($query, $options['order']),
- $this->parseLimit($query, $options['limit']),
- $this->parseLock($query, $options['lock']),
- $this->parseComment($query, $options['comment']),
- ],
- $this->deleteSql);
- }
- }
|