golang template使用自定义函数

golang template使用自定义函数

go语言中文网站www.topgoer.com

需求来源是这样的：如何拿到一个slice的最后一个元素

在模版里面，我们可以获取slice的任意元素，只要给定下标即可，例如：

<code>package main

import (
    "os"
    "log"
    "text/template"
)

const templateText = `
    Friends      : {{.Friends}}
    1st   Friend : {{index .Friends 0}}
    2nd   Friend : {{index .Friends 1}}
    3rd   Friend : {{index .Friends 2}}
`
func main() {
    type Recipient struct {
        Name        string
        Friends     []string
    }
    
    recipient := Recipient{
        Name    : "Jack",
        Friends : []string {"Bob", "Json", "Tom"},
    }

    t := template.Must(template.New("").Parse(templateText))
    err := t.Execute(os.Stdout, recipient)
    if err != nil {
        log.Println("Executing template:", err)
    }
}
/<code>

运行：

<code>    Friends      : [Bob Json Tom]
    1st   Friend : Bob
    2nd   Friend : Json
    3rd   Friend : Tom
/<code> 

然而，如何获取最后一个元素呢？因为这个下标不是一个常量，是根据slice大小动态变化的。 我们首先会想到使用len(.Friends)行不行？

<code>const templateText = `
   Nrd   Friend : {{index .Friends (len .Friends)}}
`
/<code>

执行模版会报错：

<code>Executing template: template: :<line>:<column>: executing "" at <index>: error calling index: index out of range: 3
/<index>/<column>/<line>/<code>

因为slice最后一个元素下标是(len()-1)，看来用一个算术运行就可以了，遗憾的是golang template并没有提供简单的加减运算，也没有相应的函数(golang template的所有系统函数都在这儿: https://golang.org/pkg/text/template/#hdr-Functions )；所以我们必须自定义一个减函数(sub)

<code>const templateText = `
    Nrd   Friend : {{index .Friends (sub (len .Friends) 1)}}
`

templateFunc := map[string]interface{} {
    "sub": func(a, b int) int { return a - b },
}

t := template.Must(template.New("").Funcs(template.FuncMap(templateFunc)).Parse(templateText))
/<code>

运行结果：

<code>    Nrd   Friend : Tom
/<code>

可见函数sub起效了。

这里我们定义的是通用的add/sub函数，那么我们能不能直接定义last函数呢，返回slice的最后一个元素：

<code>const templateText = `
    Nrd   Friend : {{last .Friends}}
`

templateFunc := map[string]interface{} {
    "last": func(s []interface{}) interface{} { return s[len(s)-1] },
}

t := template.Must(template.New("").Funcs(template.FuncMap(templateFunc)).Parse(templateText))
/<code>

运行：

<code>Executing template: template: :<line>:<column>: executing "" at <.friends>: wrong type for value; expected []interface {}; got []string
/<.friends>/<column>/<line>/<code>

是不是很遗憾，为啥通用类型interface{}不支持呢，一定要具体类型吗？我们重新定义last函数：

<code>templateFunc := map[string]interface{} {
    "last": func(s []string) string { return s[len(s)-1] },
}
/<code>

再运行：

<code>    Nrd   Friend : Tom
/<code>

竟然好了。这个说来我觉得还是有点遗憾了，为啥不支持通用类型interface{}呢，本来slice取其中的元素操作本身并不需要关系元素类型，用interface{}是很好的方法。 抱怨归抱怨，我也没有办法，只能遵从这个设计吧。

作者：CodingCode
链接：https://www.jianshu.com/p/1f70ad682cb6
来源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

閱讀更多 go的慢慢學習路 的文章

關鍵字: 镜音双子 自定义 函数