forked from apache/cassandra-gocql-driver
-
Notifications
You must be signed in to change notification settings - Fork 0
/
example_paging_test.go
70 lines (67 loc) · 1.99 KB
/
example_paging_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
package gocql_test
import (
"fmt"
"github.com/gocql/gocql"
"log"
)
// Example_paging demonstrates how to manually fetch pages and use page state.
//
// See also package documentation about paging.
func Example_paging() {
/* The example assumes the following CQL was used to setup the keyspace:
create keyspace example with replication = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 };
create table example.itoa(id int, description text, PRIMARY KEY(id));
insert into example.itoa (id, description) values (1, 'one');
insert into example.itoa (id, description) values (2, 'two');
insert into example.itoa (id, description) values (3, 'three');
insert into example.itoa (id, description) values (4, 'four');
insert into example.itoa (id, description) values (5, 'five');
insert into example.itoa (id, description) values (6, 'six');
*/
cluster := gocql.NewCluster("localhost:9042")
cluster.Keyspace = "example"
cluster.ProtoVersion = 4
session, err := cluster.CreateSession()
if err != nil {
log.Fatal(err)
}
defer session.Close()
var pageState []byte
for {
// We use PageSize(2) for the sake of example, use larger values in production (default is 5000) for performance
// reasons.
iter := session.Query(`SELECT id, description FROM itoa`).PageSize(2).PageState(pageState).Iter()
nextPageState := iter.PageState()
scanner := iter.Scanner()
for scanner.Next() {
var (
id int
description string
)
err = scanner.Scan(&id, &description)
if err != nil {
log.Fatal(err)
}
fmt.Println(id, description)
}
err = scanner.Err()
if err != nil {
log.Fatal(err)
}
fmt.Printf("next page state: %+v\n", nextPageState)
if len(nextPageState) == 0 {
break
}
pageState = nextPageState
}
// 5 five
// 1 one
// next page state: [4 0 0 0 1 0 240 127 255 255 253 0]
// 2 two
// 4 four
// next page state: [4 0 0 0 4 0 240 127 255 255 251 0]
// 6 six
// 3 three
// next page state: [4 0 0 0 3 0 240 127 255 255 249 0]
// next page state: []
}