Initial support for DELETE

Author: kyleconroy

parser.go

@@ -181,6 +181,16 @@ func parseFuncs(s *postgres.Schema, r *Result, source string, tree pg.ParsetreeL
 			r.Queries[i].Table = tab
 			r.Queries[i].Args = parseArgs(tab, refs)
 			r.Queries[i].SQL = strings.Replace(rawSQL, "*", strings.Join(c, ", "), 1)
+		case nodes.DeleteStmt:
+			t := tableName(n)
+
+			rawSQL, _ := pluckQuery(source, raw)
+			refs := extractArgs(n)
+
+			tab := getTable(s, t)
+			r.Queries[i].Table = tab
+			r.Queries[i].Args = parseArgs(tab, refs)
+			r.Queries[i].SQL = rawSQL
 		default:
 			log.Printf("%T\n", n)
 		}
@@ -205,6 +215,8 @@ func findRefs(r []paramRef, parent, n nodes.Node) []paramRef {
 	switch n := n.(type) {
 	case nodes.RawStmt:
 		r = findRefs(r, n.Stmt, nil)
+	case nodes.DeleteStmt:
+		r = findRefs(r, n.WhereClause, nil)
 	case nodes.SelectStmt:
 		r = findRefs(r, n.WhereClause, nil)
 		r = findRefs(r, n.LimitCount, nil)
@@ -273,12 +285,17 @@ func columnNames(s *postgres.Schema, table string) []string {
 	return cols
 }
 
-func tableName(n nodes.SelectStmt) string {
-	for _, item := range n.FromClause.Items {
-		switch i := item.(type) {
-		case nodes.RangeVar:
-			return *i.Relname
+func tableName(n nodes.Node) string {
+	switch n := n.(type) {
+	case nodes.SelectStmt:
+		for _, item := range n.FromClause.Items {
+			switch i := item.(type) {
+			case nodes.RangeVar:
+				return *i.Relname
+			}
 		}
+	case nodes.DeleteStmt:
+		return *n.Relation.Relname
 	}
 	return ""
 }
@@ -391,6 +408,22 @@ func (q *Queries) {{.MethodName}}(ctx context.Context, {{range .Args}}{{.Name}}
 }
 {{end}}
 
+{{if eq .Type ":exec"}}
+func (q *Queries) {{.MethodName}}(ctx context.Context, {{range .Args}}{{.Name}} {{.Type}},{{end}}) error {
+	var err error
+	switch {
+	case q.{{.StmtName}} != nil && q.tx != nil:
+		_, err = q.tx.StmtContext(ctx, q.{{.StmtName}}).ExecContext(ctx, {{range .Args}}{{.Name}},{{end}})
+	case q.{{.StmtName}} != nil:
+		_, err = q.{{.StmtName}}.ExecContext(ctx, {{range .Args}}{{.Name}},{{end}})
+	default:
+		_, err = q.db.ExecContext(ctx, {{.QueryName}}, {{range .Args}}{{.Name}},{{end}})
+	}
+	return err
+}
+{{end}}
+
+
 {{end}}
 `
 

testdata/ondeck/db.go

@@ -32,6 +32,9 @@ func New(db dbtx) *Queries {
 func Prepare(ctx context.Context, db dbtx) (*Queries, error) {
 	q := Queries{db: db}
 	var err error
+	if q.deleteVenue, err = db.PrepareContext(ctx, deleteVenue); err != nil {
+		return nil, err
+	}
 	if q.getCity, err = db.PrepareContext(ctx, getCity); err != nil {
 		return nil, err
 	}
@@ -47,20 +50,40 @@ func Prepare(ctx context.Context, db dbtx) (*Queries, error) {
 type Queries struct {
 	db dbtx
 
-	tx         *sql.Tx
-	getCity    *sql.Stmt
-	listCities *sql.Stmt
-	listVenues *sql.Stmt
+	tx          *sql.Tx
+	deleteVenue *sql.Stmt
+	getCity     *sql.Stmt
+	listCities  *sql.Stmt
+	listVenues  *sql.Stmt
 }
 
 func (q *Queries) WithTx(tx *sql.Tx) *Queries {
 	return &Queries{
-		tx:         tx,
-		db:         tx,
-		getCity:    q.getCity,
-		listCities: q.listCities,
-		listVenues: q.listVenues,
+		tx:          tx,
+		db:          tx,
+		deleteVenue: q.deleteVenue,
+		getCity:     q.getCity,
+		listCities:  q.listCities,
+		listVenues:  q.listVenues,
+	}
+}
+
+const deleteVenue = `-- name: DeleteVenue :exec
+DELETE FROM venue
+WHERE slug = $1
+`
+
+func (q *Queries) DeleteVenue(ctx context.Context, slug string) error {
+	var err error
+	switch {
+	case q.deleteVenue != nil && q.tx != nil:
+		_, err = q.tx.StmtContext(ctx, q.deleteVenue).ExecContext(ctx, slug)
+	case q.deleteVenue != nil:
+		_, err = q.deleteVenue.ExecContext(ctx, slug)
+	default:
+		_, err = q.db.ExecContext(ctx, deleteVenue, slug)
 	}
+	return err
 }
 
 const getCity = `-- name: GetCity :one

testdata/ondeck/query/queries.sql

@@ -13,3 +13,7 @@ SELECT *
 FROM venue
 WHERE city = $1
 ORDER BY name;
+
+-- name: DeleteVenue :exec
+DELETE FROM venue
+WHERE slug = $1;