Handle TRUNCATE() function and TRUNCATE keyword. Issue #338. (#339)

* Fix: Tests. * Fix issue #322. * Fix: Handle SubQuery in ref_clause. * Fix: Handle TRUNCATE() function and TRUNCATE keyword. Issue #338. --------- Co-authored-by: Justin Swanhart <greenlion@gmail.com>
greenlion · Mar 12, 2023 · 9ff74d9 · 9ff74d9
1 parent 8339ac3
commit 9ff74d9
Show file tree

Hide file tree

Showing 4 changed files with 51 additions and 11 deletions.
diff --git a/src/PHPSQLParser/processors/SQLProcessor.php b/src/PHPSQLParser/processors/SQLProcessor.php
@@ -242,7 +242,6 @@ public function process($tokens) {
             case 'DELETE':
             case 'ALTER':
             case 'INSERT':
-            case 'TRUNCATE':
             case 'OPTIMIZE':
             case 'GRANT':
             case 'REVOKE':
@@ -264,6 +263,17 @@ public function process($tokens) {
                 $out[$upper][0] = $trim;
                 continue 2;
 
+            case 'TRUNCATE':
+            	if ($prev_category === '') {
+            		// set the category in case these get subclauses in a future version of MySQL
+            		$token_category = $upper;
+            		$out[$upper][0] = $trim;
+            		continue 2;
+            	}
+                // part of the CREATE TABLE statement or a function
+                $out[$prev_category][] = $trim;
+                continue 2;
+
             case 'REPLACE':
             	if ($prev_category === '') {
             		// set the category in case these get subclauses in a future version of MySQL

diff --git a/tests/cases/creator/leftTest.php b/tests/cases/creator/leftTest.php
@@ -56,10 +56,9 @@ public function testLeft() {
         $this->assertSame($expected, $created, 'left joins and table-expression');
 
     }
-
-    /**
-     * @doesNotPerformAssertions
-     */
+/**
+ * @doesNotPerformAssertions
+*/
     public function testLeftIn() {
         $sql = 'SELECT *
             FROM (t1 LEFT JOIN t2 ON t1.a=t2.a)

diff --git a/tests/cases/parser/issue322Test.php b/tests/cases/parser/issue322Test.php
@@ -44,17 +44,18 @@
 
 class issue322Test extends \PHPUnit\Framework\TestCase
 {
-    /**
-     * @doesNotPerformAssertions
-     */
+ /**
+  * @doesNotPerformAssertions
+  */
+
 	public function testIssue322()
 	{
-        $sql = "SELECT IF(createdAt >= CURRENT_DATE(), '1', '0') FROM f_another_table WHERE id in(SELECT id FROM f_table WHERE createdAt > NOW())";
+    $sql = "SELECT IF(createdAt >= CURRENT_DATE(), '1', '0') FROM f_another_table WHERE id in(SELECT id FROM f_table WHERE createdAt > NOW())";
 		$parser = new PHPSQLParser();
 		$parsed = $parser->parse($sql, true);
 		$creator = new PHPSQLCreator();
 		$sql = $creator->create($parsed);
-		//echo $sql;
-    }
+
+  }
 }
 
diff --git a/tests/cases/parser/issue338Test.php b/tests/cases/parser/issue338Test.php
@@ -0,0 +1,30 @@
+<?php
+
+namespace parser;
+
+use PHPSQLParser\PHPSQLParser;
+
+class issue338Test extends \PHPUnit_Framework_TestCase
+{
+	public function testIssue338() {
+		$sql = "SELECT id, date, type as type, libelle as libelle, TRUNCATE(debit, 2) as debit, ROUND(COALESCE(credit, 0) - COALESCE(debit, 0), 2) as solde FROM compte_cp";
+
+		$parser = new PHPSQLParser();
+
+		$parser->parse($sql, true);
+		$parsed = $parser->parsed;
+
+		$this->assertNotFalse($parsed);
+		$this->assertTrue(is_array($parsed));
+    $this->assertTrue(!array_key_exists('TRUNCATE', $parsed));
+
+		$sql = "TRUNCATE TABLE truncate_table";
+		$parser->parse($sql, true);
+		$parsed = $parser->parsed;
+
+		$this->assertNotFalse($parsed);
+		$this->assertTrue(is_array($parsed));
+    $this->assertTrue(array_key_exists('TRUNCATE', $parsed));
+	}
+
+}