Go模板：利用FuncMap实现字符串大小写转换及自定义函数扩展

package main

import (
    "bytes"
    "fmt"
    "strings"
    "text/template"
)

// TemplateData 定义了将要传递给模板的数据结构
type TemplateData struct {
    Name string
}

func main() {
    // 1. 创建FuncMap，将strings.ToUpper函数映射到模板中的"ToUpper"名称
    // 这样，在模板中就可以通过"ToUpper"来调用strings.ToUpper功能
    funcMap := template.FuncMap{
        "ToUpper": strings.ToUpper,
        // 可以在这里注册更多自定义函数，例如：
        // "ToLower": strings.ToLower,
        // "Add": func(a, b int) int { return a + b },
    }

    // 2. 定义模板内容。注意，这里使用我们注册的"ToUpper"函数
    templateString := `Hello, {{ .Name | ToUpper }}!`

    // 3. 创建一个新的模板实例，并使用Funcs方法注册FuncMap
    // 重要提示：Funcs方法必须在Parse或ParseFiles方法之前调用，
    // 否则注册的函数将不会对当前模板实例生效。
    tmpl, err := template.New("myTemplate").Funcs(funcMap).Parse(templateString)
    if err != nil {
        fmt.Printf("模板解析失败: %v\n", err)
        return
    }

    // 4. 准备数据，将其传递给模板
    data := TemplateData{"world"}
    var result bytes.Buffer // 用于存储模板执行结果的缓冲区

    // 5. 执行模板并将结果写入缓冲区
    err = tmpl.Execute(&result, data)
    if err != nil {
        fmt.Printf("模板执行失败: %v\n", err)
        return
    }

    // 6. 打印模板执行后的字符串结果
    fmt.Println(result.String()) // 预期输出: Hello, WORLD!
}

package main

import (
    "bytes"
    "fmt"
    "strings"
    "text/template"
    "time" // 导入time包用于日期时间操作
)

// TemplateData 定义了将要传递给模板的数据结构
type TemplateData struct {
    Name string
    CurrentTime time.Time // 添加一个时间字段
}

// formatTime 是一个自定义函数，用于格式化时间
func formatTime(t time.Time, layout string) string {
    return t.Format(layout)
}

func main() {
    funcMap := template.FuncMap{
        "ToUpper":   strings.ToUpper,
        "FormatTime": formatTime, // 注册自定义的日期格式化函数
    }

    // 模板中同时使用ToUpper和FormatTime函数
    templateString := `
        Hello, {{ .Name | ToUpper }}!
        Current time: {{ .CurrentTime | FormatTime "2006-01-02 15:04:05" }}
    `

    tmpl, err := template.New("myTemplate").Funcs(funcMap).Parse(templateString)
    if err != nil {
        fmt.Printf("模板解析失败: %v\n", err)
        return
    }

    // 准备包含时间和名称的数据
    data := TemplateData{
        Name: "world",
        CurrentTime: time.Now(), // 获取当前时间
    }
    var result bytes.Buffer

    err = tmpl.Execute(&result, data)
    if err != nil {
        fmt.Printf("模板执行失败: %v\n", err)
        return
    }

    fmt.Println(result.String())
}