Filler.FuncByType

Author: mcuadros

defaults.go

@@ -33,13 +33,13 @@ func getDefaultFiller() *Filler {
 
 func newDefaultFiller() *Filler {
 	funcs := make(map[reflect.Kind]fillerFunc, 0)
-	funcs[reflect.Bool] = func(field *fieldData, defaultValue string) {
-		value, _ := strconv.ParseBool(defaultValue)
+	funcs[reflect.Bool] = func(field *fieldData, tagValue string) {
+		value, _ := strconv.ParseBool(tagValue)
 		field.Value.SetBool(value)
 	}
 
-	funcs[reflect.Int] = func(field *fieldData, defaultValue string) {
-		value, _ := strconv.ParseInt(defaultValue, 10, 64)
+	funcs[reflect.Int] = func(field *fieldData, tagValue string) {
+		value, _ := strconv.ParseInt(tagValue, 10, 64)
 		field.Value.SetInt(value)
 	}
 
@@ -48,15 +48,15 @@ func newDefaultFiller() *Filler {
 	funcs[reflect.Int32] = funcs[reflect.Int]
 	funcs[reflect.Int64] = funcs[reflect.Int]
 
-	funcs[reflect.Float32] = func(field *fieldData, defaultValue string) {
-		value, _ := strconv.ParseFloat(defaultValue, 64)
+	funcs[reflect.Float32] = func(field *fieldData, tagValue string) {
+		value, _ := strconv.ParseFloat(tagValue, 64)
 		field.Value.SetFloat(value)
 	}
 
 	funcs[reflect.Float64] = funcs[reflect.Float32]
 
-	funcs[reflect.Uint] = func(field *fieldData, defaultValue string) {
-		value, _ := strconv.ParseUint(defaultValue, 10, 64)
+	funcs[reflect.Uint] = func(field *fieldData, tagValue string) {
+		value, _ := strconv.ParseUint(tagValue, 10, 64)
 		field.Value.SetUint(value)
 	}
 
@@ -65,21 +65,21 @@ func newDefaultFiller() *Filler {
 	funcs[reflect.Uint32] = funcs[reflect.Uint]
 	funcs[reflect.Uint64] = funcs[reflect.Uint]
 
-	funcs[reflect.String] = func(field *fieldData, defaultValue string) {
-		field.Value.SetString(defaultValue)
+	funcs[reflect.String] = func(field *fieldData, tagValue string) {
+		field.Value.SetString(tagValue)
 	}
 
-	funcs[reflect.Slice] = func(field *fieldData, defaultValue string) {
+	funcs[reflect.Slice] = func(field *fieldData, tagValue string) {
 		if field.Value.Type().Elem().Kind() == reflect.Uint8 {
 			if field.Value.Bytes() != nil {
 				return
 			}
 
-			field.Value.SetBytes([]byte(defaultValue))
+			field.Value.SetBytes([]byte(tagValue))
 		}
 	}
 
-	funcs[reflect.Struct] = func(field *fieldData, defaultValue string) {
+	funcs[reflect.Struct] = func(field *fieldData, tagValue string) {
 		fields := getDefaultFiller().getFieldsFromValue(field.Value)
 		getDefaultFiller().setDefaultValues(fields)
 	}

factory.go

@@ -44,13 +44,13 @@ func newFactoryFiller() *Filler {
 	funcs[reflect.Int32] = funcs[reflect.Int]
 	funcs[reflect.Int64] = funcs[reflect.Int]
 
-	funcs[reflect.Float32] = func(field *fieldData, defaultValue string) {
+	funcs[reflect.Float32] = func(field *fieldData, tagValue string) {
 		field.Value.SetFloat(rand.Float64())
 	}
 
 	funcs[reflect.Float64] = funcs[reflect.Float32]
 
-	funcs[reflect.Uint] = func(field *fieldData, defaultValue string) {
+	funcs[reflect.Uint] = func(field *fieldData, tagValue string) {
 		field.Value.SetUint(uint64(rand.Uint32()))
 	}
 
@@ -59,11 +59,11 @@ func newFactoryFiller() *Filler {
 	funcs[reflect.Uint32] = funcs[reflect.Uint]
 	funcs[reflect.Uint64] = funcs[reflect.Uint]
 
-	funcs[reflect.String] = func(field *fieldData, defaultValue string) {
+	funcs[reflect.String] = func(field *fieldData, tagValue string) {
 		field.Value.SetString(randomString())
 	}
 
-	funcs[reflect.Slice] = func(field *fieldData, defaultValue string) {
+	funcs[reflect.Slice] = func(field *fieldData, tagValue string) {
 		if field.Value.Type().Elem().Kind() == reflect.Uint8 {
 			if field.Value.Bytes() != nil {
 				return
@@ -73,7 +73,7 @@ func newFactoryFiller() *Filler {
 		}
 	}
 
-	funcs[reflect.Struct] = func(field *fieldData, defaultValue string) {
+	funcs[reflect.Struct] = func(field *fieldData, tagValue string) {
 		fields := getDefaultFiller().getFieldsFromValue(field.Value)
 		getDefaultFiller().setDefaultValues(fields)
 	}

filler.go

@@ -1,6 +1,7 @@
 package defaults
 
 import (
+	"fmt"
 	"reflect"
 )
 
@@ -12,7 +13,7 @@ type fieldData struct {
 type fillerFunc func(field *fieldData, config string)
 
 type Filler struct {
-	FuncByName map[string]fillerFunc
+	FuncByType map[TypeHash]fillerFunc
 	FuncByKind map[reflect.Kind]fillerFunc
 	Tag        string
 }
@@ -90,12 +91,27 @@ func (f *Filler) isEmpty(field *fieldData) bool {
 func (f *Filler) setDefaultValue(field *fieldData) {
 	tagValue := field.Field.Tag.Get(f.Tag)
 
-	function := f.getFunctionByKind(field.Field.Type.Kind())
-	if function == nil {
+	function := f.getFunctionByType(field.Field.Type)
+	if function != nil {
+		function(field, tagValue)
 		return
 	}
 
-	function(field, tagValue)
+	function = f.getFunctionByKind(field.Field.Type.Kind())
+	if function != nil {
+		function(field, tagValue)
+		return
+	}
+
+	return
+}
+
+func (f *Filler) getFunctionByType(t reflect.Type) fillerFunc {
+	if f, ok := f.FuncByType[GetTypeHash(t)]; ok == true {
+		return f
+	}
+
+	return nil
 }
 
 func (f *Filler) getFunctionByKind(k reflect.Kind) fillerFunc {
@@ -105,3 +121,13 @@ func (f *Filler) getFunctionByKind(k reflect.Kind) fillerFunc {
 
 	return nil
 }
+
+type TypeHash string
+
+func GetTypeHash(t reflect.Type) TypeHash {
+	if t.Kind() == reflect.Ptr {
+		t = t.Elem()
+	}
+
+	return TypeHash(fmt.Sprintf("%s.%s", t.PkgPath(), t.Name()))
+}