简介

最近在整理我们项目代码的时候,发现有很多活动的代码在结构和提供的功能上都非常相似。为了方便今后的开发,我花了一点时间编写了一个生成代码框架的工具,最大程度地降低重复劳动。代码本身并不复杂,且与项目代码关联性较大,这里就不展开介绍了。在这个过程中,我发现 Go 标准的模板库text/templatehtml/template使用起来比较束手束脚,很不方便。我从 GitHub 了解到quicktemplate这个第三方模板库,功能强大,语法简单,使用方便。今天我们就来介绍一下quicktemplate

快速使用

本文代码使用 Go Modules。

先创建代码目录并初始化:

  1. $ mkdir quicktemplate && cd quicktemplate
  2. $ go mod init github.com/go-quiz/go-daily-lib/quicktemplate

quicktemplate会将我们编写的模板代码转换为 Go 语言代码。因此我们需要安装quicktemplate包和一个名为qtc的编译器:

  1. $ go get -u github.com/valyala/quicktemplate
  2. $ go get -u github.com/valyala/quicktemplate/qtc

首先,我们需要编写quicktemplate格式的模板文件,模板文件默认以.qtpl作为扩展名。下面我编写了一个简单的模板文件greeting.qtpl

  1. All text outside function is treated as comments.
  2. {% func Greeting(name string, count int) %}
  3. {% for i := 0; i < count; i++ %}
  4. Hello, {%s name %}
  5. {% endfor %}
  6. {% endfunc %}

模板语法非常简单,我们只需要简单了解以下 2 点:

  • 模板以函数为单位,函数可以接受任意类型和数量的参数,这些参数可以在函数中使用。所有函数外的文本都是注释,qtc编译时会忽视注释;
  • 函数内的内容,除了语法结构,其他都会原样输出到渲染后的文本中,包括空格和换行

greeting.qtpl保存到templates目录,然后执行qtc命令。该命令会生成对应的 Go 文件greeting.qtpl.go,包名为templates。现在,我们就可以使用这个模板了:

  1. package main
  2. import (
  3. "fmt"
  4. "github.com/go-quiz/go-daily-lib/quicktemplate/get-started/templates"
  5. )
  6. func main() {
  7. fmt.Println(templates.Greeting("dj", 5))
  8. }

调用模板函数,传入参数,返回渲染后的文本:

  1. $ go run .
  2. Hello, dj
  3. Hello, dj
  4. Hello, dj
  5. Hello, dj
  6. Hello, dj

{%s name %}执行文本替换,{% for %}循环生成重复文本。输出中出现多个空格和换行,这是因为函数内除了语法结构,其他内容都会原样保留,包括空格和换行

需要注意的是,由于quicktemplate是将模板转换为 Go 代码使用的,所以如果模板有修改,必须先执行qtc命令重新生成 Go 代码,否则修改不生效

语法结构

quicktemplate支持 Go 常见的语法结构,if/for/func/import/return。而且写法与直接写 Go 代码没太大的区别,几乎没有学习成本。只是在模板中使用这些语法时,需要使用{%%}包裹起来,而且iffor等需要添加endif/endfor明确表示结束。

变量

上面我们已经看到如何渲染传入的参数name,使用{%s name %}。由于name是 string 类型,所以在{%后使用s指定类型。quicktemplate还支持其他类型的值:

  • 整型:{%d int %}{%dl int64 %}{%dul uint64 %}
  • 浮点数:{%f float %}。还可以设置输出的精度,使用{%f.precision float %}。例如{%f.2 1.2345 %}输出1.23
  • 字节切片([]byte):{%z bytes %}
  • 字符串:{%q str %}或字节切片:{%qz bytes %},引号转义为&quot;
  • 字符串:{%j str %}或字节切片:{%jz bytes %},没有引号;
  • URL 编码:{%u str %}{%uz bytes %}
  • {%v anything %}:输出等同于fmt.Sprintf("%v", anything)

先编写模板:

  1. {% func Types(a int, b float64, c []byte, d string) %}
  2. int: {%d a %}, float64: {%f.2 b %}, bytes: {%z c %}, string with quotes: {%q d %}, string without quotes: {%j d %}.
  3. {% endfunc %}

然后使用:

  1. func main() {
  2. fmt.Println(templates.Types(1, 5.75, []byte{'a', 'b', 'c'}, "hello"))
  3. }

运行:

  1. $ go run .
  2. int: 1, float64: 5.75, bytes: abc, string with quotes: &quot;hello&quot;, string without quotes: hello.

调用函数

quicktemplate支持在模板中调用模板函数、标准库的函数。由于qtc会直接生成 Go 代码,我们甚至还可以在同目录下编写自己的函数给模板调用,模板 A 中也可以调用模板 B 中定义的函数。

我们先在templates目录下编写一个文件rank.go,定义一个Rank函数,传入分数,返回评级:

  1. package templates
  2. func Rank(score int) string {
  3. if score >= 90 {
  4. return "A"
  5. } else if score >= 80 {
  6. return "B"
  7. } else if score >= 70 {
  8. return "C"
  9. } else if score >= 60 {
  10. return "D"
  11. } else {
  12. return "E"
  13. }
  14. }

然后我们可以在模板中调用这个函数:

  1. {% import "fmt" %}
  2. {% func ScoreList(name2score map[string]int) %}
  3. {% for name, score := range name2score %}
  4. {%s fmt.Sprintf("%s: score-%d rank-%s", name, score, Rank(score)) %}
  5. {% endfor %}
  6. {% endfunc %}

编译模板:

  1. $ qtc

编写程序:

  1. func main() {
  2. name2score := make(map[string]int)
  3. name2score["dj"] = 85
  4. name2score["lizi"] = 96
  5. name2score["hjw"] = 52
  6. fmt.Println(templates.ScoreList(name2score))
  7. }

运行程序输出:

  1. $ go run .
  2. dj: score-85 rank-B
  3. lizi: score-96 rank-A
  4. hjw: score-52 rank-E

由于我们在模板中用到fmt包,需要先使用{% import %}将该包导入。

在模板中调用另一个模板的函数也是类似的,因为模板最终都会转为 Go 代码。Go 代码中有同样签名的函数。

Web

quicktemplate常用来编写 HTML 页面的模板:

  1. {% func Index(name string) %}
  2. <html>
  3. <head>
  4. <title>Awesome Web</title>
  5. </head>
  6. <body>
  7. <h1>Hi, {%s name %}
  8. <p>Welcome to the awesome web!!!</p>
  9. </body>
  10. </html>
  11. {% endfunc %}

下面编写一个简单的 Web 服务器:

  1. func index(w http.ResponseWriter, r *http.Request) {
  2. templates.WriteIndex(w, r.FormValue("name"))
  3. }
  4. func main() {
  5. mux := http.NewServeMux()
  6. mux.HandleFunc("/", index)
  7. server := &http.Server{
  8. Handler: mux,
  9. Addr: ":8080",
  10. }
  11. log.Fatal(server.ListenAndServe())
  12. }

qtc会生成一个Write*的方法,它接受一个io.Writer的参数。将模板渲染的结果写入这个io.Writer中,我们可以直接将http.ResponseWriter作为参数传入,非常便捷。

运行:

  1. $ qtc
  2. $ go run .

浏览器输入localhost:8080?name=dj查看结果。

总结

quicktemplate至少有下面 3 个优势:

  • 语法与 Go 语言非常类似,几乎没有学习成本;
  • 会先转换为 Go,渲染速度非常快,比标准库html/template快 20 倍以上;
  • 为了安全考虑,会执行一些编码,避免受到攻击。

从我个人的实际使用情况来看,确实很方便,很实用。感兴趣的还可以去看看qtc生成的 Go 代码。

大家如果发现好玩、好用的 Go 语言库,欢迎到 Go 每日一库 GitHub 上提交 issue😄

参考

  1. quicktemplate GitHub:https://github.com/valyala/quicktemplate
  2. Go 每日一库 GitHub:https://github.com/go-quiz/go-daily-lib