insert-returning

Summary:

- Support for `INSERT RETURNING`.
- Support for `UPDATE RETURNING`.
- Support for `REPLACE RETURNING`.
- Support for `DELETE RETURNING`.

Author: general-kroll-4-life

README.md

@@ -10,17 +10,15 @@ There are elements of the original work that are not required, but may take some
 ## Rebuilding parser
 
 
-```
+```bash
 make -C go/vt/sqlparser
 ```
 
 
 After changes to the ast:
 
-```
-cd go/vt/sqlparser
-
-go run ./visitorgen/main -input=ast.go -output=rewriter.go
+```bash
+cicd/build_scripts/01_ast_rebuild.sh
 ```
 
 

cicd/build_scripts/01_ast_rebuild.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+
+CURDIR="$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
+
+REPOSITORY_ROOT="$(realpath "${CURDIR}/../..")"
+
+cd "${REPOSITORY_ROOT}/go/vt/sqlparser"
+
+go run ./visitorgen/main -input=ast.go -output=rewriter.go

go/vt/sqlparser/ast.go

@@ -131,40 +131,43 @@ type (
 	// of the implications the deletion part may have on vindexes.
 	// If you add fields here, consider adding them to calls to validateUnshardedRoute.
 	Insert struct {
-		Action     string
-		Comments   Comments
-		Ignore     string
-		Table      TableName
-		Partitions Partitions
-		Columns    Columns
-		Rows       InsertRows
-		OnDup      OnDup
+		Action      string
+		Comments    Comments
+		Ignore      string
+		Table       TableName
+		Partitions  Partitions
+		Columns     Columns
+		Rows        InsertRows
+		OnDup       OnDup
+		SelectExprs SelectExprs
 	}
 
 	// Update represents an UPDATE statement.
 	// If you add fields here, consider adding them to calls to validateUnshardedRoute.
 	Update struct {
-		Action     string
-		Comments   Comments
-		Ignore     string
-		TableExprs TableExprs
-		Exprs      UpdateExprs
-		From       TableExprs
-		Where      *Where
-		OrderBy    OrderBy
-		Limit      *Limit
+		Action      string
+		Comments    Comments
+		Ignore      string
+		TableExprs  TableExprs
+		Exprs       UpdateExprs
+		From        TableExprs
+		Where       *Where
+		OrderBy     OrderBy
+		Limit       *Limit
+		SelectExprs SelectExprs
 	}
 
 	// Delete represents a DELETE statement.
 	// If you add fields here, consider adding them to calls to validateUnshardedRoute.
 	Delete struct {
-		Comments   Comments
-		Targets    TableNames
-		TableExprs TableExprs
-		Partitions Partitions
-		Where      *Where
-		OrderBy    OrderBy
-		Limit      *Limit
+		Comments    Comments
+		Targets     TableNames
+		TableExprs  TableExprs
+		Partitions  Partitions
+		Where       *Where
+		OrderBy     OrderBy
+		Limit       *Limit
+		SelectExprs SelectExprs
 	}
 
 	// Set represents a SET statement.