-
-
Notifications
You must be signed in to change notification settings - Fork 105
/
page_token.go
74 lines (55 loc) · 1.69 KB
/
page_token.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
// Copyright © 2022 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package keysetpagination
import (
"encoding/base64"
"fmt"
"strings"
)
type PageToken interface {
Parse(string) map[string]string
Encode() string
}
var _ PageToken = new(StringPageToken)
var _ PageToken = new(MapPageToken)
type StringPageToken string
func (s StringPageToken) Parse(idField string) map[string]string {
return map[string]string{idField: string(s)}
}
func (s StringPageToken) Encode() string {
return string(s)
}
func NewStringPageToken(s string) (PageToken, error) {
return StringPageToken(s), nil
}
type MapPageToken map[string]string
func (m MapPageToken) Parse(_ string) map[string]string {
return map[string]string(m)
}
const pageTokenColumnDelim = "/"
func (m MapPageToken) Encode() string {
elems := []string{}
for k, v := range m {
elems = append(elems, fmt.Sprintf("%s=%s", k, v))
}
// For now: use Base64 instead of URL escaping, as the Timestamp format we need to use can contain a `+` sign,
// which represents a space in URLs, so it's not properly encoded by the Go library.
return base64.RawStdEncoding.EncodeToString([]byte(strings.Join(elems, pageTokenColumnDelim)))
}
func NewMapPageToken(s string) (PageToken, error) {
b, err := base64.RawStdEncoding.DecodeString(s)
if err != nil {
return nil, err
}
tokens := strings.Split(string(b), pageTokenColumnDelim)
r := map[string]string{}
for _, p := range tokens {
if columnName, value, found := strings.Cut(p, "="); found {
r[columnName] = value
}
}
return MapPageToken(r), nil
}
var _ PageTokenConstructor = NewMapPageToken
var _ PageTokenConstructor = NewStringPageToken
type PageTokenConstructor func(string) (PageToken, error)