Add support for set = ? for parser

Fixes: #492
phpmyadmin · Aug 17, 2023 · 13938da · 13938da
1 parent d70e65d
commit 13938da
Show file tree

Hide file tree

Showing 5 changed files with 472 additions and 12 deletions.
diff --git a/src/Components/SetOperation.php b/src/Components/SetOperation.php
@@ -111,19 +111,27 @@ public static function parse(Parser $parser, TokensList $list, array $options =
                     $commaLastSeenAt = $token;
                 }
             } elseif ($state === 1) {
-                $tmp = Expression::parse(
-                    $parser,
-                    $list,
-                    ['breakOnAlias' => true]
-                );
-                if ($tmp === null) {
-                    $parser->error('Missing expression.', $token);
-                    break;
+                if ($token->flags & Token::FLAG_SYMBOL_PARAMETER) {
+                    $expr->column = trim($expr->column);
+                    $expr->value = $token->value;
+                    $ret[] = $expr;
+                } else {
+                    $tmp = Expression::parse(
+                        $parser,
+                        $list,
+                        ['breakOnAlias' => true]
+                    );
+
+                    if ($tmp === null) {
+                        $parser->error('Missing expression.', $token);
+                        break;
+                    }
+
+                    $expr->column = trim($expr->column);
+                    $expr->value = $tmp->expr;
+                    $ret[] = $expr;
                 }
 
-                $expr->column = trim($expr->column);
-                $expr->value = $tmp->expr;
-                $ret[] = $expr;
                 $expr = new static();
                 $state = 0;
                 $commaLastSeenAt = null;

diff --git a/tests/Parser/UpdateStatementTest.php b/tests/Parser/UpdateStatementTest.php
@@ -28,6 +28,7 @@ public static function updateProvider(): array
             ['parser/parseUpdate5'],
             ['parser/parseUpdate6'],
             ['parser/parseUpdate7'],
+            ['parser/parseUpdate8'],
             ['parser/parseUpdateErr'],
         ];
     }

diff --git a/tests/TestCase.php b/tests/TestCase.php
@@ -93,7 +93,7 @@ public function getErrorsAsArray($obj): array
      */
     public function getData(string $name): array
     {
-        $serializedData = file_get_contents('tests/data/' . $name . '.out');
+        $serializedData = file_get_contents(__DIR__ . '/data/' . $name . '.out');
         $this->assertIsString($serializedData);
 
         $serializer = new CustomJsonSerializer();

diff --git a/tests/data/parser/parseUpdate8.in b/tests/data/parser/parseUpdate8.in
@@ -0,0 +1,11 @@
+UPDATE
+    users
+SET
+    username = ?,
+    id=155;
+
+UPDATE
+    users
+SET
+    username = :user_name,
+    id=155;