Merge pull request #7 from smallfish/master

support custom struct's field tag name

structure.go

@@ -3,6 +3,10 @@ package structure
 
 import "reflect"
 
+var (
+	DefaultTagName = "structure" // struct's field default tag name
+)
+
 // Map converts the given s struct to a map[string]interface{}, where the keys
 // of the map are the field names and the values of the map the associated
 // values of the fields. The default key string is the struct field name but
@@ -40,7 +44,7 @@ func Map(s interface{}) map[string]interface{} {
 
 		// override if the user passed a structure tag value
 		// ignore if the user passed the "-" value
-		if tag := field.Tag.Get("structure"); tag != "" {
+		if tag := field.Tag.Get(DefaultTagName); tag != "" {
 			name = tag
 		}
 
@@ -248,7 +252,7 @@ func strctInfo(s interface{}) (reflect.Value, []reflect.StructField) {
 		}
 
 		// don't check if it's omitted
-		if tag := field.Tag.Get("structure"); tag == "-" {
+		if tag := field.Tag.Get(DefaultTagName); tag == "-" {
 			continue
 		}
 

structure_test.go

@@ -89,6 +89,41 @@ func TestMap_Tag(t *testing.T) {
 
 }
 
+func TestMap_CustomTag(t *testing.T) {
+	var T = struct {
+		A string `dd:"x"`
+		B int    `dd:"y"`
+		C bool   `dd:"z"`
+	}{
+		A: "a-value",
+		B: 2,
+		C: true,
+	}
+
+	defaultName := DefaultTagName
+	DefaultTagName = "dd"
+	defer func() {
+		DefaultTagName = defaultName
+	}()
+	a := Map(T)
+
+	inMap := func(key interface{}) bool {
+		for k := range a {
+			if reflect.DeepEqual(k, key) {
+				return true
+			}
+		}
+		return false
+	}
+
+	for _, key := range []string{"x", "y", "z"} {
+		if !inMap(key) {
+			t.Errorf("Map should have the key %v", key)
+		}
+	}
+
+}
+
 func TestMap_Nested(t *testing.T) {
 	type A struct {
 		Name string