一尘不染

如何在golang html / template中创建全局变量并在多个位置进行更改?

go

我在中创建一个变量,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外界条件下使用?有人可以帮我吗?


阅读 207

收藏
2020-07-02

共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上尝试一下。

2020-07-02