forked from streamingfast/kvdb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
types.go
88 lines (73 loc) · 1.44 KB
/
types.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
package store
import (
"encoding/hex"
"strconv"
)
const Unlimited = 0
type KV struct {
Key, Value []byte
}
func (kv *KV) Size() int {
if kv == nil {
return 0
}
return len(kv.Key) + len(kv.Value)
}
type Key []byte
func (k Key) String() string {
return hex.EncodeToString(k)
}
// Next returns the next key in byte-order.
//
// Copied from https://github.com/tikv/client-go/blob/master/key/key.go
func (k Key) Next() Key {
// add 0x0 to the end of key
buf := make([]byte, len([]byte(k))+1)
copy(buf, []byte(k))
return buf
}
// PrefixNext returns the next prefix key.
//
// Assume there are keys like:
//
// rowkey1
// rowkey1_column1
// rowkey1_column2
// rowKey2
//
// If we seek 'rowkey1' Next, we will get 'rowkey1_column1'.
// If we seek 'rowkey1' PrefixNext, we will get 'rowkey2'.
//
// Copied from https://github.com/tikv/client-go/blob/master/key/key.go
func (k Key) PrefixNext() Key {
buf := make([]byte, len([]byte(k)))
copy(buf, []byte(k))
var i int
for i = len(k) - 1; i >= 0; i-- {
buf[i]++
if buf[i] != 0 {
break
}
}
if i == -1 {
copy(buf, k)
buf = append(buf, 0)
}
return buf
}
type Limit int
func (l Limit) Reached(count uint64) bool {
return l.Bounded() && count >= uint64(l)
}
func (l Limit) Bounded() bool {
return int(l) > 0
}
func (l Limit) Unbounded() bool {
return int(l) <= 0
}
func (l Limit) String() string {
if l.Unbounded() {
return "unlimited"
}
return strconv.FormatInt(int64(l), 10)
}