Merge pull request #640 from ted-johnson/create_truncate

fix parsing and rendering of Truncate

Author: wumpz

src/main/java/net/sf/jsqlparser/statement/truncate/Truncate.java

@@ -31,6 +31,7 @@
 public class Truncate implements Statement {
 
     private Table table;
+    boolean cascade;  // to support TRUNCATE TABLE ... CASCADE
 
     @Override
     public void accept(StatementVisitor statementVisitor) {
@@ -45,8 +46,19 @@ public void setTable(Table table) {
         this.table = table;
     }
 
+    public boolean getCascade(){
+        return cascade;
+    }
+
+    public void setCascade(boolean c){
+        cascade=c;
+    }
+
     @Override
     public String toString() {
+        if(cascade){
+            return "TRUNCATE TABLE " + table+" CASCADE";
+        }
         return "TRUNCATE TABLE " + table;
     }
 }

src/main/java/net/sf/jsqlparser/util/deparser/StatementDeParser.java

@@ -144,6 +144,11 @@ public void visit(Select select) {
 
     @Override
     public void visit(Truncate truncate) {
+        buffer.append("TRUNCATE TABLE ");
+        buffer.append(truncate.getTable());
+        if(truncate.getCascade()){
+            buffer.append(" CASCADE");
+        }
     }
 
     @Override

src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt

@@ -3449,7 +3449,7 @@ Truncate Truncate():
 }
 {
     <K_TRUNCATE> <K_TABLE>
-    table=Table() { truncate.setTable(table); }
+	table=Table() { truncate.setTable(table); truncate.setCascade(false); } [ <K_CASCADE> {truncate.setCascade(true);} ]
     {
         return truncate;
     }

src/test/java/net/sf/jsqlparser/statement/truncate/TruncateTest.java

@@ -2,6 +2,9 @@
 
 import java.io.StringReader;
 
+import static net.sf.jsqlparser.test.TestUtils.*;
+import net.sf.jsqlparser.*;
+
 import net.sf.jsqlparser.parser.CCJSqlParserManager;
 import static org.junit.Assert.assertEquals;
 import org.junit.Test;
@@ -23,5 +26,20 @@ public void testTruncate() throws Exception {
         truncate = (Truncate) parserManager.parse(new StringReader(statement));
         assertEquals("mytab", truncate.getTable().getName());
         assertEquals(toStringStatement.toUpperCase(), truncate.toString().toUpperCase());
+
+        statement = "TRUNCATE TABLE mytab CASCADE";
+        truncate = (Truncate) parserManager.parse(new StringReader(statement));
+        assertEquals(statement, truncate.toString());
+    }
+
+    @Test
+    public void testTruncateDeparse() throws JSQLParserException {
+        assertSqlCanBeParsedAndDeparsed("TRUNCATE TABLE foo");
     }
+
+    @Test
+    public void testTruncateCascadeDeparse() throws JSQLParserException {
+        assertSqlCanBeParsedAndDeparsed("TRUNCATE TABLE foo CASCADE");
+    }
+
 }