53 lines
936 B
Go
53 lines
936 B
Go
package walk
|
|
|
|
import (
|
|
"reflect"
|
|
)
|
|
|
|
func Walk(x interface{}, fn func(string)) {
|
|
val := getValue(x)
|
|
|
|
walkValue := func(value reflect.Value) {
|
|
Walk(value.Interface(), fn)
|
|
}
|
|
|
|
switch val.Kind() {
|
|
case reflect.String:
|
|
fn(val.String())
|
|
case reflect.Struct:
|
|
for i := 0; i < val.NumField(); i++ {
|
|
// XXX: Interface() to get interface
|
|
walkValue(val.Field(i))
|
|
}
|
|
case reflect.Slice, reflect.Array:
|
|
for i := 0; i < val.Len(); i++ {
|
|
walkValue(val.Index(i))
|
|
}
|
|
case reflect.Map:
|
|
for _, key := range val.MapKeys() {
|
|
walkValue(val.MapIndex(key))
|
|
}
|
|
case reflect.Chan:
|
|
for {
|
|
if v, ok := val.Recv(); ok {
|
|
walkValue(v)
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
case reflect.Func:
|
|
valFnResult := val.Call(nil)
|
|
for _, res := range valFnResult {
|
|
walkValue(res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func getValue(x interface{}) reflect.Value {
|
|
val := reflect.ValueOf(x)
|
|
if val.Kind() == reflect.Pointer {
|
|
return val.Elem()
|
|
}
|
|
return val
|
|
}
|