Rows back from INSERT, UPDATE and DELETE: SQLite and CockroachDB
Every file of the SQLite and CockroachDB versions side by side, with the lines that differ marked.
Configuration
sgl.yamldiffers
SQLite
CockroachDB
1
version: 11
version: 12
2
3
databases:3
databases:4
- name: orders4
- name: orders5
dialect: sqlite5
dialect: postgres6
migrations: db/migrations6
migrations: db/migrations7
queries: db/queries7
queries: db/queries8
go:8
go:9
package: store9
package: store10
out: internal/store10
out: internal/store11
driver: database/sql12
migrate:13
dsn: ${env:DATABASE_URL}Migrations
db/migrations/0001_orders.up.sqldiffers
SQLite
CockroachDB
1
CREATE TABLE orders (1
CREATE TABLE orders (2
id INTEGER PRIMARY KEY,2
id BIGINT GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,3
item TEXT NOT NULL,3
item TEXT NOT NULL,4
quantity INTEGER NOT NULL DEFAULT 1,4
quantity BIGINT NOT NULL DEFAULT 1,5
status TEXT NOT NULL DEFAULT 'waiting'5
status TEXT NOT NULL DEFAULT 'waiting'6
);6
);db/migrations/0001_orders.down.sqlidentical
SQLite
CockroachDB
1
DROP TABLE orders;1
DROP TABLE orders;Queries
PlaceOrderdiffers
SQLite · db/queries/orders.sql
CockroachDB · db/queries/orders.sql
1
-- sgl.query(PlaceOrder, one)1
-- sgl.query(PlaceOrder, one)2
INSERT INTO orders (item, quantity) VALUES (?, ?)2
INSERT INTO orders (item, quantity) VALUES ($1, $2)3
RETURNING id, status, quantity * 250 AS price_pence;3
RETURNING id, status, quantity * 250 AS price_pence;ServeWaitingOrdersdiffers
SQLite · db/queries/orders.sql
CockroachDB · db/queries/orders.sql
1
-- sgl.query(ServeWaitingOrders, many)1
-- sgl.query(ServeWaitingOrders, many, row: OrderLine)2
UPDATE orders SET status = 'served' WHERE status = 'waiting'2
UPDATE orders SET status = 'served' WHERE status = 'waiting'3
RETURNING id, item, quantity;3
RETURNING id, item, quantity;CancelOrderdiffers
SQLite · db/queries/orders.sql
CockroachDB · db/queries/orders.sql
1
-- sgl.query(CancelOrder, one)1
-- sgl.query(CancelOrder, one, row: OrderLine)2
DELETE FROM orders WHERE id = ? AND status = 'waiting'2
DELETE FROM orders WHERE id = $1 AND status = 'waiting'3
RETURNING item, quantity;3
RETURNING id, item, quantity;ReopenOrderonly CockroachDB
SQLite · not in this version
CockroachDB · db/queries/orders.sql
1
-- sgl.query(ReopenOrder, one, model: orders)2
UPDATE orders SET status = 'waiting' WHERE id = $13
RETURNING *;What sgl generates
PlaceOrderidentical
SQLite
CockroachDB
1
func (queries *Queries) PlaceOrder(ctx context.Context, params PlaceOrderParams) (PlaceOrderRow, error)1
func (queries *Queries) PlaceOrder(ctx context.Context, params PlaceOrderParams) (PlaceOrderRow, error)2
2
3
type PlaceOrderParams struct {3
type PlaceOrderParams struct {4
Item string4
Item string5
Quantity int645
Quantity int646
}6
}7
7
8
type PlaceOrderRow struct {8
type PlaceOrderRow struct {9
ID int64 `json:"id"`9
ID int64 `json:"id"`10
Status string `json:"status"`10
Status string `json:"status"`11
PricePence int64 `json:"price_pence"`11
PricePence int64 `json:"price_pence"`12
}12
}ServeWaitingOrdersdiffers
SQLite
CockroachDB
1
func (queries *Queries) ServeWaitingOrders(ctx context.Context) ([]ServeWaitingOrdersRow, error)1
func (queries *Queries) ServeWaitingOrders(ctx context.Context) ([]OrderLine, error)2
2
3
type ServeWaitingOrdersRow struct {3
type OrderLine struct {4
ID int64 `json:"id"`4
ID int64 `json:"id"`5
Item string `json:"item"`5
Item string `json:"item"`6
Quantity int64 `json:"quantity"`6
Quantity int64 `json:"quantity"`7
}7
}CancelOrderdiffers
SQLite
CockroachDB
1
func (queries *Queries) CancelOrder(ctx context.Context, id int64) (CancelOrderRow, error)1
func (queries *Queries) CancelOrder(ctx context.Context, id int64) (OrderLine, error)2
2
3
type CancelOrderRow struct {3
type OrderLine struct {4
ID int64 `json:"id"`4
Item string `json:"item"`5
Item string `json:"item"`5
Quantity int64 `json:"quantity"`6
Quantity int64 `json:"quantity"`6
}7
}ReopenOrderonly CockroachDB
SQLite · not in this version
CockroachDB
1
func (queries *Queries) ReopenOrder(ctx context.Context, id int64) (Orders, error)2
3
type Orders struct {4
ID int64 `json:"id"`5
Item string `json:"item"`6
Quantity int64 `json:"quantity"`7
Status string `json:"status"`8
}The program
example.godiffers
SQLite
CockroachDB
1
package main1
package main2
2
3
import (3
import (4
"cmp"4
"cmp"5
"context"5
"context"6
"database/sql"6
"database/sql"7
"errors"7
"errors"8
"fmt"8
"fmt"9
"io"9
"io"10
"slices"10
"slices"11
11
12
"seagull.sh/seagull/examples/sqlite/020_returning/internal/store"12
"seagull.sh/seagull/examples/postgres/020_returning/internal/store"13
)13
)14
14
15
// run places three orders, cancels one, serves the rest and then tries to cancel an order15
// run places three orders, cancels one, serves the rest, tries to cancel an order that16
// that has already been served. Every write uses RETURNING, so the rows it touched come16
// has already been served and reopens one. Every write uses RETURNING, so the rows it17
// back from the same statement.17
// touched come back from the same statement.18
//18
//19
// A query reads its rows into a struct of its own, such as PlaceOrderRow, unless its20
// header names another. ServeWaitingOrders and CancelOrder return the same three columns,21
// so both name the row type OrderLine with row: OrderLine, and the generated code22
// declares that struct once for the two. ReopenOrder returns the whole order with23
// RETURNING *, so model: orders reads it into Orders, the table's model in models.go,24
// rather than a ReopenOrderRow with the same fields.25
//19
// Takes database (*sql.DB) which is a migrated database.26
// Takes database (*sql.DB) which is a migrated database.20
// Takes output (io.Writer) which receives what each write returned.27
// Takes output (io.Writer) which receives what each write returned.21
//28
//22
// Returns error when a query fails.29
// Returns error when a query fails.23
func run(ctx context.Context, database *sql.DB, output io.Writer) error {30
func run(ctx context.Context, database *sql.DB, output io.Writer) error {24
queries := store.New(database)31
queries := store.New(database)25
32
26
// PlaceOrder is an INSERT declared as one: the id SQLite assigned, the default status33
// PlaceOrder is an INSERT declared as one: the id the identity column assigned, the27
// and a computed price all come back in a PlaceOrderRow.34
// default status and a computed price all come back in a PlaceOrderRow.28
orders := []store.PlaceOrderParams{35
orders := []store.PlaceOrderParams{29
{Item: "chips", Quantity: 2},36
{Item: "chips", Quantity: 2},30
{Item: "battered cod", Quantity: 1},37
{Item: "battered cod", Quantity: 1},31
{Item: "mushy peas", Quantity: 1},38
{Item: "mushy peas", Quantity: 1},32
}39
}33
for _, order := range orders {40
for _, order := range orders {34
placed, err := queries.PlaceOrder(ctx, order)41
placed, err := queries.PlaceOrder(ctx, order)35
if err != nil {42
if err != nil {36
return fmt.Errorf("placing an order for %s: %w", order.Item, err)43
return fmt.Errorf("placing an order for %s: %w", order.Item, err)37
}44
}38
fmt.Fprintf(output, "placed #%d %-12s %s, %dp\n", placed.ID, order.Item, placed.Status, placed.PricePence)45
fmt.Fprintf(output, "placed #%d %-12s %s, %dp\n", placed.ID, order.Item, placed.Status, placed.PricePence)39
}46
}40
47
41
// CancelOrder is a DELETE declared as one, so it returns the row it removed.48
// CancelOrder is a DELETE declared as one, so it returns the row it removed.42
cancelled, err := queries.CancelOrder(ctx, 2)49
cancelled, err := queries.CancelOrder(ctx, 2)43
if err != nil {50
if err != nil {44
return fmt.Errorf("cancelling order 2: %w", err)51
return fmt.Errorf("cancelling order 2: %w", err)45
}52
}46
fmt.Fprintf(output, "cancelled #2 %-12s x%d\n", cancelled.Item, cancelled.Quantity)53
fmt.Fprintf(output, "cancelled #2 %-12s x%d\n", cancelled.Item, cancelled.Quantity)47
54
48
// ServeWaitingOrders is an UPDATE declared as many: one row for every order it changed.55
// ServeWaitingOrders is an UPDATE declared as many: one row for every order it changed.49
// SQLite does not promise an order for RETURNING rows, so sort them.56
// PostgreSQL does not promise an order for RETURNING rows, so sort them.50
served, err := queries.ServeWaitingOrders(ctx)57
served, err := queries.ServeWaitingOrders(ctx)51
if err != nil {58
if err != nil {52
return fmt.Errorf("serving orders: %w", err)59
return fmt.Errorf("serving orders: %w", err)53
}60
}54
slices.SortFunc(served, func(a, b store.ServeWaitingOrdersRow) int { return cmp.Compare(a.ID, b.ID) })61
slices.SortFunc(served, func(a, b store.OrderLine) int { return cmp.Compare(a.ID, b.ID) })55
for _, order := range served {62
for _, order := range served {56
fmt.Fprintf(output, "served #%d %-12s x%d\n", order.ID, order.Item, order.Quantity)63
fmt.Fprintf(output, "served #%d %-12s x%d\n", order.ID, order.Item, order.Quantity)57
}64
}58
65
59
// A one query that matches no row returns sql.ErrNoRows, writes included: order 1 has66
// A one query that matches no row returns sql.ErrNoRows, writes included: order 1 has60
// been served, so the DELETE's WHERE clause no longer matches it.67
// been served, so the DELETE's WHERE clause no longer matches it.61
_, err = queries.CancelOrder(ctx, 1)68
_, err = queries.CancelOrder(ctx, 1)62
if errors.Is(err, sql.ErrNoRows) {69
switch {70
case errors.Is(err, sql.ErrNoRows):63
fmt.Fprintln(output, "cannot cancel #1: it is no longer waiting")71
fmt.Fprintln(output, "cannot cancel #1: it is no longer waiting")64
return nil72
case err != nil:73
return fmt.Errorf("cancelling order 1: %w", err)74
default:75
return errors.New("cancelled an order that had already been served")65
}76
}77
78
reopened, err := queries.ReopenOrder(ctx, 3)66
if err != nil {79
if err != nil {67
return fmt.Errorf("cancelling order 1: %w", err)80
return fmt.Errorf("reopening order 3: %w", err)68
}81
}69
return errors.New("cancelled an order that had already been served")82
fmt.Fprintf(output, "reopened #%d %-12s x%d, %s\n", reopened.ID, reopened.Item, reopened.Quantity, reopened.Status)83
return nil70
}84
}