如何在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>]

原始答案如下。

简短的答案是：您不能。

根据设计原理，模板不应包含复杂的逻辑。在模板中，您只能创建新变量，而不能更改现有模板变量的值。当您{{$currentUserId := .UserData.UserId}}在一个{{if}}块中执行操作时，这将创建一个新变量，该变量将外部变量遮盖起来，并且其作用域扩展到{{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，然后在UserData具有的情况下执行模板UserId = 99。输出为：

Current user id: 0
Current user id: 99

在Go Playground上尝试一下。

模拟可变变量

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

这是一个示例（它使用a 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，然后再使用UserData具有的模板UserId = 99。输出为：

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

在Go Playground上尝试一下。