Go语言中如何检查文本模板数据里的键是否存在？

当使用Go模板的{{if .Num}}判断时，int类型的0会被视为false，无法区分“键/字段存在但值为0”和“键/字段不存在”的场景，以下是几种可行的解决方法：

1. 针对Map类型数据：自定义isset函数 #

如果传入模板的是map[string]interface{}，可以先检查键是否存在，再判断值是否非零：

步骤1：注册isset函数 #

func main() {
    tpl := template.Must(template.New("").Funcs(template.FuncMap{
        "isset": func(m map[string]interface{}, key string) bool {
            _, ok := m[key]
            return ok
        },
    }).Parse(`
{{if isset . "Num"}}
    {{if ne .Num 0}}
        Num存在且值不为0
    {{else}}
        Num存在但值为0
    {{end}}
{{else}}
    Num不存在
{{end}}
`))

    // 测试数据
    data1 := map[string]interface{}{"Num": 0}
    data2 := map[string]interface{}{"Num": 10}
    data3 := map[string]interface{}{}

    tpl.Execute(os.Stdout, data1)
    tpl.Execute(os.Stdout, data2)
    tpl.Execute(os.Stdout, data3)
}

模板中使用 #

通过isset先确认键存在，再用ne判断值是否不等于0，就能精准区分三种情况。

2. 针对结构体类型数据：自定义字段存在性检查函数 #

如果传入的是结构体，字段存在性是编译期确定的，需要通过反射实现字段存在性判断：

注册hasField函数 #

func hasField(v interface{}, fieldName string) bool {
    rv := reflect.ValueOf(v)
    if rv.Kind() != reflect.Struct {
        return false
    }
    _, ok := rv.Type().FieldByName(fieldName)
    return ok
}

// 在模板中注册
tpl := template.Must(template.New("").Funcs(template.FuncMap{
    "hasField": hasField,
}).Parse(`
{{if and (hasField . "Num") (ne .Num 0)}}
    Num字段存在且值不为0
{{else if hasField . "Num"}}
    Num字段存在但值为0
{{else}}
    Num字段不存在
{{end}}
`))

3. 通用方案：封装existsAndNonZero函数 #

如果需要同时兼容Map和结构体，可以封装一个通用函数，一次性判断“存在且非零”：

注册通用函数 #

func existsAndNonZero(v interface{}, name string) bool {
    rv := reflect.ValueOf(v)
    var fieldVal reflect.Value

    switch rv.Kind() {
    case reflect.Map:
        key := reflect.ValueOf(name)
        if !rv.MapIndex(key).IsValid() {
            return false
        }
        fieldVal = rv.MapIndex(key)
    case reflect.Struct:
        field, ok := rv.Type().FieldByName(name)
        if !ok {
            return false
        }
        fieldVal = rv.FieldByIndex(field.Index)
    default:
        return false
    }

    // 判断值是否为对应类型的零值
    zeroVal := reflect.Zero(fieldVal.Type()).Interface()
    return !reflect.DeepEqual(fieldVal.Interface(), zeroVal)
}

// 模板中使用
tpl := template.Must(template.New("").Funcs(template.FuncMap{
    "existsAndNonZero": existsAndNonZero,
}).Parse(`
{{if existsAndNonZero . "Num"}}
    Num存在且值不为0
{{else}}
    Num不存在或值为零值
{{end}}
`))

内容的提问来源于stack exchange，提问作者codewise69420