golang文档转pdf

PHP中文网

2023-09-20 帮助9人

安装依赖

首先，我们需要安装两个库来完成文件转换和PDF生成的任务。这两个库分别是：

github.com/SebastiaanKlippert/go-wkhtmltopdf：这个库是一个go语言封装的wkhtmltopdf的库，用于将html格式文件转换成PDF格式。
github.com/unidoc/unioffice: 这个库是一个golang的office文件读写库，支持多种类型的office文件，包括doc, docx, xls, xlsx, ppt, pptx等。

我们可以使用go mod命令来安装这两个库：

go get -u github.com/SebastiaanKlippert/go-wkhtmltopdf
go get -u github.com/unidoc/unioffice

实现文档转HTML

现在我们已经准备好了两个库，接下来就是实现文档转HTML的功能。我们将以docx文件为例。使用unioffice库可以轻松地读取docx文档。下面是一个简单的实现代码：

package main

import (
    "log"
    "github.com/unidoc/unioffice/document"
)

func DocxToHtml(inputFilePath string) (string, error) {
    doc, err := document.Open(inputFilePath)
    if err != nil {
        return "", err
    }
    defer func() {
        err := doc.Close()
        if err != nil {
            log.Fatalf("unable to close document: %s", err)
        }
    }()

    html, err := doc.Html()
    if err != nil {
        return "", err
    }
    
    return html, nil
}

这个函数的作用是将输入的docx文件转换成HTML字符串。函数的输入参数是docx文件的路径，输出是HTML字符串和一个错误值。在这个函数中，我们首先使用document.Open函数打开docx文件，然后使用doc.Html函数将文件内容转换成HTML字符串。最后，将HTML字符串作为输出值返回。

实现HTML转PDF

现在我们已经完成了将docx文件转换成HTML的功能，下一步是将HTML转换成PDF。我们将使用go-wkhtmltopdf库来实现这个功能。这个库是一个go语言封装的wkhtmltopdf的库，可以很容易地将HTML格式的文档转换成PDF格式。下面是一个简单的实现代码：

package main

import (
    "io/ioutil"
    "strings"
    "github.com/SebastiaanKlippert/go-wkhtmltopdf"
)

func HtmlToPdf(html string, outputFilePath string) error {
    err := ioutil.WriteFile("input.html", []byte(html), 0644)
    if err != nil {
        return err
    }

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

    pdfg.AddPage(wkhtmltopdf.NewPageReader(strings.NewReader(html)))
    pdfg.Dpi.Set(300)
    pdfg.Orientation.Set(wkhtmltopdf.OrientationPortrait)
    pdfg.PageSize.Set(wkhtmltopdf.PageSizeA4)
    err = pdfg.Create()
    if err != nil {
        return err
    }

    err = pdfg.WriteFile(outputFilePath)
    if err != nil {
        return err
    }

    return nil
}

这个函数的作用是将输入的HTML字符串转换成PDF文件。函数的输入参数是HTML字符串和输出PDF文件的路径，输出是一个错误值。在函数中，我们首先将HTML字符串写入到一个文件中，然后使用go-wkhtmltopdf库来将HTML文件转换成PDF文件。在调用AddPage函数时，我们将HTML文件作为一个Reader对象传递进去。对于输出的PDF文件，我们可以使用pdfg.WriteFile函数将其写入到指定的路径中。

完整代码

现在我们已经完成了文档转换成PDF的两个主要步骤。下面是一个完整的例子代码：

package main

import (
    "io/ioutil"
    "log"
    "os"
    "strings"
    "github.com/SebastiaanKlippert/go-wkhtmltopdf"
    "github.com/unidoc/unioffice/document"
)

func main() {
    inputFilePath := "input.docx"
    outputFilePath := "output.pdf"
    html, err := DocxToHtml(inputFilePath)
    if err != nil {
        log.Fatalf("unable to convert docx to html: %s", err)
    }

    err = HtmlToPdf(html, outputFilePath)
    if err != nil {
        log.Fatalf("unable to convert html to pdf: %s", err)
    }

    err = os.Remove("input.html")
    if err != nil {
        log.Fatalf("unable to delete input.html: %s", err)
    }
}

func DocxToHtml(inputFilePath string) (string, error) {
    doc, err := document.Open(inputFilePath)
    if err != nil {
        return "", err
    }
    defer func() {
        err := doc.Close()
        if err != nil {
            log.Fatalf("unable to close document: %s", err)
        }
    }()

    html, err := doc.Html()
    if err != nil {
        return "", err
    }
    
    return html, nil
}

func HtmlToPdf(html string, outputFilePath string) error {
    err := ioutil.WriteFile("input.html", []byte(html), 0644)
    if err != nil {
        return err
    }

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

    pdfg.AddPage(wkhtmltopdf.NewPageReader(strings.NewReader(html)))
    pdfg.Dpi.Set(300)
    pdfg.Orientation.Set(wkhtmltopdf.OrientationPortrait)
    pdfg.PageSize.Set(wkhtmltopdf.PageSizeA4)
    err = pdfg.Create()
    if err != nil {
        return err
    }

    err = pdfg.WriteFile(outputFilePath)
    if err != nil {
        return err
    }

    return nil
}

在这个例子中，我们定义了两个函数：DocxToHtml和HtmlToPdf。DocxToHtml函数的作用是将输入的docx文件转换成HTML字符串；HtmlToPdf函数的作用是将HTML字符串转换成PDF文件。在main函数中，我们首先调用DocxToHtml函数将docx文件转换成HTML字符串，然后再调用HtmlToPdf函数将HTML字符串转换成PDF文件。最后，我们将生成的中间文件input.html删除。

总结

这篇好文章是转载于：学新通技术网

golang文档转pdf

photoshop保存的图片太大微信发不了怎么办

《学习通》视频自动暂停处理方法

Android 11 保存文件到外部存储，并分享文件

word里面弄一个表格后上面的标题会跑到下面怎么办

photoshop扩展功能面板显示灰色怎么办

微信公众号没有声音提示怎么办

excel下划线不显示怎么办

excel打印预览压线压字怎么办

TikTok加速器哪个好免费的TK加速器推荐

怎样阻止微信小程序自动打开