-
Notifications
You must be signed in to change notification settings - Fork 0
/
pagination.go
94 lines (71 loc) · 1.48 KB
/
pagination.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
package client
import (
"fmt"
"net/url"
"strconv"
"github.com/google/go-querystring/query"
)
const defaultPerPage = 25
type PageToken struct {
// Page number for paginated result sets.
number int `url:"-"`
// Number of items on paginated result sets.
size int `url:"-"`
}
var _ query.Encoder = (*PageToken)(nil)
func (t *PageToken) values() (int, int) {
number := 1
size := defaultPerPage
if t != nil {
// Zero is an invalid page number.
if t.number > 0 {
number = t.number
}
if t.size > 0 {
size = t.size
}
}
return number, size
}
func (t *PageToken) EncodeValues(_ string, v *url.Values) error {
number, size := t.values()
if number > 0 {
v.Set("page", strconv.FormatUint(uint64(number), 10))
}
// The page size is always set in the URL to never rely on the server's
// default.
v.Set("page_size", strconv.FormatUint(uint64(size), 10))
return nil
}
func pageTokenFromURL(raw string) (*PageToken, error) {
if raw == "" {
return nil, nil
}
parsed, err := url.Parse(raw)
if err != nil {
return nil, err
}
query := parsed.Query()
var t PageToken
for _, i := range []struct {
dest *int
name string
}{
{&t.number, "page"},
{&t.size, "page_size"},
} {
raw := query.Get(i.name)
if raw == "" {
continue
}
num, err := strconv.ParseInt(raw, 10, 64)
if err != nil {
return nil, fmt.Errorf("parsing %q value %q: %w", i.name, raw, err)
}
*i.dest = int(num)
}
return &t, nil
}
type ListOptions struct {
Page *PageToken
}