Enums as generated Go types, and your own types with go_type: DuckDB and ClickHouse
Every file of the DuckDB and ClickHouse versions side by side, with the lines that differ marked.
Configuration
sgl.yamldiffers
DuckDB
ClickHouse
1
version: 11
version: 12
2
3
databases:3
databases:4
- name: menu4
- name: harbour5
dialect: duckdb5
dialect: clickhouse6
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
types:12
overrides:13
- sql: item_status14
go: seagull.sh/seagull/examples/duckdb/007_enums/internal/menu.StatusMigrations
db/migrations/0001_menu.up.sqldiffers
DuckDB
ClickHouse · db/migrations/0001_boats.up.sql
1
CREATE TYPE section AS ENUM ('fish', 'sides', 'puddings');1
CREATE TABLE boats (2
2
id UInt64,3
CREATE TYPE item_status AS ENUM ('available', 'sold_out');3
name String,4
4
status Enum8('moored' = 1, 'fishing' = 2, 'in_dock' = 3)5
CREATE SEQUENCE menu_item_ids;5
) ENGINE = MergeTree() ORDER BY id;6
7
CREATE TABLE menu_items (8
id BIGINT PRIMARY KEY DEFAULT nextval('menu_item_ids'),9
name VARCHAR NOT NULL UNIQUE,10
section section NOT NULL,11
status item_status NOT NULL DEFAULT 'available'12
);db/migrations/0001_menu.down.sqldiffers
DuckDB
ClickHouse · db/migrations/0001_boats.down.sql
1
DROP TABLE menu_items;1
DROP TABLE boats;2
DROP SEQUENCE menu_item_ids;3
DROP TYPE item_status;4
DROP TYPE section;Queries
AddItemonly DuckDB
DuckDB · db/queries/menu.sql
ClickHouse · not in this version
1
-- sgl.query(AddItem, exec)2
INSERT INTO menu_items (name, section) VALUES ($1, $2);SetStatusonly DuckDB
DuckDB · db/queries/menu.sql
ClickHouse · not in this version
1
-- sgl.query(SetStatus, execrows)2
UPDATE menu_items SET status = $1 WHERE name = $2;ListMenuonly DuckDB
DuckDB · db/queries/menu.sql
ClickHouse · not in this version
1
-- sgl.query(ListMenu, many)2
SELECT name, section, status FROM menu_items ORDER BY section, name;ItemsInSectiononly DuckDB
DuckDB · db/queries/menu.sql
ClickHouse · not in this version
1
-- sgl.query(ItemsInSection, many)2
-- sgl.column(section, go_type: "seagull.sh/seagull/examples/duckdb/007_enums/internal/menu.Section")3
SELECT name, section, status FROM menu_items WHERE section = $1 ORDER BY name;AddBoatonly ClickHouse
DuckDB · not in this version
ClickHouse · db/queries/boats.sql
1
-- sgl.query(AddBoat, exec)2
INSERT INTO boats (id, name, status) VALUES ({id:UInt64}, {name:String}, {status:String});ListBoatsonly ClickHouse
DuckDB · not in this version
ClickHouse · db/queries/boats.sql
1
-- sgl.query(ListBoats, many)2
SELECT id, name, status FROM boats ORDER BY id;BoatsNotInonly ClickHouse
DuckDB · not in this version
ClickHouse · db/queries/boats.sql
1
-- sgl.query(BoatsNotIn, many)2
-- sgl.column(status, go_type: "seagull.sh/seagull/examples/clickhouse/007_enums/internal/domain.Status")3
SELECT id, name, status FROM boats WHERE status != {status:String} ORDER BY id;What sgl generates
AddItemonly DuckDB
DuckDB
ClickHouse · not in this version
1
func (queries *Queries) AddItem(ctx context.Context, params AddItemParams) error2
3
type AddItemParams struct {4
Name string5
Section Section6
}7
8
type Section stringSetStatusonly DuckDB
DuckDB
ClickHouse · not in this version
1
func (queries *Queries) SetStatus(ctx context.Context, params SetStatusParams) (int64, error)2
3
type SetStatusParams struct {4
Status menu.Status5
Name string6
}ListMenuonly DuckDB
DuckDB
ClickHouse · not in this version
1
func (queries *Queries) ListMenu(ctx context.Context) ([]ListMenuRow, error)2
3
type ListMenuRow struct {4
Name string `json:"name"`5
Section Section `json:"section"`6
Status menu.Status `json:"status"`7
}8
9
type Section stringItemsInSectiononly DuckDB
DuckDB
ClickHouse · not in this version
1
func (queries *Queries) ItemsInSection(ctx context.Context, section Section) ([]ItemsInSectionRow, error)2
3
type Section string4
5
type ItemsInSectionRow struct {6
Name string `json:"name"`7
Section menu.Section `json:"section"`8
Status menu.Status `json:"status"`9
}AddBoatonly ClickHouse
DuckDB · not in this version
ClickHouse
1
func (queries *Queries) AddBoat(ctx context.Context, params AddBoatParams) error2
3
type AddBoatParams struct {4
ID uint645
Name string6
Status string7
}ListBoatsonly ClickHouse
DuckDB · not in this version
ClickHouse
1
func (queries *Queries) ListBoats(ctx context.Context) ([]ListBoatsRow, error)2
3
type ListBoatsRow struct {4
ID uint64 `json:"id"`5
Name string `json:"name"`6
Status BoatsStatus `json:"status"`7
}8
9
type BoatsStatus stringBoatsNotInonly ClickHouse
DuckDB · not in this version
ClickHouse
1
func (queries *Queries) BoatsNotIn(ctx context.Context, status string) ([]BoatsNotInRow, error)2
3
type BoatsNotInRow struct {4
ID uint64 `json:"id"`5
Name string `json:"name"`6
Status domain.Status `json:"status"`7
}The program
example.godiffers
DuckDB
ClickHouse
1
package main1
package main2
2
3
import (3
import (4
"context"4
"context"5
"database/sql"5
"database/sql"6
"errors"7
"fmt"6
"fmt"8
"io"7
"io"8
"strings"9
9
10
"seagull.sh/seagull/examples/duckdb/007_enums/internal/menu"10
"seagull.sh/seagull/examples/clickhouse/007_enums/internal/domain"11
"seagull.sh/seagull/examples/duckdb/007_enums/internal/store"11
"seagull.sh/seagull/examples/clickhouse/007_enums/internal/store"12
)12
)13
13
14
// run fills a chip shop menu whose section and status columns are DuckDB enums, marks an14
// run berths four boats and lists them twice. boats.status is an Enum8, and Seagull15
// item sold out and reads the menu back. Each enum becomes a named Go type in the15
// declares a named Go type for it in models.go, named after the table and column that16
// generated package unless an override names a type of your own:16
// hold it: store.BoatsStatus, with a constant for each label, such as17
// store.BoatsStatusMoored, and a Valid method. ListBoats returns Status as a18
// store.BoatsStatus. The status parameters are written {status:String}, so they are plain19
// strings, and the server converts the label to the Enum8.17
//20
//18
// - section has no override, so models.go declares store.Section, with a constant for21
// BoatsNotIn adds sgl.column(status, go_type: "…/internal/domain.Status"), so its rows19
// each value, such as store.SectionFish, and a Valid method; ListMenuRow.Section,22
// carry the harbour's own domain.Status type, with its constants and methods. The20
// AddItemParams.Section and ItemsInSection's parameter are all a store.Section;23
// override changes only that query's result rows: parameters stay string, so the example21
// - sgl.yaml's types.overrides names the enum type item_status, so every column of that24
// converts with string(domain.StatusInDock) on the way in. A types.overrides entry in22
// enum gives menu.Status, as in ListMenuRow.Status, and so does every parameter25
// sgl.yaml would change every query's rows the same way.23
// compared with or assigned to one, as in SetStatusParams.Status; the generated24
// package then declares no type for item_status;25
// - sgl.column(section, go_type: ...) in ItemsInSection gives that one query's returned26
// Section the type menu.Section; the parameter it compares with stays a27
// store.Section.28
//26
//29
// An enum sorts in the order its values are declared, not alphabetically, so ORDER BY27
// ClickHouse checks every label against the Enum8, so a status that is not one of its30
// section lists fish, then sides, then puddings. Valid reports whether a value is one the28
// labels is refused by the server, not stored.31
// enum declares, so the shop can check a section before it binds it. DuckDB checks each32
// value against the enum as it is stored as well, so a section the enum does not list is33
// refused; DuckDB keeps an enum as a small integer, and its error names that type.34
//29
//35
// Takes database (*sql.DB) which is a migrated database.30
// Takes database (*sql.DB) which is a migrated database.36
// Takes output (io.Writer) which receives the menu, the fish section and the refusal.31
// Takes output (io.Writer) which receives both listings and the refusal.37
//32
//38
// Returns error when a query fails other than as the example expects.33
// Returns error when a query fails or a bad status is accepted.39
func run(ctx context.Context, database *sql.DB, output io.Writer) error {34
func run(ctx context.Context, database *sql.DB, output io.Writer) error {40
queries := store.New(database)35
queries := store.New(database)41
36
42
items := []store.AddItemParams{37
boats := []store.AddBoatParams{43
{Name: "cod", Section: store.SectionFish},38
{ID: 1, Name: "Kittiwake", Status: string(domain.StatusMoored)},44
{Name: "haddock", Section: store.SectionFish},39
{ID: 2, Name: "Salty Chip", Status: string(domain.StatusFishing)},45
{Name: "chips", Section: store.SectionSides},40
{ID: 3, Name: "Gull Wing", Status: string(domain.StatusInDock)},46
{Name: "mushy peas", Section: store.SectionSides},41
{ID: 4, Name: "Driftwood", Status: string(domain.StatusFishing)},47
{Name: "sticky toffee pudding", Section: store.SectionPuddings},48
}42
}49
for _, item := range items {43
for _, boat := range boats {50
if err := queries.AddItem(ctx, item); err != nil {44
if err := queries.AddBoat(ctx, boat); err != nil {51
return fmt.Errorf("adding %s: %w", item.Name, err)45
return fmt.Errorf("adding %s: %w", boat.Name, err)52
}46
}53
}47
}54
if _, err := queries.SetStatus(ctx, store.SetStatusParams{Status: menu.SoldOut, Name: "haddock"}); err != nil {55
return fmt.Errorf("selling out of haddock: %w", err)56
}57
48
58
listed, err := queries.ListMenu(ctx)49
listed, err := queries.ListBoats(ctx)59
if err != nil {50
if err != nil {60
return fmt.Errorf("listing the menu: %w", err)51
return fmt.Errorf("listing boats: %w", err)61
}52
}62
var menuRow store.ListMenuRow53
fmt.Fprintln(output, "all boats, status as store.BoatsStatus:")63
fmt.Fprintf(output, "ListMenu: section %T, status %T\n", menuRow.Section, menuRow.Status)54
for _, boat := range listed {64
for _, item := range listed {55
fmt.Fprintf(output, " %d %-11s %s\n", boat.ID, boat.Name, boat.Status)65
fmt.Fprintf(output, " %-9s %s%s\n", item.Section, item.Name, soldOut(item.Status))66
}56
}67
57
68
fish, err := queries.ItemsInSection(ctx, store.SectionFish)58
afloat, err := queries.BoatsNotIn(ctx, string(domain.StatusInDock))69
if err != nil {59
if err != nil {70
return fmt.Errorf("listing the fish: %w", err)60
return fmt.Errorf("listing boats not in dock: %w", err)71
}61
}72
var sectionRow store.ItemsInSectionRow62
fmt.Fprintln(output, "not in dock, status as domain.Status:")73
fmt.Fprintf(output, "ItemsInSection: section %T, status %T\n", sectionRow.Section, sectionRow.Status)63
for _, boat := range afloat {74
for _, item := range fish {64
fmt.Fprintf(output, " %d %-11s %-8s at sea: %t\n", boat.ID, boat.Name, boat.Status, boat.Status.AtSea())75
fmt.Fprintf(output, " %-9s %s%s\n", item.Section, item.Name, soldOut(item.Status))76
}65
}77
66
78
drinks := store.Section("drinks")67
sunk := store.AddBoatParams{ID: 5, Name: "Old Barnacle", Status: "sunk"}79
fmt.Fprintf(output, "section drinks is valid: %t\n", drinks.Valid())68
err = queries.AddBoat(ctx, sunk)80
err = queries.AddItem(ctx, store.AddItemParams{Name: "lemonade", Section: drinks})81
if err == nil {69
if err == nil {82
return errors.New("added an item to a section the enum does not list")70
return fmt.Errorf("adding status %q: got no error, want the server's refusal", sunk.Status)83
}71
}84
fmt.Fprintf(output, "refused section drinks: %v\n", err)72
if !strings.Contains(err.Error(), "Unknown element 'sunk' for enum") {85
return nil73
return fmt.Errorf("adding status %q: got %w, want the server's refusal", sunk.Status, err)86
}87
88
// soldOut returns a note for an item that has sold out, comparing against the typed89
// constant rather than a string.90
//91
// Takes status (menu.Status) which is the item's status.92
//93
// Returns string which is the note, or empty when the item is available.94
func soldOut(status menu.Status) string {95
if status == menu.SoldOut {96
return " (sold out)"97
}74
}98
return ""75
fmt.Fprintf(output, "status %q refused: not one of the Enum8 labels\n", sunk.Status)76
return nil99
}77
}