Merge pull request #158 from devenbansod/add_set_querytype

Add recognizer for SET Statements

Author: nijel

src/Utils/Query.php

@@ -26,6 +26,7 @@
 use PhpMyAdmin\SqlParser\Statements\RepairStatement;
 use PhpMyAdmin\SqlParser\Statements\ReplaceStatement;
 use PhpMyAdmin\SqlParser\Statements\SelectStatement;
+use PhpMyAdmin\SqlParser\Statements\SetStatement;
 use PhpMyAdmin\SqlParser\Statements\ShowStatement;
 use PhpMyAdmin\SqlParser\Statements\TruncateStatement;
 use PhpMyAdmin\SqlParser\Statements\UpdateStatement;
@@ -358,6 +359,8 @@ public static function getFlags($statement, $all = false)
         } elseif ($statement instanceof UpdateStatement) {
             $flags['querytype'] = 'UPDATE';
             $flags['is_affected'] = true;
+        } elseif ($statement instanceof SetStatement) {
+            $flags['querytype'] = 'SET';
         }
 
         if (($statement instanceof SelectStatement)

tests/Utils/QueryTest.php

@@ -263,6 +263,12 @@ public function testGetFlagsProvider()
                     'select_from' => true,
                 ),
             ),
+            array(
+                'SET NAMES \'latin\'',
+                array(
+                    'querytype' => 'SET',
+                ),
+            ),
         );
     }