1

utils/pdf/html_to_pdf.go

@@ -1,15 +1,16 @@
 package pdf
 
 import (
+	"encoding/base64"
 	"github.com/SebastiaanKlippert/go-wkhtmltopdf"
 	"strings"
 )
 
 // html转 Pdf dataType = 1 为网页url dataType = 2为网页内容 outFile为空则不保存
-func HtmlToPdf(content string, dataType int, outFile ...string) ([]byte, error) {
+func HtmlToPdfBase64(content string, dataType int, outFile ...string) (string, error) {
 	pdfg, err := wkhtmltopdf.NewPDFGenerator()
 	if err != nil {
-		return nil, err
+		return "", err
 	}
 	switch dataType {
 	case 1: //网页地址
@@ -23,13 +24,13 @@ func HtmlToPdf(content string, dataType int, outFile ...string) ([]byte, error)
 	pdfg.PageSize.Set(wkhtmltopdf.PageSizeA4)
 	// Create PDF document in memory
 	if err = pdfg.Create(); err != nil {
-		return nil, err
+		return "", err
 	}
 	// Write PDF to file
 	if len(outFile) != 0 {
 		if err = pdfg.WriteFile(outFile[0]); err != nil {
-			return nil, err
+			return "", err
 		}
 	}
-	return pdfg.Bytes(), nil
+	return base64.StdEncoding.EncodeToString(pdfg.Bytes()), nil
 }