dolphin: Implement inserts

internal/endtoend/testdata/insert_values/mysql/query.sql

@@ -0,0 +1,4 @@
+CREATE TABLE foo (a text, b integer);
+
+/* name: InsertValues :exec */
+INSERT INTO foo (a, b) VALUES (?, ?);

internal/endtoend/testdata/insert_values/mysql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "engine": "_dolphin",
+      "path": "go",
+      "name": "querytest",
+      "schema": "query.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/insert_values/postgresql/query.sql

@@ -0,0 +1,4 @@
+CREATE TABLE foo (a text, b integer);
+
+-- name: InsertValues :exec
+INSERT INTO foo (a, b) VALUES ($1, $2);

internal/endtoend/testdata/insert_values/postgresql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "engine": "postgresql",
+      "path": "go",
+      "name": "querytest",
+      "schema": "query.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/engine/dolphin/convert.go

@@ -132,6 +132,17 @@ func (c *cc) convertColumnNameExpr(n *pcast.ColumnNameExpr) *pg.ColumnRef {
 	}
 }
 
+func (c *cc) convertColumnNames(cols []*pcast.ColumnName) *ast.List {
+	list := &ast.List{Items: []ast.Node{}}
+	for i := range cols {
+		name := cols[i].Name.String()
+		list.Items = append(list.Items, &pg.ResTarget{
+			Name: &name,
+		})
+	}
+	return list
+}
+
 func (c *cc) convertDropTableStmt(n *pcast.DropTableStmt) ast.Node {
 	drop := &ast.DropTableStmt{IfExists: n.IfExists}
 	for _, name := range n.Tables {
@@ -156,6 +167,43 @@ func (c *cc) convertFieldList(n *pcast.FieldList) *ast.List {
 	return &ast.List{Items: fields}
 }
 
+func (c *cc) convertInsertStmt(n *pcast.InsertStmt) *pg.InsertStmt {
+	debug.Dump(n)
+
+	rels := c.convertTableRefsClause(n.Table)
+	if len(rels.Items) != 1 {
+		panic("expected one range var")
+	}
+	rel := rels.Items[0]
+	rangeVar, ok := rel.(*pg.RangeVar)
+	if !ok {
+		panic("expected range var")
+	}
+
+	return &pg.InsertStmt{
+		Relation:      rangeVar,
+		Cols:          c.convertColumnNames(n.Columns),
+		ReturningList: &ast.List{Items: []ast.Node{}},
+		SelectStmt: &pg.SelectStmt{
+			FromClause:  &ast.List{Items: []ast.Node{}},
+			TargetList:  &ast.List{Items: []ast.Node{}},
+			ValuesLists: c.convertLists(n.Lists),
+		},
+	}
+}
+
+func (c *cc) convertLists(lists [][]pcast.ExprNode) *ast.List {
+	list := &ast.List{Items: []ast.Node{}}
+	for _, exprs := range lists {
+		inner := &ast.List{Items: []ast.Node{}}
+		for _, expr := range exprs {
+			inner.Items = append(inner.Items, c.convert(expr))
+		}
+		list.Items = append(list.Items, inner)
+	}
+	return list
+}
+
 func (c *cc) convertParamMarkerExpr(n *driver.ParamMarkerExpr) *pg.ParamRef {
 	// Parameter numbers start at one
 	c.paramCount += 1
@@ -256,6 +304,9 @@ func (c *cc) convert(node pcast.Node) ast.Node {
 	case *pcast.ExistsSubqueryExpr:
 		return c.convertExistsSubqueryExpr(n)
 
+	case *pcast.InsertStmt:
+		return c.convertInsertStmt(n)
+
 	case *pcast.SelectStmt:
 		return c.convertSelectStmt(n)