-
-
Notifications
You must be signed in to change notification settings - Fork 33
/
object_iterator.go
103 lines (87 loc) · 1.98 KB
/
object_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
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
95
96
97
98
99
100
101
102
103
package uo
import (
log "github.com/sirupsen/logrus"
)
type ObjectIteratorFunc func(it *ObjectIterator) error
type ObjectIterator struct {
path []interface{}
keys []interface{}
cb ObjectIteratorFunc
}
func NewObjectIterator(o map[string]interface{}) *ObjectIterator {
return &ObjectIterator{
path: []interface{}{o},
}
}
func (it *ObjectIterator) KeyPath() []interface{} {
return it.keys
}
func (it *ObjectIterator) Key() interface{} {
if len(it.keys) == 0 {
return nil
}
return it.keys[len(it.keys)-1]
}
func (it *ObjectIterator) Parent() interface{} {
if len(it.path) < 2 {
return nil
}
return it.path[len(it.path)-2]
}
func (it *ObjectIterator) Value() interface{} {
return it.path[len(it.path)-1]
}
func (it *ObjectIterator) SetValue(v interface{}) error {
return SetChild(it.Parent(), it.Key(), v)
}
func (it *ObjectIterator) JsonPath() string {
return KeyListToJsonPath(it.keys)
}
func (it *ObjectIterator) IterateLeafs(cb ObjectIteratorFunc) error {
it.cb = cb
return it.iterateInterface()
}
func (it *ObjectIterator) iterateInterface() error {
value := it.Value()
if _, ok := value.(map[string]interface{}); ok {
return it.iterateMap()
} else if _, ok := value.([]interface{}); ok {
return it.iterateList()
} else {
return it.cb(it)
}
}
func (it *ObjectIterator) iterateMap() error {
m, ok := it.Value().(map[string]interface{})
if !ok {
log.Fatalf("!ok")
}
for k, v := range m {
it.path = append(it.path, v)
it.keys = append(it.keys, k)
err := it.iterateInterface()
it.path = it.path[:len(it.path)-1]
it.keys = it.keys[:len(it.keys)-1]
if err != nil {
return err
}
}
return nil
}
func (it *ObjectIterator) iterateList() error {
l, ok := it.Value().([]interface{})
if !ok {
log.Fatalf("!ok")
}
for i, e := range l {
it.path = append(it.path, e)
it.keys = append(it.keys, i)
err := it.iterateInterface()
it.path = it.path[:len(it.path)-1]
it.keys = it.keys[:len(it.keys)-1]
if err != nil {
return err
}
}
return nil
}