Slice location

mitchellh · Jun 12, 2014 · e311b7b · e311b7b
1 parent f74a140
commit e311b7b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/reflectwalk.go b/reflectwalk.go
@@ -15,6 +15,7 @@ const (
 	Map
 	MapKey
 	MapValue
+	Slice
 	SliceElem
 	StructField
 	WalkLoc
@@ -177,6 +178,11 @@ func walkPrimitive(v reflect.Value, w interface{}) error {
 }
 
 func walkSlice(v reflect.Value, w interface{}) (err error) {
+	ew, ok := w.(EnterExitWalker)
+	if ok {
+		ew.Enter(Slice)
+	}
+
 	if sw, ok := w.(SliceWalker); ok {
 		if err := sw.Slice(v); err != nil {
 			return err
@@ -206,6 +212,11 @@ func walkSlice(v reflect.Value, w interface{}) (err error) {
 		}
 	}
 
+	ew, ok = w.(EnterExitWalker)
+	if ok {
+		ew.Exit(Slice)
+	}
+
 	return nil
 }