go/vt/sqlparser: Parse KILL {,CONNECTION,QUERY} statements.

go/vt/sqlparser/ast.go

@@ -5542,6 +5542,30 @@ func (node *UnlockTables) walkSubtree(visit Visit) error {
 	return nil
 }
 
+type Kill struct {
+	Connection bool
+	ConnID     Expr
+}
+
+func (k *Kill) Format(buf *TrackedBuffer) {
+	buf.WriteString("kill ")
+	if k.Connection {
+		buf.WriteString("connection ")
+	} else {
+		buf.WriteString("query ")
+	}
+	buf.Myprintf("%v", k.ConnID)
+}
+
+func (*Kill) iStatement() {}
+
+func (k *Kill) walkSubtree(visit Visit) error {
+	if k == nil {
+		return nil
+	}
+	return Walk(visit, k.ConnID)
+}
+
 func compliantName(in string) string {
 	var buf strings.Builder
 	for i, c := range in {

go/vt/sqlparser/parse_test.go

@@ -2227,8 +2227,13 @@ var (
 		}, {
 			input:  "SHOW GRANTS FOR UserName USING role1, role2",
 			output: "show grants for `UserName`@`%` using `role1`@`%`, `role2`@`%`",
+		}, {
+			input:  "kill query 123",
+			output: "kill query 123",
+		}, {
+			input:  "kill connection 423",
+			output: "kill connection 423",
 		},
-
 	}
 	// Any tests that contain multiple statements within the body (such as BEGIN/END blocks) should go here.
 	// validSQL is used by TestParseNextValid, which expects a semicolon to mean the end of a full statement.