36 lines
895 B
Go
36 lines
895 B
Go
|
package pdf
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/SebastiaanKlippert/go-wkhtmltopdf"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// html转 Pdf dataType = 1 为网页url dataType = 2为网页内容 outFile为空则不保存
|
||
|
|
func HtmlToPdf(content string, dataType int, outFile string) ([]byte, error) {
|
||
|
|
pdfg, err := wkhtmltopdf.NewPDFGenerator()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
switch dataType {
|
||
|
|
case 1: //网页地址
|
||
|
|
pdfg.AddPage(wkhtmltopdf.NewPage(content))
|
||
|
|
case 2: //网页内容
|
||
|
|
pdfg.AddPage(wkhtmltopdf.NewPageReader(strings.NewReader(content)))
|
||
|
|
}
|
||
|
|
//模式
|
||
|
|
pdfg.Orientation.Set(wkhtmltopdf.OrientationPortrait)
|
||
|
|
//pdf尺寸默认A4纸
|
||
|
|
pdfg.PageSize.Set(wkhtmltopdf.PageSizeA4)
|
||
|
|
// Create PDF document in memory
|
||
|
|
if err = pdfg.Create(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
// Write PDF to file
|
||
|
|
if outFile != "" {
|
||
|
|
if err = pdfg.WriteFile(outFile); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return pdfg.Bytes(), nil
|
||
|
|
}
|