-
Notifications
You must be signed in to change notification settings - Fork 348
/
unmarshal_iterator.go
62 lines (54 loc) · 1.07 KB
/
unmarshal_iterator.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
package committed
import (
"github.com/treeverse/lakefs/pkg/graveler"
)
// UnmarshalIterator wrap value iterator and unmarshal each value
type UnmarshalIterator struct {
it ValueIterator
value *graveler.ValueRecord
err error
}
func NewUnmarshalIterator(it ValueIterator) *UnmarshalIterator {
return &UnmarshalIterator{
it: it,
}
}
func (r *UnmarshalIterator) Next() bool {
if !r.it.Next() {
r.err = r.it.Err()
r.value = nil
return false
}
val := r.it.Value()
// unmarshal value
var v *graveler.Value
if val.Value != nil {
v, r.err = UnmarshalValue(val.Value)
if r.err != nil {
r.value = nil
return false
}
}
r.value = &graveler.ValueRecord{
Key: graveler.Key(val.Key.Copy()),
Value: v,
}
return true
}
func (r *UnmarshalIterator) SeekGE(id graveler.Key) {
r.it.SeekGE(Key(id))
r.value = nil
r.err = r.Err()
}
func (r *UnmarshalIterator) Value() *graveler.ValueRecord {
if r.err != nil {
return nil
}
return r.value
}
func (r *UnmarshalIterator) Err() error {
return r.err
}
func (r *UnmarshalIterator) Close() {
r.it.Close()
}