-
Notifications
You must be signed in to change notification settings - Fork 0
/
example_rows_iterator_test.go
100 lines (79 loc) · 2.08 KB
/
example_rows_iterator_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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package iterators_test
import (
"fmt"
"log"
"github.com/fredbi/go-patterns/iterators"
"github.com/fredbi/go-patterns/iterators/internal/testdb"
"github.com/jmoiron/sqlx"
)
func ExampleRowsIterator_Next() {
dbName := testdb.UniqueDBName()
// create a DB and fill a table with some data
db, err := testdb.CreateDBAndData(dbName)
if err != nil {
log.Fatalf("could not create test DB: %v", err)
}
// open a cursor selecting over the test data
rows, err := testdb.OpenDBCursor(db)
if err != nil {
log.Fatalf("could not query DB: %v", err)
}
iterator := iterators.NewRowsIterator[*sqlx.Rows, testdb.DummyRow](rows)
defer func() {
_ = iterator.Close()
}()
count := 0
for iterator.Next() {
item, err := iterator.Item()
if err != nil {
fmt.Printf("err: %v\n", err)
}
count++
fmt.Printf("item: %#v\n", item)
}
fmt.Printf("count: %d\n", count)
// Output:
// item: testdb.DummyRow{A:1, B:"x"}
// item: testdb.DummyRow{A:2, B:"y"}
// count: 2
}
func ExampleRowsIterator_Collect() {
dbName := testdb.UniqueDBName()
db, err := testdb.CreateDBAndData(dbName)
if err != nil {
log.Fatalf("could not create test DB: %v", err)
}
rows, err := testdb.OpenDBCursor(db)
if err != nil {
log.Fatalf("could not create test DB: %v", err)
}
iterator := iterators.NewSqlxIterator[testdb.DummyRow](rows)
items, err := iterator.Collect()
if err != nil {
fmt.Printf("err: %v\n", err)
}
fmt.Printf("items: %#v\n", items)
fmt.Printf("count: %d\n", len(items))
// Output:
// items: []testdb.DummyRow{testdb.DummyRow{A:1, B:"x"}, testdb.DummyRow{A:2, B:"y"}}
// count: 2
}
func ExampleRowsIterator_CollectPtr() {
dbName := testdb.UniqueDBName()
db, err := testdb.CreateDBAndData(dbName)
if err != nil {
log.Fatalf("could not create test DB: %v", err)
}
rows, err := testdb.OpenDBCursor(db)
if err != nil {
log.Fatalf("could not create test DB: %v", err)
}
iterator := iterators.NewSqlxIterator[testdb.DummyRow](rows)
items, err := iterator.CollectPtr()
if err != nil {
fmt.Printf("err: %v\n", err)
}
fmt.Printf("count: %d\n", len(items))
// Output:
// count: 2
}