首页 > Golang > Go问答

使用 Go 语言和 wkhtmltopdf 将数据注入页脚

来源：stackoverflow

时间：2024-02-06 13:33:24 439浏览收藏

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《使用 Go 语言和 wkhtmltopdf 将数据注入页脚》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

问题内容

我正在使用 go-wkhtmltopdf 包通过 go lang 生成 pdf。我使用下面的代码在 pdf 的每个页面上显示页脚。我想将一些动态数据传递到页脚以显示相同的数据。我没有找到任何方法可以做到这一点。请帮忙

"bytes"
"encoding/base64"
"fmt"
htmlTemplate "html/template"
"path"
"strings"
textTemplate "text/template"
wkhtml "github.com/SebastiaanKlippert/go-wkhtmltopdf"

pdfg, err := wkhtml.NewPDFGenerator()
if err != nil {
    return nil, err
}
page := wkhtml.NewPageReader(strings.NewReader(util.FilterSmartQuotesEtc(html)))
page.FooterHTML.Set("templates/footer.html")
pdfg.AddPage(page)
err = pdfg.Create()

templates/footer.html 是我要传递数据的页脚文件。非常感谢任何帮助。

正确答案

html 代码中的“动态数据”建议使用 HTLM templates（数据驱动模板，用于生成安全的 html 输出，防止代码注入）

您可以使用模板 ParseFiles() 从页脚获取该模板，并注入动态数据。（参见“Golang HTML Template ParseFiles and Execute”作为示例）

假设您有一个结构来保存动态数据，例如：

type footerdata struct {
    id string
}

并假设您的 templates/footer.html 有一个（非常）简单的模板：

<!doctype html>
<html>
<head>
    <meta charset="utf-8">
    <style>
        /* your footer styles */
    </style>
</head>
<body>
    <div>
        id: {{.id}}
    </div>
</body>
</html>

您可以创建一个函数，将渲染的 html 保存到临时文件并返回文件路径（用作 page.footerhtml.set(footerhtml) 的 url）：

import (
    "io/ioutil"
    //...
)

func renderfooterhtml(templatepath string, data *footerdata) (string, error) {
    tmpl, err := htmltemplate.parsefiles(templatepath)
    if err != nil {
        return "", err
    }

    var buf bytes.buffer
    err = tmpl.execute(&buf, data)
    if err != nil {
        return "", err
    }

    // save the rendered html to a temporary file
    tmpfile, err := ioutil.tempfile("", "footer-*.html")
    if err != nil {
        return "", err
    }

    if _, err := tmpfile.write(buf.bytes()); err != nil {
        return "", err
    }
    if err := tmpfile.close(); err != nil {
        return "", err
    }

    return tmpfile.name(), nil
}

然后修改现有代码以使用呈现的页脚 html：

import (
    "bytes"
    "encoding/base64"
    "fmt"
    htmlTemplate "html/template"
    "path"
    "strings"
    textTemplate "text/template"
    wkhtml "github.com/SebastiaanKlippert/go-wkhtmltopdf"
)

// ...

pdfg, err := wkhtml.NewPDFGenerator()
if err != nil {
    return nil, err
}

page := wkhtml.NewPageReader(strings.NewReader(util.FilterSmartQuotesEtc(html)))

// Render the footer HTML with your dynamic data
footerData := &FooterData{
    ID: "Your dynamic ID here",
}
footerFilePath, err := renderFooterHTML("templates/footer.html", footerData)
if err != nil {
    return nil, err
}

// Use the rendered footer HTML file
page.FooterHTML.Set(footerFilePath)

pdfg.AddPage(page)
err = pdfg.Create()

//...

// Delete the temporary footer file when you're done with it
os.Remove(footerFilePath)

请务必将 id: "yourdynamic id here" 中的字符串替换为您的实际动态 id 值。

到这里，我们也就讲完了《使用 Go 语言和 wkhtmltopdf 将数据注入页脚》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！

声明：本文转载于：stackoverflow 如有侵犯，请联系study_golang@163.com删除