Merge pull request #2 from damoye/master

feat: handle multiple json tag like omitempty
teambition · Feb 3, 2018 · de5ce4e · de5ce4e
2 parents 5f29f24 + e3b0100
commit de5ce4e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/util.go b/util.go
@@ -2,6 +2,7 @@ package jsonmask
 
 import (
 	"reflect"
+	"strings"
 	"sync"
 
 	"gopkg.in/oleiade/reflections.v1"
@@ -32,6 +33,11 @@ func getFiledNamesByJSONKeys(obj interface{}, jsonKeys []string) (map[string]str
 	}
 
 	for fieldName, key := range fieldTagMap {
+		keys := strings.Split(key, ",")
+		if len(keys) == 0 {
+			continue
+		}
+		key := keys[0]
 		for _, jsonKey := range jsonKeys {
 			if key == jsonKey {
 				fieldNames[jsonKey] = fieldName