Arrays and lists as typed slices: ClickHouse and CockroachDB
Every file of the ClickHouse and CockroachDB versions side by side, with the lines that differ marked.
Configuration
sgl.yamldiffers
ClickHouse
CockroachDB
1
version: 11
version: 12
2
3
databases:3
databases:4
- name: quay4
- name: voyages5
dialect: clickhouse5
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/sqlMigrations
db/migrations/0001_hauls.up.sqldiffers
ClickHouse
CockroachDB · db/migrations/0001_voyages.up.sql
1
CREATE TABLE hauls (1
CREATE TABLE voyages (2
id UInt64,2
id BIGINT GENERATED ALWAYS AS IDENTITY PRIMARY KEY,3
boat String,3
boat TEXT NOT NULL,4
species Array(String),4
ports TEXT[] NOT NULL,5
weights_kg Array(UInt32)5
catch_kg BIGINT[] NOT NULL,6
) ENGINE = MergeTree() ORDER BY id;6
sightings TEXT[]7
);8
-- sgl.column(voyages.ports, non_null_elements: true)db/migrations/0001_hauls.down.sqldiffers
ClickHouse
CockroachDB · db/migrations/0001_voyages.down.sql
1
DROP TABLE hauls;1
DROP TABLE voyages;Queries
LandHaulonly ClickHouse
ClickHouse · db/queries/hauls.sql
CockroachDB · not in this version
1
-- sgl.query(LandHaul, exec)2
INSERT INTO hauls (id, boat, species, weights_kg)3
VALUES ({id:UInt64}, {boat:String}, {species:Array(String)}, {weights_kg:Array(UInt32)});GetHaulonly ClickHouse
ClickHouse · db/queries/hauls.sql
CockroachDB · not in this version
1
-- sgl.query(GetHaul, one)2
SELECT id, boat, species, weights_kg FROM hauls WHERE id = {id:UInt64};HaulsWithonly ClickHouse
ClickHouse · db/queries/hauls.sql
CockroachDB · not in this version
1
-- sgl.query(HaulsWith, many)2
SELECT id, boat, species, weights_kg3
FROM hauls4
WHERE has(species, {fish:String})5
ORDER BY id;LogVoyageonly CockroachDB
ClickHouse · not in this version
CockroachDB · db/queries/voyages.sql
1
-- sgl.query(LogVoyage, exec)2
INSERT INTO voyages (boat, ports, catch_kg, sightings) VALUES ($1, $2, $3, $4);ListVoyagesonly CockroachDB
ClickHouse · not in this version
CockroachDB · db/queries/voyages.sql
1
-- sgl.query(ListVoyages, many)2
SELECT id, boat, ports, catch_kg, sightings FROM voyages ORDER BY id;VoyagesCallingAtonly CockroachDB
ClickHouse · not in this version
CockroachDB · db/queries/voyages.sql
1
-- sgl.query(VoyagesCallingAt, many)2
-- $1 as sgl.param(port)3
SELECT boat, ports[1] AS first_port, cardinality(ports) AS stops4
FROM voyages5
WHERE $1::text = ANY(ports)6
ORDER BY id;VoyagesPerBoatonly CockroachDB
ClickHouse · not in this version
CockroachDB · db/queries/voyages.sql
1
-- sgl.query(VoyagesPerBoat, many)2
SELECT boat, array_agg(id ORDER BY id) AS voyages3
FROM voyages4
GROUP BY boat5
ORDER BY boat;What sgl generates
LandHaulonly ClickHouse
ClickHouse
CockroachDB · not in this version
1
func (queries *Queries) LandHaul(ctx context.Context, params LandHaulParams) error2
3
type LandHaulParams struct {4
ID uint645
Boat string6
Species []string7
WeightsKg []uint328
}GetHaulonly ClickHouse
ClickHouse
CockroachDB · not in this version
1
func (queries *Queries) GetHaul(ctx context.Context, id uint64) (GetHaulRow, error)2
3
type GetHaulRow struct {4
ID uint64 `json:"id"`5
Boat string `json:"boat"`6
Species []string `json:"species"`7
WeightsKg []uint32 `json:"weights_kg"`8
}HaulsWithonly ClickHouse
ClickHouse
CockroachDB · not in this version
1
func (queries *Queries) HaulsWith(ctx context.Context, fish string) ([]HaulsWithRow, error)2
3
type HaulsWithRow struct {4
ID uint64 `json:"id"`5
Boat string `json:"boat"`6
Species []string `json:"species"`7
WeightsKg []uint32 `json:"weights_kg"`8
}LogVoyageonly CockroachDB
ClickHouse · not in this version
CockroachDB
1
func (queries *Queries) LogVoyage(ctx context.Context, params LogVoyageParams) error2
3
type LogVoyageParams struct {4
Boat string5
Ports []string6
CatchKg []*int647
Sightings []*string8
}ListVoyagesonly CockroachDB
ClickHouse · not in this version
CockroachDB
1
func (queries *Queries) ListVoyages(ctx context.Context) ([]ListVoyagesRow, error)2
3
type ListVoyagesRow struct {4
ID int64 `json:"id"`5
Boat string `json:"boat"`6
Ports []string `json:"ports"`7
CatchKg []*int64 `json:"catch_kg"`8
Sightings []*string `json:"sightings"`9
}VoyagesCallingAtonly CockroachDB
ClickHouse · not in this version
CockroachDB
1
func (queries *Queries) VoyagesCallingAt(ctx context.Context, port string) ([]VoyagesCallingAtRow, error)2
3
type VoyagesCallingAtRow struct {4
Boat string `json:"boat"`5
FirstPort *string `json:"first_port"`6
Stops int32 `json:"stops"`7
}VoyagesPerBoatonly CockroachDB
ClickHouse · not in this version
CockroachDB
1
func (queries *Queries) VoyagesPerBoat(ctx context.Context) ([]VoyagesPerBoatRow, error)2
3
type VoyagesPerBoatRow struct {4
Boat string `json:"boat"`5
Voyages []int64 `json:"voyages"`6
}The program
example.godiffers
ClickHouse
CockroachDB
1
package main1
package main2
2
3
import (3
import (4
"context"4
"context"5
"database/sql"5
"database/sql"6
"fmt"6
"fmt"7
"io"7
"io"8
8
9
"seagull.sh/seagull/examples/clickhouse/008_arrays/internal/store"9
"seagull.sh/seagull/examples/postgres/008_arrays/internal/store"10
)10
)11
11
12
// run lands four hauls of fish and reads them back. An Array(T) column arrives as a Go12
// run logs three fishing voyages and reads them back. An array column becomes a typed Go13
// slice of T's type, so species, an Array(String), is []string and weights_kg, an13
// slice: TEXT[] gives []*string and BIGINT[] gives []*int64, and array_agg gives a slice14
// Array(UInt32), is []uint32, in result rows and in parameters alike: LandHaul takes the14
// of its argument's type, here []int64: the id column it collects is NOT NULL, so the15
// slices and binds each as one {species:Array(String)} value.15
// array can hold no NULL and its elements are plain values. A slice parameter is sent as16
// a PostgreSQL array, and a nil slice is sent as NULL.16
//17
//17
// HaulsWith filters with has(species, {fish:String}), which looks inside the array on the18
// The elements are pointers because PostgreSQL can hold NULL in any array, whatever the18
// server. An Array column cannot be Nullable in ClickHouse, so an empty haul is an empty19
// column's type says, so a NULL element reads as nil, which elements prints as NULL.19
// array, never NULL.20
// Parameters take the same pointers, so LogVoyage takes []*int64 for catch_kg and21
// []*string for sightings, and a nil element writes NULL; values builds them from plain22
// values. When the data never holds a NULL element, non_null_elements: true opts out, in23
// an sgl.column line of a query or migration or as go.types.non_null_elements in24
// sgl.yaml: the elements then read and bind as plain values, and a NULL one fails the25
// read. The migration declares it for ports, as a voyage never calls at a NULL port, so26
// Ports is a plain []string in the model, the row and the parameters alike.20
//27
//28
// database/sql cannot scan a PostgreSQL array, so the generated SQL wraps each array29
// column in to_json() and decodes the JSON into the slice; open internal/store to see it.30
// A NULL array therefore scans to a nil slice, and an empty array to an empty one.31
//32
// VoyagesCallingAt searches inside an array with $1::text = ANY(ports). The cast gives33
// the parameter its type, for Seagull, which would otherwise leave it as any, and for34
// CockroachDB, which cannot work it out from ANY. ports[1] reads one element, and is35
// *string because a missing element is NULL.36
//21
// Takes database (*sql.DB) which is a migrated database.37
// Takes database (*sql.DB) which is a migrated database.22
// Takes output (io.Writer) which receives the hauls.38
// Takes output (io.Writer) which receives the voyages and the searches.23
//39
//24
// Returns error when a query fails.40
// Returns error when a query fails.25
func run(ctx context.Context, database *sql.DB, output io.Writer) error {41
func run(ctx context.Context, database *sql.DB, output io.Writer) error {26
queries := store.New(database)42
queries := store.New(database)27
43
28
hauls := []store.LandHaulParams{44
voyages := []store.LogVoyageParams{29
{ID: 1, Boat: "Kittiwake", Species: []string{"mackerel", "pollock"}, WeightsKg: []uint32{120, 45}},45
{Boat: "Kittiwake", Ports: []string{"Newlyn", "St Ives", "Mousehole"}, CatchKg: values[int64](120, 85), Sightings: values("dolphin", "basking shark")},30
{ID: 2, Boat: "Salty Chip", Species: []string{"crab", "lobster"}, WeightsKg: []uint32{30, 12}},46
{Boat: "Salty Chip", Ports: []string{"Penzance", "Mousehole"}, CatchKg: values[int64](40), Sightings: nil},31
{ID: 3, Boat: "Gull Wing", Species: []string{"mackerel"}, WeightsKg: []uint32{80}},47
{Boat: "Kittiwake", Ports: []string{"Newlyn", "Lamorna"}, CatchKg: values[int64](60, 30, 15), Sightings: []*string{}},32
{ID: 4, Boat: "Driftwood", Species: []string{}, WeightsKg: []uint32{}},33
}48
}34
for _, haul := range hauls {49
for _, voyage := range voyages {35
if err := queries.LandHaul(ctx, haul); err != nil {50
if err := queries.LogVoyage(ctx, voyage); err != nil {36
return fmt.Errorf("landing haul %d: %w", haul.ID, err)51
return fmt.Errorf("logging %s's voyage: %w", voyage.Boat, err)37
}52
}38
}53
}39
54
40
first, err := queries.GetHaul(ctx, 1)55
logged, err := queries.ListVoyages(ctx)41
if err != nil {56
if err != nil {42
return fmt.Errorf("getting haul 1: %w", err)57
return fmt.Errorf("listing voyages: %w", err)43
}58
}44
fmt.Fprintf(output, "haul 1: species %T %v, weights_kg %T %v\n", first.Species, first.Species, first.WeightsKg, first.WeightsKg)59
for _, voyage := range logged {60
sightings := elements(voyage.Sightings)61
if voyage.Sightings == nil {62
sightings = "NULL"63
}64
fmt.Fprintf(output, "voyage %d %-10s ports %s, catch %s kg, sightings %s\n", voyage.ID, voyage.Boat, voyage.Ports, elements(voyage.CatchKg), sightings)65
}45
66
46
mackerel, err := queries.HaulsWith(ctx, "mackerel")67
calling, err := queries.VoyagesCallingAt(ctx, "Mousehole")47
if err != nil {68
if err != nil {48
return fmt.Errorf("finding mackerel: %w", err)69
return fmt.Errorf("finding voyages calling at Mousehole: %w", err)49
}70
}50
fmt.Fprintln(output, "hauls with mackerel:")71
fmt.Fprintln(output, "\ncalling at Mousehole:")51
for _, haul := range mackerel {72
for _, voyage := range calling {52
fmt.Fprintf(output, " #%d %-10s %v %v kg\n", haul.ID, haul.Boat, haul.Species, haul.WeightsKg)73
fmt.Fprintf(output, " %-10s from %s, %d stops\n", voyage.Boat, *voyage.FirstPort, voyage.Stops)53
}74
}54
75
55
empty, err := queries.GetHaul(ctx, 4)76
perBoat, err := queries.VoyagesPerBoat(ctx)56
if err != nil {77
if err != nil {57
return fmt.Errorf("getting haul 4: %w", err)78
return fmt.Errorf("grouping voyages by boat: %w", err)58
}79
}59
fmt.Fprintf(output, "haul 4 landed %d species, an empty array: %v\n", len(empty.Species), empty.Species)80
fmt.Fprintln(output, "\nvoyages per boat:")81
for _, boat := range perBoat {82
fmt.Fprintf(output, " %-10s %v\n", boat.Boat, boat.Voyages)83
}60
return nil84
return nil61
}85
}86
87
// elements returns an array's elements as fmt prints a slice, with NULL for each element88
// the database holds NULL in.89
//90
// Takes array ([]*T) which holds the elements, nil for a NULL one.91
//92
// Returns string which is the elements in brackets, separated by spaces.93
func elements[T any](array []*T) string {94
texts := make([]string, 0, len(array))95
for _, element := range array {96
if element == nil {97
texts = append(texts, "NULL")98
continue99
}100
texts = append(texts, fmt.Sprint(*element))101
}102
return fmt.Sprint(texts)103
}104
105
// values returns pointers to copies of plain values, as an array parameter whose elements106
// may be NULL takes them.107
//108
// Takes items (...T) which are the elements.109
//110
// Returns []*T which points at the copies, none of them nil.111
func values[T any](items ...T) []*T {112
pointed := make([]*T, len(items))113
for index := range items {114
pointed[index] = &items[index]115
}116
return pointed117
}