walk: refacto

This commit is contained in:
Muyao CHEN 2024-09-20 20:32:37 +02:00
parent 049c346e63
commit 1c3ad9b9d8

View File

@ -7,22 +7,18 @@ import (
func Walk(x interface{}, fn func(string)) {
val := getValue(x)
for i := 0; i < val.NumField(); i++ {
field := val.Field(i)
if val.Kind() == reflect.Slice {
for i := 0; i < val.Len(); i++ {
Walk(val.Index(i).Interface(), fn)
}
}
switch field.Kind() {
case reflect.String:
fn(field.String())
case reflect.Struct:
switch val.Kind() {
case reflect.Struct:
for i := 0; i < val.NumField(); i++ {
// XXX: Interface() to get interface
Walk(field.Interface(), fn)
Walk(val.Field(i).Interface(), fn)
}
case reflect.Slice:
for i := 0; i < val.Len(); i++ {
Walk(val.Index(i).Interface(), fn)
}
case reflect.String:
fn(val.String())
}
}