如何在golang html/template 中的多个地方创建全局变量和更改？

Question

我正在html/template创建一个变量，并根据条件更改值。 但是值的范围只停留在if条件内：

{{if .UserData}}
    {{$currentUserId := .UserData.UserId}}
    [<a href="#ask_question">Inside {{$currentUserId}}</a>]
{{else}}
    {{$currentUserId := 0}}
{{end}}
[<a href="#ask_question">outside {{$currentUserId}}</a>]

在 if 条件内，我得到正确的值，但在它之外是0 。 如何在条件之外使用$currentUserId ？ 有人可以帮我解决这个问题吗？

Answer 1

Go 1.11 添加了对更改模板变量值的支持。 要定义变量，请使用:= ：

{{$currentUserId := 0}}

要更改其值，请使用 assignment = ：

{{$currentUserId = .UserData.UserId}}

如果变量是在{{if}}块之外创建但在其中更改，则更改将在{{if}}块之后可见。

{{$currentUserId := 0 -}}
Before: {{$currentUserId}}
{{if .UserData -}}
    {{$currentUserId = .UserData.UserId}}
    [<a href="#ask_question">Inside {{$currentUserId}}</a>]
{{else}}
    {{$currentUserId = 0}}
{{end}}
[<a href="#ask_question">outside {{$currentUserId}}</a>]

像这样测试：

m := map[string]interface{}{}
t := template.Must(template.New("").Parse(src))

m["UserData"] = UserData{99}
if err := t.Execute(os.Stdout, m); err != nil {
    panic(err)

输出是（在Go Playground上试试）：

Before: 0

    [<a href="#ask_question">Inside 99</a>]

[<a href="#ask_question">outside 99</a>]

原答案如下。

---

简短的回答是：你不能。

根据设计理念，模板不应包含复杂的逻辑。 在模板中，您只能创建新变量，不能更改现有模板变量的值。 当您在{{if}}块内执行{{$currentUserId := .UserData.UserId}} ，会创建一个新变量，该变量{{$currentUserId := .UserData.UserId}}外部变量，其范围扩展到{{end}}操作。

这在text/template包的变量部分中有描述（同样适用于html/template包）：

变量的范围扩展到声明它的控制结构（“if”、“with”或“range”）的“结束”操作，如果没有这样的控制结构，则扩展到模板的末尾。 模板调用不会从其调用点继承变量。

可能的解决方法

在您的情况下，最简单的方法是将自定义函数CurrentUserId()注册到您的模板，如果UserData存在，则返回其 ID UserData.UserId() ，如果不存在，则返回0与您的情况一样。

这是一个如何完成的示例：

type UserData struct {
    UserId int
}

func main() {
    m := map[string]interface{}{}
    t := template.Must(template.New("").Funcs(template.FuncMap{
        "CurrentUserId": func() int {
            if u, ok := m["UserData"]; ok {
                return u.(UserData).UserId
            }
            return 0
        },
    }).Parse(src))

    if err := t.Execute(os.Stdout, m); err != nil {
        panic(err)
    }
    m["UserData"] = UserData{99}
    if err := t.Execute(os.Stdout, m); err != nil {
        panic(err)
    }
}

const src = `Current user id: {{CurrentUserId}}
`

它首先执行没有UserData的模板，然后执行具有UserId = 99 UserData的模板。 输出是：

Current user id: 0
Current user id: 99

在Go Playground上试试。

模拟可变变量

您还可以模拟可变变量，但也可以使用已注册的自定义函数。 你可以注册一个像SetCurrentUserId()这样的函数，它会改变一个变量的值，最好是在传递给模板执行的参数中，这样它就可以安全地并发使用。

这是一个如何做到的示例（它使用map作为模板数据，并且SetCurrentUserId()将当前用户 ID 设置为该地图中的一个值）：

func main() {
    m := map[string]interface{}{}
    t := template.Must(template.New("").Funcs(template.FuncMap{
        "SetCurrentUserId": func(id int) string {
            m["CurrentUserId"] = id
            return ""
        },
    }).Parse(src))

    if err := t.Execute(os.Stdout, m); err != nil {
        panic(err)
    }
    m["UserData"] = UserData{99}
    if err := t.Execute(os.Stdout, m); err != nil {
        panic(err)
    }
}

const src = `Before: {{.CurrentUserId}}
{{if .UserData}}
    {{SetCurrentUserId .UserData.UserId}}Inside: {{.CurrentUserId}}
{{else}}
    {{SetCurrentUserId 0}}Inside: {{.CurrentUserId}}
{{end}}
After: {{.CurrentUserId}}
`

这再次首先执行没有UserData的模板，然后执行具有UserId = 99 UserData模板。 输出是：

Before: 
    Inside: 0
After: 0
Before: 0
    Inside: 99
After: 99

在Go Playground上试试。