diff --git a/internal/shared/pdf/html_pdf_generator.go b/internal/shared/pdf/html_pdf_generator.go index f9bb761..58f5bbe 100644 --- a/internal/shared/pdf/html_pdf_generator.go +++ b/internal/shared/pdf/html_pdf_generator.go @@ -49,8 +49,11 @@ func (g *HTMLPDFGenerator) GenerateFromURL(ctx context.Context, url string) ([]b chromedp.WaitVisible(".page", chromedp.ByQuery), chromedp.ActionFunc(func(ctx context.Context) error { g.logger.Info("开始通过 headless Chrome 生成企业报告 PDF", zap.String("url", url)) - var err error - pdfBuf, err = page.PrintToPDF(). + var ( + buf []byte + err error + ) + buf, _, err = page.PrintToPDF(). WithPrintBackground(true). WithPaperWidth(8.27). // A4 宽度(英寸 -> 约 210mm) WithPaperHeight(11.69). // A4 高度(英寸 -> 约 297mm) @@ -59,6 +62,9 @@ func (g *HTMLPDFGenerator) GenerateFromURL(ctx context.Context, url string) ([]b WithMarginLeft(0.4). WithMarginRight(0.4). Do(ctx) + if err == nil { + pdfBuf = buf + } return err }), }