-
Notifications
You must be signed in to change notification settings - Fork 270
/
retrieve.go
89 lines (68 loc) · 1.84 KB
/
retrieve.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
package funk
import (
"reflect"
"strings"
)
// Get retrieves the value at path of struct(s).
func Get(out interface{}, path string) interface{} {
result := get(reflect.ValueOf(out), path)
if result.Kind() != reflect.Invalid {
return result.Interface()
}
return nil
}
func get(value reflect.Value, path string) reflect.Value {
if value.Kind() == reflect.Slice || value.Kind() == reflect.Array {
var resultSlice reflect.Value
length := value.Len()
if length == 0 {
return resultSlice
}
for i := 0; i < length; i++ {
item := value.Index(i)
resultValue := get(item, path)
if resultValue.Kind() == reflect.Invalid {
continue
}
resultType := resultValue.Type()
if resultSlice.Kind() == reflect.Invalid {
resultType := reflect.SliceOf(resultType)
resultSlice = reflect.MakeSlice(resultType, 0, 0)
}
resultSlice = reflect.Append(resultSlice, resultValue)
}
// if the result is a slice of a slice, we need to flatten it
if resultSlice.Kind() != reflect.Invalid && resultSlice.Type().Elem().Kind() == reflect.Slice {
return flattenDeep(resultSlice)
}
return resultSlice
}
parts := strings.Split(path, ".")
for _, part := range parts {
value = redirectValue(value)
kind := value.Kind()
switch kind {
case reflect.Invalid:
continue
case reflect.Struct:
value = value.FieldByName(part)
case reflect.Map:
value = value.MapIndex(reflect.ValueOf(part))
case reflect.Slice, reflect.Array:
value = get(value, part)
default:
return reflect.ValueOf(nil)
}
}
return value
}
// Get retrieves the value of the pointer or default.
func GetOrElse(v interface{}, def interface{}) interface{} {
val := reflect.ValueOf(v)
if v == nil || (val.Kind() == reflect.Ptr && val.IsNil()) {
return def
} else if val.Kind() != reflect.Ptr {
return v
}
return val.Elem().Interface()
}