Golang模板函数自定义与渲染技巧-Golang-PHP中文网

自定义golang模板函数需通过template.funcmap注册函数，如toupper；处理复杂数据可用管道访问嵌套字段，如.address.city；条件判断用{{if}} {{else}} {{end}}，循环用{{range}}遍历数据；为防xss，默认自动转义html，可显式使用{{.userinput | html}}；结构体方法若首字母大写，可在模板中直接调用，如{{.fullname}}。

golang模板函数自定义与渲染技巧

Golang模板函数自定义与渲染技巧：简单来说，就是教你如何扩展Go模板的功能，让它能做更多的事情，以及如何更灵活地使用这些模板。

模板函数自定义与渲染技巧

如何自定义Golang模板函数？

自定义模板函数，其实就是给

template.FuncMap

添加你自己的函数。这有点像给工具箱添加新工具。你需要先定义一个Go函数，然后把它注册到

FuncMap

里，最后在模板里就可以像使用内置函数一样使用它了。

举个例子，假设你需要一个函数来把字符串转换成大写。你可以这样：

package main

import (
    "fmt"
    "html/template"
    "os"
    "strings"
)

// 定义一个函数，将字符串转为大写
func toUpper(s string) string {
    return strings.ToUpper(s)
}

func main() {
    // 创建一个FuncMap，并将自定义函数注册进去
    funcMap := template.FuncMap{
        "toUpper": toUpper,
    }

    // 创建一个模板，并使用FuncMap
    tmpl, err := template.New("test").Funcs(funcMap).Parse(`
        {{ .Name }} 的大写是：{{ toUpper .Name }}
    `)
    if err != nil {
        panic(err)
    }

    // 定义数据
    data := struct {
        Name string
    }{
        Name: "golang",
    }

    // 渲染模板
    err = tmpl.Execute(os.Stdout, data)
    if err != nil {
        panic(err)
    }
}

这段代码首先定义了一个

toUpper

函数，然后创建了一个

FuncMap

，把

toUpper

函数注册到里面，键名是"toUpper"。之后，创建模板的时候，通过

.Funcs(funcMap)

把

FuncMap

传进去。在模板里，就可以直接使用

{{ toUpper .Name }}

来调用这个函数了。

如何在Golang模板中处理复杂数据结构？

处理复杂数据结构的关键在于理解模板的“管道”概念。管道允许你将一个表达式的结果传递给另一个函数或表达式。这有点像流水线，数据在流水线上一步步处理。

例如，假设你有一个

User

结构体，里面有一个

Address

结构体，你需要访问

Address

里面的

City

字段。你可以这样：

package main

import (
    "fmt"
    "html/template"
    "os"
)

type Address struct {
    City string
}

type User struct {
    Name    string
    Address Address
}

func main() {
    tmpl, err := template.New("test").Parse(`
        用户 {{ .Name }} 住在 {{ .Address.City }}
    `)
    if err != nil {
        panic(err)
    }

    data := User{
        Name: "张三",
        Address: Address{
            City: "北京",
        },
    }

    err = tmpl.Execute(os.Stdout, data)
    if err != nil {
        panic(err)
    }
}

这里，

.Address.City

就是通过管道访问嵌套结构体的字段。如果数据结构更复杂，你还可以结合自定义函数来处理，比如自定义一个函数来格式化地址信息。

Golang模板中的条件判断和循环技巧

条件判断和循环是模板中常用的控制结构。Go模板使用

{{ if ... }}

{{ else }}

{{ end }}

来进行条件判断，使用

{{ range ... }}

{{ end }}

来进行循环。

比如，你想根据用户的年龄来显示不同的消息：

ApiPost接口调试与文档生成工具

ApiPost是一个支持团队协作，支持模拟POST、GET、PUT等常见请求，并可直接生成文档的API调试、管理工具，ApiPost是后台接口开发者或前端、接口测试人员的工作必备工具。快速生成、一键导出API文档。感兴趣的朋友快来下载吧。软件说明ApiPost官方版是一款十分出色的接口调试与文档生成工具，ApiPost官方版界面美观大方，功能强劲实用，支持团队协作，支持模拟POST、GET、PUT等常见请求，是后台接口开发者或前端、接口测试人员的工作必备工具。软件特色更方便支持接口调试的同时快速生成、一键

下载

package main

import (
    "fmt"
    "html/template"
    "os"
)

type User struct {
    Name string
    Age  int
}

func main() {
    tmpl, err := template.New("test").Parse(`
        {{ .Name }}，
        {{ if gt .Age 18 }}
            你已经成年了。
        {{ else }}
            你还未成年。
        {{ end }}
    `)
    if err != nil {
        panic(err)
    }

    data := User{
        Name: "李四",
        Age:  20,
    }

    err = tmpl.Execute(os.Stdout, data)
    if err != nil {
        panic(err)
    }
}

这里，

{{ if gt .Age 18 }}

使用了内置的

gt

函数（greater than），判断年龄是否大于18。

循环的例子：

package main

import (
    "fmt"
    "html/template"
    "os"
)

func main() {
    tmpl, err := template.New("test").Parse(`
        {{ range . }}
            {{ . }}
        {{ end }}
    `)
    if err != nil {
        panic(err)
    }

    data := []string{"苹果", "香蕉", "橙子"}

    err = tmpl.Execute(os.Stdout, data)
    if err != nil {
        panic(err)
    }
}

这个例子循环遍历一个字符串切片，并输出每个元素。

如何避免Golang模板中的安全漏洞（例如XSS）？

避免XSS攻击的关键在于对用户输入进行适当的转义。Go模板默认会自动转义HTML，防止XSS攻击。但是，在某些情况下，你可能需要手动转义。

Go模板提供了几个转义函数：

```
html
```
: 将字符串转义为HTML。
```
js
```
: 将字符串转义为JavaScript。
```
urlquery
```
: 将字符串转义为URL查询参数。

例如，如果你的模板中包含用户输入的HTML内容，你需要使用

html

函数进行转义：

package main

import (
    "fmt"
    "html/template"
    "os"
)

func main() {
    tmpl, err := template.New("test").Parse(`
        <div>
            用户输入：{{ .UserInput | html }}
        </div>
    `)
    if err != nil {
        panic(err)
    }

    data := struct {
        UserInput string
    }{
        UserInput: "<script>alert('XSS')</script>",
    }

    err = tmpl.Execute(os.Stdout, data)
    if err != nil {
        panic(err)
    }
}

这里，

{{ .UserInput | html }}

使用了

html

函数对用户输入进行转义，防止XSS攻击。记住，永远不要信任用户输入，并始终进行适当的转义。此外，尽量避免在模板中直接拼接字符串，因为这可能会引入安全漏洞。

如何在Golang模板中使用自定义的结构体方法？

你可以在模板中直接调用结构体的方法，这使得模板可以更方便地处理结构体数据。前提是方法要是导出的（首字母大写）。

package main

import (
    "fmt"
    "html/template"
    "os"
)

type User struct {
    FirstName string
    LastName  string
}

// 定义一个结构体方法，返回用户的全名
func (u User) FullName() string {
    return u.FirstName + " " + u.LastName
}

func main() {
    tmpl, err := template.New("test").Parse(`
        {{ .FullName }}
    `)
    if err != nil {
        panic(err)
    }

    data := User{
        FirstName: "张",
        LastName:  "三",
    }

    err = tmpl.Execute(os.Stdout, data)
    if err != nil {
        panic(err)
    }
}

这里，