Question

我有一个结构：

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软件包，嗯...有什么帮助吗？：）

Answer 1

您可以使用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
    }
}

https://play.golang.com/p/qlt_mWsXGju

Golang通过JSON标记获取结构的字段名称

1 个答案: