-
Notifications
You must be signed in to change notification settings - Fork 1
/
pages.go
94 lines (74 loc) · 1.64 KB
/
pages.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 pages
import (
"context"
"fmt"
"strconv"
"strings"
"github.com/bloqs-sites/bloqsenjin/pkg/db"
"github.com/google/uuid"
)
type Cursor string
type Paginator[T any] struct {
Identifier Identifier[T]
db db.KVDBer
bitset *Bitset
initialized bool
}
func NewPaginator[T any](id Identifier[T], db db.KVDBer) *Paginator[T] {
return &Paginator[T]{
Identifier: id,
db: db,
bitset: NewBitset(),
initialized: false,
}
}
type Identifier[T any] interface {
GetPrefix() string
GetID(T) int32
}
func GenCursor() Cursor {
return Cursor(uuid.NewString())
}
func validPrefix(prefix string) bool {
return !strings.Contains(prefix, ":")
}
func createPrefix[T any](id Identifier[T], c Cursor) (string, error) {
prefix := id.GetPrefix()
if !validPrefix(prefix) {
return "", fmt.Errorf("invalid prefix `%s` given by %T", prefix, id)
}
return fmt.Sprintf("pagination:%#v:%#v:", prefix, c), nil
}
func (p *Paginator[T]) Init(ctx context.Context, c Cursor) error {
prefix, err := createPrefix[T](p.Identifier, c)
if err != nil {
return err
}
// list, cursor, err := p.db.List(ctx, &prefix, nil)
list, _, err := p.db.List(ctx, &prefix, nil)
if err != nil {
return err
}
kv, err := p.db.Get(ctx, list...)
if err != nil {
return nil
}
for k, v := range kv {
power, err := strconv.ParseUint(strings.Split(k, ":")[3], 10, 0)
if err != nil {
continue
}
used, err := strconv.ParseUint(string(v), 10, 0)
if err != nil {
continue
}
p.bitset.Bits[power] = used
}
p.initialized = true
return nil
}
func (Paginator[T]) Add(i T) {
}
func (p *Paginator[T]) Close() error {
return p.db.Close()
}