|
@@ -6,11 +6,13 @@
|
|
|
// +----------------------------------------------------------------------
|
|
|
// | Author: liu21st <liu21st@gmail.com>
|
|
|
// +----------------------------------------------------------------------
|
|
|
+declare (strict_types = 1);
|
|
|
|
|
|
namespace think\db\builder;
|
|
|
|
|
|
-use think\db\BaseQuery;
|
|
|
use think\db\Builder;
|
|
|
+use think\db\Query;
|
|
|
+use think\db\exception\DbException as Exception;
|
|
|
use think\db\Raw;
|
|
|
|
|
|
/**
|
|
@@ -18,136 +20,98 @@ use think\db\Raw;
|
|
|
*/
|
|
|
class Dm extends Builder
|
|
|
{
|
|
|
- protected $insertAllSql = '%INSERT% INTO %TABLE% (%FIELD%) VALUES %DATA% %COMMENT%';
|
|
|
- protected $updateSql = 'UPDATE %TABLE% %JOIN% SET %SET% %WHERE% %ORDER%%LIMIT% %LOCK%%COMMENT%';
|
|
|
+// protected $selectSql = 'SELECT * FROM (SELECT thinkphp.*, rownum AS numrow FROM (SELECT %DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%) thinkphp ) %LIMIT%%COMMENT%';
|
|
|
|
|
|
/**
|
|
|
- * 生成insertall SQL
|
|
|
- * @access public
|
|
|
- * @param array $dataSet 数据集
|
|
|
- * @param array $options 表达式
|
|
|
- * @param bool $replace 是否replace
|
|
|
+ * limit分析
|
|
|
+ * @access protected
|
|
|
+ * @param Query $query 查询对象
|
|
|
+ * @param mixed $limit
|
|
|
* @return string
|
|
|
- * @throws Exception
|
|
|
*/
|
|
|
-// public function insertAll($dataSet, $options = [], $replace = false):string
|
|
|
-// {
|
|
|
-// // 获取合法的字段
|
|
|
-// if ('*' == $options['field']) {
|
|
|
-// $fields = array_keys($this->getFieldsType($options['table']));
|
|
|
-// } else {
|
|
|
-// $fields = $options['field'];
|
|
|
-// }
|
|
|
-// foreach ($dataSet as $data) {
|
|
|
-// foreach ($data as $key => $val) {
|
|
|
-// if (!in_array($key, $fields, true) && !in_array( strtoupper($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[] = '( ' . implode(',', $value) . ' )';
|
|
|
-//
|
|
|
-// if (!isset($insertFields)) {
|
|
|
-// $insertFields = array_map([$this, 'parseKey'], array_keys($data));
|
|
|
-// }
|
|
|
-// }
|
|
|
-//
|
|
|
-// return str_replace(
|
|
|
-// ['%INSERT%', '%TABLE%', '%FIELD%', '%DATA%', '%COMMENT%'],
|
|
|
-// [
|
|
|
-// $replace ? 'REPLACE' : 'INSERT',
|
|
|
-// $this->parseTable($options['table'], $options),
|
|
|
-// implode(' , ', $insertFields),
|
|
|
-// implode(' , ', $values),
|
|
|
-// $this->parseComment($options['comment']),
|
|
|
-// ], $this->insertAllSql);
|
|
|
-// }
|
|
|
+ protected function parseLimit(Query $query, string $limit): string
|
|
|
+ {
|
|
|
+ return (!empty($limit) && false === strpos($limit, '(')) ? ' LIMIT ' . $limit . ' ' : '';
|
|
|
+
|
|
|
+ }
|
|
|
|
|
|
/**
|
|
|
- * 字段和表名处理
|
|
|
+ * 设置锁机制
|
|
|
* @access protected
|
|
|
- * @param mixed $key
|
|
|
- * @param array $options
|
|
|
+ * @param Query $query 查询对象
|
|
|
+ * @param bool|false $lock
|
|
|
* @return string
|
|
|
*/
|
|
|
- public function parseKey(BaseQuery $query, string|int|Raw $key, bool $strict = false): string
|
|
|
+ protected function parseLock(Query $query, $lock = false): string
|
|
|
+ {
|
|
|
+ if (!$lock) {
|
|
|
+ return '';
|
|
|
+ }
|
|
|
+
|
|
|
+ return ' FOR UPDATE NOWAIT ';
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 字段和表名处理
|
|
|
+ * @access public
|
|
|
+ * @param Query $query 查询对象
|
|
|
+ * @param string $key
|
|
|
+ * @param bool $strict
|
|
|
+ * @return string
|
|
|
+ * @throws Exception
|
|
|
+ */
|
|
|
+ public function parseKey(Query $query, $key, bool $strict = false): string
|
|
|
{
|
|
|
- if (is_numeric($key)) {
|
|
|
- return $key;
|
|
|
- } elseif ($key instanceof Expression) {
|
|
|
- return $key->getValue();
|
|
|
+ if (is_int($key)) {
|
|
|
+ return (string) $key;
|
|
|
+ } elseif ($key instanceof Raw) {
|
|
|
+ return $this->parseRaw($query, $key);
|
|
|
}
|
|
|
|
|
|
$key = trim($key);
|
|
|
- if (strpos($key, '$.') && false === strpos($key, '(')) {
|
|
|
+
|
|
|
+ if (strpos($key, '->') && false === strpos($key, '(')) {
|
|
|
// JSON字段支持
|
|
|
- list($field, $name) = explode('$.', $key);
|
|
|
- return 'json_extract(' . $field . ', \'$.' . $name . '\')';
|
|
|
- } elseif (strpos($key, '.') && !preg_match('/[,\'\"\(\)`\s]/', $key)) {
|
|
|
- list($table, $key) = explode('.', $key, 2);
|
|
|
+ [$field, $name] = explode($key, '->');
|
|
|
+ $key = $field . '."' . $name . '"';
|
|
|
+ } elseif (strpos($key, '.') && !preg_match('/[,\'\"\(\)\[\s]/', $key)) {
|
|
|
+ [$table, $key] = explode('.', $key, 2);
|
|
|
+
|
|
|
+ $alias = $query->getOptions('alias');
|
|
|
+
|
|
|
if ('__TABLE__' == $table) {
|
|
|
- $table =$query->getTable();
|
|
|
+ $table = $query->getOptions('table');
|
|
|
+ $table = is_array($table) ? array_shift($table) : $table;
|
|
|
}
|
|
|
- if (isset($options['alias'][$table])) {
|
|
|
- $table = $options['alias'][$table];
|
|
|
+
|
|
|
+ if (isset($alias[$table])) {
|
|
|
+ $table = $alias[$table];
|
|
|
}
|
|
|
}
|
|
|
|
|
|
if ($strict && !preg_match('/^[\w\.\*]+$/', $key)) {
|
|
|
throw new Exception('not support data:' . $key);
|
|
|
}
|
|
|
- if ('*' != $key && ($strict || !preg_match('/[,\'\"\*\(\)`.\s]/', $key))) {
|
|
|
- $key = '`' . $key . '`';
|
|
|
+
|
|
|
+ if ('*' != $key && !preg_match('/[,\'\"\*\(\)\[.\s]/', $key)) {
|
|
|
+ $key = '"' . $key . '"';
|
|
|
}
|
|
|
+
|
|
|
if (isset($table)) {
|
|
|
- if (strpos($table, '.')) {
|
|
|
- $table = str_replace('.', '`.`', $table);
|
|
|
- }
|
|
|
- $key = '`' . $table . '`.' . $key;
|
|
|
+ $key = '"' . $table . '".' . $key;
|
|
|
}
|
|
|
+
|
|
|
return $key;
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* 随机排序
|
|
|
* @access protected
|
|
|
+ * @param Query $query 查询对象
|
|
|
* @return string
|
|
|
*/
|
|
|
- protected function parseRand(BaseQuery $query): string
|
|
|
- {
|
|
|
- return 'rand()';
|
|
|
- }
|
|
|
- /**
|
|
|
- * Partition 分析.
|
|
|
- *
|
|
|
- * @param BaseQuery $query 查询对象
|
|
|
- * @param string|array $partition 分区
|
|
|
- *
|
|
|
- * @return string
|
|
|
- */
|
|
|
- protected function parsePartition(BaseQuery $query, $partition): string
|
|
|
+ protected function parseRand(Query $query): string
|
|
|
{
|
|
|
- if ('' == $partition) {
|
|
|
- return '';
|
|
|
- }
|
|
|
-
|
|
|
- if (is_string($partition)) {
|
|
|
- $partition = explode(',', $partition);
|
|
|
- }
|
|
|
-
|
|
|
- return ' PARTITION (' . implode(' , ', $partition) . ') ';
|
|
|
+ return 'DBMS_RANDOM.value';
|
|
|
}
|
|
|
}
|