golang template使用自定义函数

golang template使用自定义函数

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

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

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)
    }
}

运行：

    Friends      : [Bob Json Tom]
    1st   Friend : Bob
    2nd   Friend : Json
    3rd   Friend : Tom

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

const templateText = `
   Nrd   Friend : {{index .Friends (len .Friends)}}
`

执行模版会报错：

Executing template: template: :<line>:<column>: executing "" at <index .Friends (len .Friends)>: error calling index: index out of range: 3

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

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))

运行结果：

    Nrd   Friend : Tom

可见函数sub起效了。

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

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))

运行：

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

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

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

再运行：

    Nrd   Friend : Tom

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

有疑问加站长微信联系（非本文作者）