Merge pull request #482 from jhernand/add_dig_map_support

Add support for digging inside maps

data/digger.go

@@ -106,6 +106,8 @@ func (d *Digger) digFieldFromValue(value reflect.Value, field string) interface{
 		return d.digFieldFromPtr(value, field)
 	case reflect.Struct:
 		return d.digFieldFromStruct(value, field)
+	case reflect.Map:
+		return d.digFieldFromMap(value, field)
 	default:
 		return nil
 	}
@@ -173,6 +175,15 @@ func (d *Digger) digFieldFromMethod(value reflect.Value, method reflect.Method)
 	return result.Interface()
 }
 
+func (d *Digger) digFieldFromMap(value reflect.Value, name string) interface{} {
+	key := reflect.ValueOf(name)
+	result := value.MapIndex(key)
+	if !result.IsValid() {
+		return nil
+	}
+	return result.Interface()
+}
+
 // lookupMethod tries to find a public method of the given value that matches the given path
 // segment. For example, if the path segment is `my_field` it will look for a method named
 // `GetMyField` or `MyField`. Only methods that don't have input parameters will be considered.

data/digger_test.go

@@ -206,6 +206,24 @@ var _ = Describe("Digger", func() {
 			"does_not_exist",
 			nil,
 		),
+		Entry(
+			"Map field",
+			map[string]interface{}{
+				"id": "123",
+			},
+			"id",
+			"123",
+		),
+		Entry(
+			"Nested map field",
+			map[string]interface{}{
+				"object": map[string]interface{}{
+					"id": "123",
+				},
+			},
+			"object.id",
+			"123",
+		),
 	)
 })