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
}),
}