feat: add support for values statement with list argument bind variab…

…le. Added new Row_Tuple bind variable type for formatting in vttablet

Signed-off-by: Harshit Gangal <harshit@planetscale.com>

go/vt/sqlparser/ast.go

@@ -1703,9 +1703,10 @@ type InsertRows interface {
  SQLNode
 }
 
-func (*Select) iInsertRows() {}
-func (*Union) iInsertRows() {}
-func (Values) iInsertRows() {}
+func (*Select) iInsertRows() {}
+func (*Union) iInsertRows() {}
+func (Values) iInsertRows() {}
+func (*ValuesStatement) iInsertRows() {}
 
 // OptLike works for create table xxx like xxx
 type OptLike struct {
@@ -3577,17 +3578,10 @@ type Limit struct {
 type Values []ValTuple
 
 type ValuesStatement struct {
- Rows Values
- Order OrderBy
- Limit *Limit
-}
-
-func (s *ValuesStatement) AddOrder(order *Order) {
- s.Order = append(s.Order, order)
-}
-
-func (s *ValuesStatement) SetLimit(limit *Limit) {
- s.Limit = limit
+ Rows Values
+ ListArg ListArg
+ Order OrderBy
+ Limit *Limit
 }
 
 // UpdateExprs represents a list of update expressions.

go/vt/sqlparser/ast_format.go

@@ -123,10 +123,14 @@ func (node *VStream) Format(buf *TrackedBuffer) {
 // Format formats the node.
 func (node *ValuesStatement) Format(buf *TrackedBuffer) {
  buf.WriteString("values ")
- for i, row := range node.Rows {
- buf.astPrintf(node, "row%v", row)
- if i < len(node.Rows)-1 {
- buf.WriteString(", ")
+ if node.ListArg != "" {
+ buf.astPrintf(node, "%v", node.ListArg)
+ } else {
+ for i, row := range node.Rows {
+ buf.astPrintf(node, "row%v", row)
+ if i < len(node.Rows)-1 {
+ buf.WriteString(", ")
+ }
  }
  }
  buf.astPrintf(node, "%v%v",

go/vt/sqlparser/ast_funcs.go

@@ -2839,3 +2839,67 @@ func ExtractAllTables(stmt Statement) []string {
  }, stmt)
  return tables
 }
+
+var _ SelectStatement = (*ValuesStatement)(nil)
+
+func (node *ValuesStatement) GetLock() Lock {
+ return NoLock
+}
+
+func (node *ValuesStatement) SetLock(lock Lock) {
+ if lock != NoLock {
+ panic("cannot set lock on Values statement")
+ }
+}
+
+func (node *ValuesStatement) SetInto(into *SelectInto) {
+ panic("cannot set Into on Values statement")
+}
+
+func (node *ValuesStatement) SetWith(with *With) {
+ panic("cannot set With on Values statement")
+}
+
+func (node *ValuesStatement) MakeDistinct() {
+ panic("cannot set Distinct on Values statement")
+}
+
+func (node *ValuesStatement) GetColumnCount() int {
+ panic("cannot call Column count on Values statement")
+}
+
+func (node *ValuesStatement) GetColumns() SelectExprs {
+ panic("cannot call Columns on Values statement")
+}
+
+func (node *ValuesStatement) SetComments(comments Comments) {
+ panic("cannot set Comments on Values statement")
+}
+
+func (node *ValuesStatement) GetParsedComments() *ParsedComments {
+ panic("cannot call Parsed comments on Values statement")
+}
+
+func (node *ValuesStatement) IsDistinct() bool {
+ return false
+}
+
+func (node *ValuesStatement) GetOrderBy() OrderBy {
+ return node.Order
+}
+
+func (node *ValuesStatement) SetOrderBy(by OrderBy) {
+ node.Order = by
+}
+
+func (node *ValuesStatement) GetLimit() *Limit {
+ return node.Limit
+}
+
+func (node *ValuesStatement) AddOrder(order *Order) {
+ node.Order = append(node.Order, order)
+}
+
+func (node *ValuesStatement) SetLimit(limit *Limit) {
+ node.Limit = limit
+}