Add support for TRUNCATE as DDL statement

vitessio · Oct 23, 2017 · 083454b · 083454b
1 parent cdb860e
commit 083454b
Show file tree

Hide file tree

Showing 5 changed files with 1,997 additions and 1,955 deletions.
diff --git a/data/test/tabletserver/exec_cases.txt b/data/test/tabletserver/exec_cases.txt
@@ -1195,8 +1195,8 @@
 # truncate
 "truncate a"
 {
-  "PlanID": "OTHER_ADMIN",
-  "TableName": ""
+  "PlanID": "DDL",
+  "TableName": "a"
 }
 
 # table not found select

diff --git a/go/vt/sqlparser/ast.go b/go/vt/sqlparser/ast.go
@@ -534,8 +534,8 @@ func (node *Set) WalkSubtree(visit Visit) error {
 	)
 }
 
-// DDL represents a CREATE, ALTER, DROP or RENAME statement.
-// Table is set for AlterStr, DropStr, RenameStr
+// DDL represents a CREATE, ALTER, DROP, RENAME or TRUNCATE statement.
+// Table is set for AlterStr, DropStr, RenameStr, TruncateStr
 // NewName is set for AlterStr, CreateStr, RenameStr.
 type DDL struct {
 	Action        string
@@ -548,10 +548,11 @@ type DDL struct {
 
 // DDL strings.
 const (
-	CreateStr = "create"
-	AlterStr  = "alter"
-	DropStr   = "drop"
-	RenameStr = "rename"
+	CreateStr   = "create"
+	AlterStr    = "alter"
+	DropStr     = "drop"
+	RenameStr   = "rename"
+	TruncateStr = "truncate"
 )
 
 // Format formats the node.

diff --git a/go/vt/sqlparser/parse_test.go b/go/vt/sqlparser/parse_test.go
@@ -826,9 +826,12 @@ var (
 	}, {
 		input:  "explain foobar",
 		output: "otherread",
+	}, {
+		input:  "truncate table foo",
+		output: "truncate table foo",
 	}, {
 		input:  "truncate foo",
-		output: "otheradmin",
+		output: "truncate table foo",
 	}, {
 		input:  "repair foo",
 		output: "otheradmin",