// +---------------------------------------------------------------------- namespace think\db; use PDO; use think\Exception; abstract class Builder { // connection对象实例 protected $connection; // 查询对象实例 protected $query; // 数据库表达式 protected $exp = ['eq' => '=', 'neq' => '<>', 'gt' => '>', 'egt' => '>=', 'lt' => '<', 'elt' => '<=', 'notlike' => 'NOT LIKE', 'not like' => 'NOT LIKE', 'like' => 'LIKE', 'in' => 'IN', 'exp' => 'EXP', 'notin' => 'NOT IN', 'not in' => 'NOT IN', 'between' => 'BETWEEN', 'not between' => 'NOT BETWEEN', 'notbetween' => 'NOT BETWEEN', 'exists' => 'EXISTS', 'notexists' => 'NOT EXISTS', 'not exists' => 'NOT EXISTS', 'null' => 'NULL', 'notnull' => 'NOT NULL', 'not null' => 'NOT NULL', '> time' => '> TIME', '< time' => '< TIME', '>= time' => '>= TIME', '<= time' => '<= TIME', 'between time' => 'BETWEEN TIME', 'not between time' => 'NOT BETWEEN TIME', 'notbetween time' => 'NOT BETWEEN TIME']; // 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 数据库连接对象实例 * @param Query $query 数据库查询对象实例 */ public function __construct(Connection $connection, Query $query) { $this->connection = $connection; $this->query = $query; } /** * 获取当前的连接对象实例 * @access public * @return Connection */ public function getConnection() { return $this->connection; } /** * 获取当前的Query对象实例 * @access public * @return Query */ public function getQuery() { return $this->query; } /** * 将SQL语句中的__TABLE_NAME__字符串替换成带前缀的表名(小写) * @access protected * @param string $sql sql语句 * @return string */ protected function parseSqlTable($sql) { return $this->query->parseSqlTable($sql); } /** * 数据分析 * @access protected * @param array $data 数据 * @param array $options 查询参数 * @return array * @throws Exception */ protected function parseData($data, $options) { if (empty($data)) { return []; } // 获取绑定信息 $bind = $this->query->getFieldsBind($options['table']); 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($key, $options, true); if ($val instanceof Expression) { $result[$item] = $val->getValue(); continue; } elseif (is_object($val) && method_exists($val, '__toString')) { // 对象数据写入 $val = $val->__toString(); } if (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 (strtolower($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)) { // 过滤非标量数据 if (0 === strpos($val, ':') && $this->query->isBind(substr($val, 1))) { $result[$item] = $val; } else { $key = str_replace('.', '_', $key); $this->query->bind('data__' . $key, $val, isset($bind[$key]) ? $bind[$key] : PDO::PARAM_STR); $result[$item] = ':data__' . $key; } } } return $result; } /** * 字段名分析 * @access protected * @param string $key * @param array $options * @return string */ protected function parseKey($key, $options = [], $strict = false) { return $key; } /** * value分析 * @access protected * @param mixed $value * @param string $field * @return string|array */ protected function parseValue($value, $field = '') { if (is_string($value)) { $value = strpos($value, ':') === 0 && $this->query->isBind(substr($value, 1)) ? $value : $this->connection->quote($value); } 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 * @param array $options * @return string */ protected function parseField($fields, $options = []) { if ('*' == $fields || empty($fields)) { $fieldsStr = '*'; } elseif (is_array($fields)) { // 支持 'field1'=>'field2' 这样的字段别名定义 $array = []; foreach ($fields as $key => $field) { if ($field instanceof Expression) { $array[] = $field->getValue(); } elseif (!is_numeric($key)) { $array[] = $this->parseKey($key, $options) . ' AS ' . $this->parseKey($field, $options, true); } else { $array[] = $this->parseKey($field, $options); } } $fieldsStr = implode(',', $array); } return $fieldsStr; } /** * table分析 * @access protected * @param mixed $tables * @param array $options * @return string */ protected function parseTable($tables, $options = []) { $item = []; foreach ((array) $tables as $key => $table) { if (!is_numeric($key)) { $key = $this->parseSqlTable($key); $item[] = $this->parseKey($key) . ' ' . (isset($options['alias'][$table]) ? $this->parseKey($options['alias'][$table]) : $this->parseKey($table)); } else { $table = $this->parseSqlTable($table); if (isset($options['alias'][$table])) { $item[] = $this->parseKey($table) . ' ' . $this->parseKey($options['alias'][$table]); } else { $item[] = $this->parseKey($table); } } } return implode(',', $item); } /** * where分析 * @access protected * @param mixed $where 查询条件 * @param array $options 查询参数 * @return string */ protected function parseWhere($where, $options) { $whereStr = $this->buildWhere($where, $options); if (!empty($options['soft_delete'])) { // 附加软删除条件 list($field, $condition) = $options['soft_delete']; $binds = $this->query->getFieldsBind($options['table']); $whereStr = $whereStr ? '( ' . $whereStr . ' ) AND ' : ''; $whereStr = $whereStr . $this->parseWhereItem($field, $condition, '', $options, $binds); } return empty($whereStr) ? '' : ' WHERE ' . $whereStr; } /** * 生成查询条件SQL * @access public * @param mixed $where * @param array $options * @return string */ public function buildWhere($where, $options) { if (empty($where)) { $where = []; } if ($where instanceof Query) { return $this->buildWhere($where->getOptions('where'), $options); } $whereStr = ''; $binds = $this->query->getFieldsBind($options['table']); foreach ($where as $key => $val) { $str = []; foreach ($val as $field => $value) { if ($value instanceof Expression) { $str[] = ' ' . $key . ' ( ' . $value->getValue() . ' )'; } elseif ($value instanceof \Closure) { // 使用闭包查询 $query = new Query($this->connection); call_user_func_array($value, [ & $query]); $whereClause = $this->buildWhere($query->getOptions('where'), $options); if (!empty($whereClause)) { $str[] = ' ' . $key . ' ( ' . $whereClause . ' )'; } } elseif (strpos($field, '|')) { // 不同字段使用相同查询条件(OR) $array = explode('|', $field); $item = []; foreach ($array as $k) { $item[] = $this->parseWhereItem($k, $value, '', $options, $binds); } $str[] = ' ' . $key . ' ( ' . implode(' OR ', $item) . ' )'; } elseif (strpos($field, '&')) { // 不同字段使用相同查询条件(AND) $array = explode('&', $field); $item = []; foreach ($array as $k) { $item[] = $this->parseWhereItem($k, $value, '', $options, $binds); } $str[] = ' ' . $key . ' ( ' . implode(' AND ', $item) . ' )'; } else { // 对字段使用表达式查询 $field = is_string($field) ? $field : ''; $str[] = ' ' . $key . ' ' . $this->parseWhereItem($field, $value, $key, $options, $binds); } } $whereStr .= empty($whereStr) ? substr(implode(' ', $str), strlen($key) + 1) : implode(' ', $str); } return $whereStr; } // where子单元分析 protected function parseWhereItem($field, $val, $rule = '', $options = [], $binds = [], $bindName = null) { // 字段分析 $key = $field ? $this->parseKey($field, $options, 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) { $bindName = 'where_' . str_replace('.', '_', $field) . '_' . $k; $str[] = $this->parseWhereItem($field, $item, $rule, $options, $binds, $bindName); } return '( ' . implode(' ' . $rule . ' ', $str) . ' )'; } // 检测操作符 if (!in_array($exp, $this->exp)) { $exp = strtolower($exp); if (isset($this->exp[$exp])) { $exp = $this->exp[$exp]; } else { throw new Exception('where express error:' . $exp); } } $bindName = $bindName ?: 'where_' . $rule . '_' . str_replace(['.', '-'], '_', $field); if (preg_match('/\W/', $bindName)) { // 处理带非单词字符的字段名 $bindName = md5($bindName); } if ($value instanceof Expression) { } elseif (is_object($value) && method_exists($value, '__toString')) { // 对象数据写入 $value = $value->__toString(); } $bindType = isset($binds[$field]) ? $binds[$field] : PDO::PARAM_STR; if (is_scalar($value) && array_key_exists($field, $binds) && !in_array($exp, ['EXP', 'NOT NULL', 'NULL', 'IN', 'NOT IN', 'BETWEEN', 'NOT BETWEEN']) && strpos($exp, 'TIME') === false) { if (strpos($value, ':') !== 0 || !$this->query->isBind(substr($value, 1))) { if ($this->query->isBind($bindName)) { $bindName .= '_' . str_replace('.', '_', uniqid('', true)); } $this->query->bind($bindName, $value, $bindType); $value = ':' . $bindName; } } $whereStr = ''; if (in_array($exp, ['=', '<>', '>', '>=', '<', '<='])) { // 比较运算 if ($value instanceof \Closure) { $whereStr .= $key . ' ' . $exp . ' ' . $this->parseClosure($value); } else { $whereStr .= $key . ' ' . $exp . ' ' . $this->parseValue($value, $field); } } elseif ('LIKE' == $exp || 'NOT LIKE' == $exp) { // 模糊匹配 if (is_array($value)) { foreach ($value as $item) { $array[] = $key . ' ' . $exp . ' ' . $this->parseValue($item, $field); } $logic = isset($val[2]) ? $val[2] : 'AND'; $whereStr .= '(' . implode($array, ' ' . strtoupper($logic) . ' ') . ')'; } else { $whereStr .= $key . ' ' . $exp . ' ' . $this->parseValue($value, $field); } } elseif ('EXP' == $exp) { // 表达式查询 if ($value instanceof Expression) { $whereStr .= '( ' . $key . ' ' . $value->getValue() . ' )'; } else { throw new Exception('where express error:' . $exp); } } elseif (in_array($exp, ['NOT NULL', 'NULL'])) { // NULL 查询 $whereStr .= $key . ' IS ' . $exp; } elseif (in_array($exp, ['NOT IN', 'IN'])) { // IN 查询 if ($value instanceof \Closure) { $whereStr .= $key . ' ' . $exp . ' ' . $this->parseClosure($value); } else { $value = array_unique(is_array($value) ? $value : explode(',', $value)); if (array_key_exists($field, $binds)) { $bind = []; $array = []; $i = 0; foreach ($value as $v) { $i++; if ($this->query->isBind($bindName . '_in_' . $i)) { $bindKey = $bindName . '_in_' . uniqid() . '_' . $i; } else { $bindKey = $bindName . '_in_' . $i; } $bind[$bindKey] = [$v, $bindType]; $array[] = ':' . $bindKey; } $this->query->bind($bind); $zone = implode(',', $array); } else { $zone = implode(',', $this->parseValue($value, $field)); } $whereStr .= $key . ' ' . $exp . ' (' . (empty($zone) ? "''" : $zone) . ')'; } } elseif (in_array($exp, ['NOT BETWEEN', 'BETWEEN'])) { // BETWEEN 查询 $data = is_array($value) ? $value : explode(',', $value); if (array_key_exists($field, $binds)) { if ($this->query->isBind($bindName . '_between_1')) { $bindKey1 = $bindName . '_between_1' . uniqid(); $bindKey2 = $bindName . '_between_2' . uniqid(); } else { $bindKey1 = $bindName . '_between_1'; $bindKey2 = $bindName . '_between_2'; } $bind = [ $bindKey1 => [$data[0], $bindType], $bindKey2 => [$data[1], $bindType], ]; $this->query->bind($bind); $between = ':' . $bindKey1 . ' AND :' . $bindKey2; } else { $between = $this->parseValue($data[0], $field) . ' AND ' . $this->parseValue($data[1], $field); } $whereStr .= $key . ' ' . $exp . ' ' . $between; } elseif (in_array($exp, ['NOT EXISTS', 'EXISTS'])) { // EXISTS 查询 if ($value instanceof \Closure) { $whereStr .= $exp . ' ' . $this->parseClosure($value); } else { $whereStr .= $exp . ' (' . $value . ')'; } } elseif (in_array($exp, ['< TIME', '> TIME', '<= TIME', '>= TIME'])) { $whereStr .= $key . ' ' . substr($exp, 0, 2) . ' ' . $this->parseDateTime($value, $field, $options, $bindName, $bindType); } elseif (in_array($exp, ['BETWEEN TIME', 'NOT BETWEEN TIME'])) { if (is_string($value)) { $value = explode(',', $value); } $whereStr .= $key . ' ' . substr($exp, 0, -4) . $this->parseDateTime($value[0], $field, $options, $bindName . '_between_1', $bindType) . ' AND ' . $this->parseDateTime($value[1], $field, $options, $bindName . '_between_2', $bindType); } return $whereStr; } // 执行闭包子查询 protected function parseClosure($call, $show = true) { $query = new Query($this->connection); call_user_func_array($call, [ & $query]); return $query->buildSql($show); } /** * 日期时间条件解析 * @access protected * @param string $value * @param string $key * @param array $options * @param string $bindName * @param integer $bindType * @return string */ protected function parseDateTime($value, $key, $options = [], $bindName = null, $bindType = null) { // 获取时间字段类型 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->query->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); } } $bindName = $bindName ?: $key; if ($this->query->isBind($bindName)) { $bindName .= '_' . str_replace('.', '_', uniqid('', true)); } $this->query->bind($bindName, $value, $bindType); return ':' . $bindName; } /** * limit分析 * @access protected * @param mixed $limit * @return string */ protected function parseLimit($limit) { return (!empty($limit) && false === strpos($limit, '(')) ? ' LIMIT ' . $limit . ' ' : ''; } /** * join分析 * @access protected * @param array $join * @param array $options 查询条件 * @return string */ protected function parseJoin($join, & $options = []) { $joinStr = ''; if (!empty($join)) { foreach ($join as $item) { list($table, $type, $on) = $item; if (is_array($table)) { $origin = key($table); if ($origin && $origin != $table[$origin]) { $options['alias'][$origin] = $table[$origin]; } } $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($val1, $options) . '=' . $this->parseKey($val2, $options); } else { $condition[] = $val; } } $table = $this->parseTable($table, $options); $joinStr .= ' ' . $type . ' JOIN ' . $table . ' ON ' . implode(' AND ', $condition); } } return $joinStr; } /** * order分析 * @access protected * @param mixed $order * @param array $options 查询条件 * @return string */ protected function parseOrder($order, $options = []) { if (empty($order)) { return ''; } $array = []; foreach ($order as $key => $val) { if ($val instanceof Expression) { $array[] = $val->getValue(); } elseif ('[rand]' == $val) { $array[] = $this->parseRand(); } else { if (is_numeric($key)) { list($key, $sort) = explode(' ', strpos($val, ' ') ? $val : $val . ' '); } else { $sort = $val; } $sort = strtoupper($sort); $sort = in_array($sort, ['ASC', 'DESC'], true) ? ' ' . $sort : ''; $array[] = $this->parseKey($key, $options, true) . $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 ' . $this->parseKey($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) { if (false !== strpos($comment, '*/')) { $comment = strstr($comment, '*/', true); } 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 ''; } $type = $union['type']; unset($union['type']); foreach ($union as $u) { if ($u instanceof \Closure) { $sql[] = $type . ' ' . $this->parseClosure($u); } elseif (is_string($u)) { $sql[] = $type . ' ( ' . $this->parseSqlTable($u) . ' )'; } } return ' ' . implode(' ', $sql); } /** * index分析,可在操作链中指定需要强制使用的索引 * @access protected * @param mixed $index * @return string */ protected function parseForce($index) { if (empty($index)) { return ''; } return sprintf(" FORCE INDEX ( %s ) ", is_array($index) ? implode(',', $index) : $index); } /** * 设置锁机制 * @access protected * @param bool|string $lock * @return string */ protected function parseLock($lock = false) { if (is_bool($lock)) { return $lock ? ' FOR UPDATE ' : ''; } elseif (is_string($lock)) { return ' ' . trim($lock) . ' '; } } /** * 生成查询SQL * @access public * @param array $options 表达式 * @return string */ public function select($options = []) { $sql = str_replace( ['%TABLE%', '%DISTINCT%', '%FIELD%', '%JOIN%', '%WHERE%', '%GROUP%', '%HAVING%', '%ORDER%', '%LIMIT%', '%UNION%', '%LOCK%', '%COMMENT%', '%FORCE%'], [ $this->parseTable($options['table'], $options), $this->parseDistinct($options['distinct']), $this->parseField($options['field'], $options), $this->parseJoin($options['join'], $options), $this->parseWhere($options['where'], $options), $this->parseGroup($options['group']), $this->parseHaving($options['having']), $this->parseOrder($options['order'], $options), $this->parseLimit($options['limit']), $this->parseUnion($options['union']), $this->parseLock($options['lock']), $this->parseComment($options['comment']), $this->parseForce($options['force']), ], $this->selectSql); return $sql; } /** * 生成insert SQL * @access public * @param array $data 数据 * @param array $options 表达式 * @param bool $replace 是否replace * @return string */ public function insert(array $data, $options = [], $replace = false) { // 分析并处理数据 $data = $this->parseData($data, $options); if (empty($data)) { return 0; } $fields = array_keys($data); $values = array_values($data); $sql = str_replace( ['%INSERT%', '%TABLE%', '%FIELD%', '%DATA%', '%COMMENT%'], [ $replace ? 'REPLACE' : 'INSERT', $this->parseTable($options['table'], $options), implode(' , ', $fields), implode(' , ', $values), $this->parseComment($options['comment']), ], $this->insertSql); return $sql; } /** * 生成insertall SQL * @access public * @param array $dataSet 数据集 * @param array $options 表达式 * @param bool $replace 是否replace * @return string * @throws Exception */ public function insertAll($dataSet, $options = [], $replace = false) { // 获取合法的字段 if ('*' == $options['field']) { $fields = array_keys($this->query->getFieldsType($options['table'])); } else { $fields = $options['field']; } foreach ($dataSet as $data) { foreach ($data as $key => $val) { if (!in_array($key, $fields, true)) { if ($options['strict']) { throw new Exception('fields not exists:[' . $key . ']'); } unset($data[$key]); } elseif (is_null($val)) { $data[$key] = 'NULL'; } elseif (is_scalar($val)) { $data[$key] = $this->parseValue($val, $key); } elseif (is_object($val) && method_exists($val, '__toString')) { // 对象数据写入 $data[$key] = $val->__toString(); } else { // 过滤掉非标量数据 unset($data[$key]); } } $value = array_values($data); $values[] = 'SELECT ' . implode(',', $value); if (!isset($insertFields)) { $insertFields = array_keys($data); } } foreach ($insertFields as $field) { $fields[] = $this->parseKey($field, $options, true); } return str_replace( ['%INSERT%', '%TABLE%', '%FIELD%', '%DATA%', '%COMMENT%'], [ $replace ? 'REPLACE' : 'INSERT', $this->parseTable($options['table'], $options), implode(' , ', $insertFields), implode(' UNION ALL ', $values), $this->parseComment($options['comment']), ], $this->insertAllSql); } /** * 生成select insert SQL * @access public * @param array $fields 数据 * @param string $table 数据表 * @param array $options 表达式 * @return string */ public function selectInsert($fields, $table, $options) { if (is_string($fields)) { $fields = explode(',', $fields); } $fields = array_map([$this, 'parseKey'], $fields); $sql = 'INSERT INTO ' . $this->parseTable($table, $options) . ' (' . implode(',', $fields) . ') ' . $this->select($options); return $sql; } /** * 生成update SQL * @access public * @param array $data 数据 * @param array $options 表达式 * @return string */ public function update($data, $options) { $table = $this->parseTable($options['table'], $options); $data = $this->parseData($data, $options); if (empty($data)) { return ''; } foreach ($data as $key => $val) { $set[] = $key . '=' . $val; } $sql = str_replace( ['%TABLE%', '%SET%', '%JOIN%', '%WHERE%', '%ORDER%', '%LIMIT%', '%LOCK%', '%COMMENT%'], [ $this->parseTable($options['table'], $options), implode(',', $set), $this->parseJoin($options['join'], $options), $this->parseWhere($options['where'], $options), $this->parseOrder($options['order'], $options), $this->parseLimit($options['limit']), $this->parseLock($options['lock']), $this->parseComment($options['comment']), ], $this->updateSql); return $sql; } /** * 生成delete SQL * @access public * @param array $options 表达式 * @return string */ public function delete($options) { $sql = str_replace( ['%TABLE%', '%USING%', '%JOIN%', '%WHERE%', '%ORDER%', '%LIMIT%', '%LOCK%', '%COMMENT%'], [ $this->parseTable($options['table'], $options), !empty($options['using']) ? ' USING ' . $this->parseTable($options['using'], $options) . ' ' : '', $this->parseJoin($options['join'], $options), $this->parseWhere($options['where'], $options), $this->parseOrder($options['order'], $options), $this->parseLimit($options['limit']), $this->parseLock($options['lock']), $this->parseComment($options['comment']), ], $this->deleteSql); return $sql; } }