From 36db8aa4a823ed03af441465d53a84c748a3f165 Mon Sep 17 00:00:00 2001 From: "as2252258@163.com" Date: Sat, 3 Jul 2021 16:22:50 +0800 Subject: [PATCH] modify --- Database/Connection.php | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/Database/Connection.php b/Database/Connection.php index 35fc6b94..87acdaa5 100644 --- a/Database/Connection.php +++ b/Database/Connection.php @@ -267,14 +267,14 @@ class Connection extends Component { $substr = strtoupper(substr($sql, 0, 6)); $sql = match ($substr) { - 'SELECT', 'SHOW F', 'DELETE' => preg_replace('/FROM\s+(\w+)\s+/', 'FROM `' . $dbname . '`.$1 ', $sql), - 'UPDATE' => preg_replace('/UPDATE\s+(\w+)\s+SET/', 'UPDATE `' . $dbname . '`.$1 SET', $sql), - 'INSERT' => preg_replace('/INSERT INTO\s+(\w+)/', 'INSERT INTO `' . $dbname . '`.$1', $sql), - 'TRUNCA' => preg_replace('/TRUNCATE\s+(\w+)\s+/', 'TRUNCATE `' . $dbname . '`.$1', $sql), + 'SELECT', 'SHOW F', 'DELETE' => preg_replace('/FROM\s+([a-zA-Z\-_]+)\s+/', 'FROM `' . $dbname . '`.$1 ', $sql), + 'UPDATE' => preg_replace('/UPDATE\s+([a-zA-Z\-_]+)\s+SET/', 'UPDATE `' . $dbname . '`.$1 SET', $sql), + 'INSERT' => preg_replace('/INSERT INTO\s+([a-zA-Z\-_]+)/', 'INSERT INTO `' . $dbname . '`.$1', $sql), + 'TRUNCA' => preg_replace('/TRUNCATE\s+([a-zA-Z\-_]+)\s+/', 'TRUNCATE `' . $dbname . '`.$1', $sql), default => throw new Exception('database error') }; - $sql = preg_replace('/INNER JOIN\s+(\w+)\s/', 'INNER JOIN `'.$dbname.'`.$1 ', $sql); + $sql = preg_replace('/INNER JOIN\s+([a-zA-Z\-_]+)\s/', 'INNER JOIN `'.$dbname.'`.$1 ', $sql); $command = new Command(['db' => $this, 'sql' => $sql]); return $command->bindValues($attributes);