-
-
Notifications
You must be signed in to change notification settings - Fork 401
/
example_array_test.go
75 lines (62 loc) · 1.71 KB
/
example_array_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package pg_test
import (
"fmt"
"github.com/go-pg/pg/v10"
"github.com/go-pg/pg/v10/types"
)
func ExampleDB_Model_postgresArrayStructTag() {
type Item struct {
Id int64
Emails []string `pg:",array"` // marshalled as PostgreSQL array
Numbers [][]int `pg:",array"` // marshalled as PostgreSQL array
}
_, err := pgdb.Exec(`CREATE TEMP TABLE items (id serial, emails text[], numbers int[][])`)
panicIf(err)
defer pgdb.Exec("DROP TABLE items")
item1 := &Item{
Id: 1,
Emails: []string{"[email protected]", "[email protected]"},
Numbers: [][]int{{1, 2}, {3, 4}},
}
_, err = pgdb.Model(item1).Insert()
panicIf(err)
item := new(Item)
err = pgdb.Model(item).Where("id = ?", 1).Select()
panicIf(err)
fmt.Println(item)
// Output: &{1 [[email protected] [email protected]] [[1 2] [3 4]]}
}
func ExampleArray() {
src := []string{"[email protected]", "[email protected]"}
var dst []string
_, err := pgdb.QueryOne(pg.Scan(pg.Array(&dst)), `SELECT ?`, pg.Array(src))
panicIf(err)
fmt.Println(dst)
// Output: [[email protected] [email protected]]
}
type MyArrayValueScanner struct {
sum int
}
var _ types.ArrayValueScanner = (*MyArrayValueScanner)(nil)
func (s *MyArrayValueScanner) BeforeScanArrayValue(rd types.Reader, n int) error {
return nil
}
func (s *MyArrayValueScanner) ScanArrayValue(rd types.Reader, n int) error {
num, err := types.ScanInt(rd, n)
if err != nil {
return err
}
s.sum += num
return nil
}
func (s *MyArrayValueScanner) AfterScanArrayValue() error {
return nil
}
func ExampleDB_arrayValueScanner() {
var dst MyArrayValueScanner
_, err := pgdb.QueryOne(pg.Scan(pg.Array(&dst)),
`SELECT array_agg(id) from generate_series(0, 10) AS id`)
panicIf(err)
fmt.Println(dst.sum)
// Output: 55
}