我有一个结构:
type Human struct {
Head string `json:"a1"`
Body string `json:"a2"`
Leg string `json:"a3"`
}
如何通过提供JSON标记名来获取结构的字段名称?可能是这样的:
fmt.Println(getFieldName("a1")) // "Head"
fmt.Println(getFieldName("a2")) // "Body"
fmt.Println(getFieldName("a99")) // ""
func getFieldName(tag string) (fieldname string) {
/* ... */
}
我该如何实现getFieldName
功能?我在线阅读,似乎我需要使用reflect
包,嗯......任何帮助之手? :)
您可以使用reflect
包循环结构的字段并匹配其标记值。
func getFieldName(tag, key string, s interface{}) (fieldname string) {
rt := reflect.TypeOf(s)
if rt.Kind() != reflect.Struct {
panic("bad type")
}
for i := 0; i < rt.NumField(); i++ {
f := rt.Field(i)
v := strings.Split(f.Tag.Get(key), ",")[0] // use split to ignore tag "options" like omitempty, etc.
if v == tag {
return f.Name
}
}
return ""
}
https://play.golang.com/p/2zCC7pZKJTz
或者,正如@icza所指出的,您可以构建一个地图,然后使用它来更快地查找。
// Human json a1 Head
var fieldsByTag = make(map[reflect.Type]map[string]map[string]string)
func buildFieldsByTagMap(key string, s interface{}) {
rt := reflect.TypeOf(s)
if rt.Kind() != reflect.Struct {
panic("bad type")
}
if fieldsByTag[rt] == nil {
fieldsByTag[rt] = make(map[string]map[string]string)
}
if fieldsByTag[rt][key] == nil {
fieldsByTag[rt][key] = make(map[string]string)
}
for i := 0; i < rt.NumField(); i++ {
f := rt.Field(i)
v := strings.Split(f.Tag.Get(key), ",")[0] // use split to ignore tag "options"
if v == "" || v == "-" {
continue
}
fieldsByTag[rt][key][v] = f.Name
}
}