go - 优雅地处理 gin-gonic 中的模板渲染错误

Question

我正在学习 Go 并将 gin-gonic 用于网络应用程序。我正在尝试从模板错误中优雅地恢复，但一直无法弄清楚如何缓冲输出或正确重定向以实现此目的。

使用这段代码:

package main

import (
    "net/http"

    "github.com/gin-gonic/gin"
)

func main() {
    g := gin.New()
    g.LoadHTMLGlob("templates/*")
    g.Use(func(c *gin.Context) {
        defer func() {
            if err := recover(); err != nil {
                c.HTML(http.StatusInternalServerError, "error.tmpl", nil)
            }
        }()

        c.Next()
    })
    g.GET("/", func(c *gin.Context) {
        c.HTML(http.StatusOK, "index.tmpl", gin.H{"var": 4})
    })
    g.Run(":80")
}

templates/index.tmpl 在哪里

Before
<br>
Bad: {{.var.x}}
<br>
After

和 templates/error.tmpl 是

Oops! We encountered an error.

当我加载我的页面时，我看到了

Before
Bad: Oops! We encountered an error.

响应代码最终为 200。我更愿意清楚地捕获错误，以便向用户显示的唯一内容是

Oops! We encountered an error.

响应代码为 500，错误会记录在服务器上以供日后调查。

在 gin 中捕获模板错误而不向用户显示部分输出的最佳方法是什么？我已经看到几个通过使用内置 net/http 东西进行缓冲来完成此操作的示例，但是还没有找到任何在 gin 中处理它的好方法。

用解决方案编辑

根据@big pigeon 对已接受答案的评论，我最终自己将模板执行到一个缓冲区中，并在没有错误的情况下使用 c.Data() 显示它。这似乎仍然不太理想，因为它绕过了 multitemplate 在开发构建运行时动态重新加载已解析模板的功能，但它确实有效。更新后的概念验证代码如下所示:

package main

import (
    "bytes"
    "html/template"
    "net/http"

    "github.com/gin-gonic/gin"
)

func main() {
    g := gin.New()
    g.LoadHTMLGlob("templates/*")
    g.Use(func(c *gin.Context) {
        defer func() {
            if err := recover(); err != nil {
                c.HTML(http.StatusInternalServerError, "error.tmpl", nil)
            }
        }()

        c.Next()
    })
    g.GET("/", func(c *gin.Context) {
        if tmpl, err := template.ParseFiles("templates/index.tmpl"); err != nil {
            panic(err)
        } else {
            buf := &bytes.Buffer{}
            if err = tmpl.Execute(buf, gin.H{"var": 4}); err != nil {
                panic(err)
            } else {
                c.Data(http.StatusOK, "text/html; charset=utf-8", buf.Bytes())
            }
        }
    })
    g.Run(":80")
}

使用缓冲池、预解析模板和其他类似的技巧留给 future 的读者作为练习。

如果有人知道在不绕过 gin 的解析/渲染功能的情况下处理这个问题的更好方法，我很乐意接受。

Answer 1

您必须确保模板正确呈现，因为 c.HTML 将直接写入响应，此时一些字节已发送到客户端。

你可以使用“html/template”，并使用buff来缓存响应数据，而不是直接写入响应编写器