team/hyapi-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

f

Browse Source
This commit is contained in:
liangzai
2026-06-12 15:54:40 +08:00
parent 0478674a55
commit 18f7c26539
7 changed files with 496 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
internal/infrastructure/http/handlers/qygl_report_handler.go
View File

@@ -19,6 +19,7 @@ import (
"hyapi-server/internal/domains/api/services/processors"
"hyapi-server/internal/domains/api/services/processors/qygl/reportstore"
"hyapi-server/internal/shared/pdf"
"hyapi-server/internal/shared/qyglreport"
)
// QYGLReportHandler 企业全景报告页面渲染处理器
@@ -97,9 +98,7 @@ func (h *QYGLReportHandler) GetQYGLReportPage(c *gin.Context) {
reportJSON := template.JS(reportJSONBytes)
c.HTML(http.StatusOK, "qiye.html", gin.H{
"ReportJSON": reportJSON,
})
c.HTML(http.StatusOK, "qiye.html", qyglreport.PageTemplateData(reportJSON))
}
// GetQYGLReportPageByID 通过编号查看企业全景报告页面
@@ -115,9 +114,7 @@ func (h *QYGLReportHandler) GetQYGLReportPageByID(c *gin.Context) {
if h.reportRepo != nil {
if entity, err := h.reportRepo.FindByReportID(c.Request.Context(), id); err == nil && entity != nil {
h.maybeScheduleQYGLPDFPregen(c.Request.Context(), id)
c.HTML(http.StatusOK, "qiye.html", gin.H{
"ReportJSON": template.JS(entity.ReportData),
})
c.HTML(http.StatusOK, "qiye.html", qyglreport.PageTemplateData(template.JS(entity.ReportData)))
return
}
}
@@ -140,9 +137,7 @@ func (h *QYGLReportHandler) GetQYGLReportPageByID(c *gin.Context) {
reportJSON := template.JS(reportJSONBytes)
c.HTML(http.StatusOK, "qiye.html", gin.H{
"ReportJSON": reportJSON,
})
c.HTML(http.StatusOK, "qiye.html", qyglreport.PageTemplateData(reportJSON))
}
// qyglReportExists 报告是否仍在库或本进程内存中（用于决定是否补开预生成）

137
internal/shared/pdf/html_pdf_generator.go
View File

@@ -10,6 +10,12 @@ import (
"go.uber.org/zap"
)
// A4 纵向 @ 96dpi，与 CSS @page / 报告页 max-width 对齐
const (
htmlPDFViewportWidth = 794
htmlPDFViewportHeight = 1123
)
// HTMLPDFGenerator 使用 headless Chrome 将 HTML 页面渲染为 PDF
type HTMLPDFGenerator struct {
logger *zap.Logger
@@ -26,41 +32,54 @@ func NewHTMLPDFGenerator(logger *zap.Logger) *HTMLPDFGenerator {
}
// GenerateFromURL 使用 headless Chrome 打开指定 URL，并导出为 PDF 字节流
// 这里固定使用 A4 纵向纸张，开启背景打印
// 流程：A4 视口渲染 → 等待报告 DOM → 滚动触发展开 → 切换 print 媒体 → PrintToPDF（边距由 CSS @page 控制）
func (g *HTMLPDFGenerator) GenerateFromURL(ctx context.Context, url string) ([]byte, error) {
if ctx == nil {
ctx = context.Background()
}
// 整个生成过程增加超时时间，避免长时间卡死
timeoutCtx, cancel := context.WithTimeout(ctx, 60*time.Second)
timeoutCtx, cancel := context.WithTimeout(ctx, 90*time.Second)
defer cancel()
// 创建 Chrome 上下文（使用系统默认的 headless Chrome/Chromium）
chromeCtx, cancelChrome := chromedp.NewContext(timeoutCtx)
opts := append(chromedp.DefaultExecAllocatorOptions[:],
chromedp.Flag("headless", true),
chromedp.Flag("disable-gpu", true),
chromedp.Flag("no-sandbox", true),
chromedp.Flag("disable-dev-shm-usage", true),
)
allocCtx, allocCancel := chromedp.NewExecAllocator(timeoutCtx, opts...)
defer allocCancel()
chromeCtx, cancelChrome := chromedp.NewContext(allocCtx)
defer cancelChrome()
var pdfBuf []byte
tasks := chromedp.Tasks{
chromedp.EmulateViewport(htmlPDFViewportWidth, htmlPDFViewportHeight),
chromedp.Navigate(url),
// 等待页面主体和报告容器就绪，确保数据渲染完成
chromedp.WaitReady("body", chromedp.ByQuery),
chromedp.WaitVisible(".page", chromedp.ByQuery),
waitReportReadyForPDF(),
waitReportPosterImages(),
scrollPageForPrintLayout(),
chromedp.ActionFunc(func(ctx context.Context) error {
return chromedp.Evaluate(`document.body.classList.add('pdf-export')`, nil).Do(ctx)
}),
chromedp.ActionFunc(func(ctx context.Context) error {
time.Sleep(150 * time.Millisecond)
return nil
}),
chromedp.ActionFunc(func(ctx context.Context) error {
g.logger.Info("开始通过 headless Chrome 生成企业报告 PDF", zap.String("url", url))
var (
buf []byte
err error
)
buf, _, err = page.PrintToPDF().
buf, _, err := page.PrintToPDF().
WithPrintBackground(true).
WithPaperWidth(8.27). // A4 宽度（英寸 -> 约 210mm）
WithPaperHeight(11.69). // A4 高度（英寸 -> 约 297mm）
WithMarginTop(0.4).
WithMarginBottom(0.4).
WithMarginLeft(0.4).
WithMarginRight(0.4).
WithPaperWidth(8.27).
WithPaperHeight(11.69).
WithMarginTop(0.39).
WithMarginBottom(0.39).
WithMarginLeft(0.47).
WithMarginRight(0.47).
Do(ctx)
if err == nil {
pdfBuf = buf
@@ -86,3 +105,87 @@ func (g *HTMLPDFGenerator) GenerateFromURL(ctx context.Context, url string) ([]b
return pdfBuf, nil
}
// waitReportReadyForPDF 等待企业报告脚本渲染完成（qiye.html 设置 data-qygl-report-ready）
func waitReportReadyForPDF() chromedp.Action {
return chromedp.ActionFunc(func(ctx context.Context) error {
deadline := time.Now().Add(45 * time.Second)
for {
if ctx.Err() != nil {
return ctx.Err()
}
var ready string
_ = chromedp.Evaluate(`document.body && document.body.dataset.qyglReportReady || ''`, &ready).Do(ctx)
if ready == "1" {
return nil
}
var sectionCount float64
_ = chromedp.Evaluate(`document.querySelectorAll('#reportSections .section-card, .report-sections .section-block').length`, &sectionCount).Do(ctx)
if sectionCount > 0 {
return nil
}
if time.Now().After(deadline) {
return fmt.Errorf("等待报告内容渲染超时")
}
time.Sleep(200 * time.Millisecond)
}
})
}
func waitReportPosterImages() chromedp.Action {
return chromedp.ActionFunc(func(ctx context.Context) error {
deadline := time.Now().Add(20 * time.Second)
for {
if ctx.Err() != nil {
return ctx.Err()
}
var pending float64
js := `(function(){
var imgs = document.querySelectorAll('img.header-poster, img.footer-poster');
if (!imgs.length) return 0;
for (var i = 0; i < imgs.length; i++) {
var img = imgs[i];
if (!img.complete || img.naturalWidth === 0) return 1;
}
return 0;
})()`
if err := chromedp.Evaluate(js, &pending).Do(ctx); err != nil {
return err
}
if pending == 0 {
return nil
}
if time.Now().After(deadline) {
return fmt.Errorf("等待报告海报图片加载超时")
}
time.Sleep(150 * time.Millisecond)
}
})
}
func scrollPageForPrintLayout() chromedp.Action {
return chromedp.ActionFunc(func(ctx context.Context) error {
var height float64
if err := chromedp.Evaluate(`Math.max(document.body.scrollHeight, document.documentElement.scrollHeight)`, &height).Do(ctx); err != nil {
return err
}
step := float64(900)
if height < step {
step = height
}
if step <= 0 {
return nil
}
for y := float64(0); y <= height; y += step {
js := fmt.Sprintf(`window.scrollTo(0, %f);`, y)
if err := chromedp.Evaluate(js, nil).Do(ctx); err != nil {
return err
}
time.Sleep(120 * time.Millisecond)
}
if err := chromedp.Evaluate(`window.scrollTo(0, 0);`, nil).Do(ctx); err != nil {
return err
}
time.Sleep(300 * time.Millisecond)
return nil
})
}

77
internal/shared/qyglreport/poster_assets.go Normal file
View File

@@ -0,0 +1,77 @@
package qyglreport
import (
"encoding/base64"
"html/template"
"mime"
"os"
"path/filepath"
"sync"
"github.com/gin-gonic/gin"
)
const (
defaultBannerRel = "resources/qygl-media/banner.jpg"
defaultFooterRel = "resources/qygl-media/footer.png"
)
var (
loadOnce sync.Once
bannerURL template.URL
footerURL template.URL
)
func loadPosterAssets() {
bannerURL = dataURLFromFile(resolveMediaFile([]string{
defaultBannerRel,
"resources/qygl-media/banner.png",
}), "/reports/qygl/media/banner.jpg")
footerURL = dataURLFromFile(defaultFooterRel, "/reports/qygl/media/footer.png")
}
func resolveMediaFile(candidates []string) string {
for _, p := range candidates {
if _, err := os.Stat(p); err == nil {
return p
}
}
if len(candidates) > 0 {
return candidates[0]
}
return ""
}
func dataURLFromFile(path string, httpFallback string) template.URL {
if path == "" {
return template.URL(httpFallback)
}
data, err := os.ReadFile(path)
if err != nil {
return template.URL(httpFallback)
}
ext := filepath.Ext(path)
mimeType := mime.TypeByExtension(ext)
if mimeType == "" {
switch ext {
case ".jpg", ".jpeg":
mimeType = "image/jpeg"
case ".png":
mimeType = "image/png"
default:
mimeType = "application/octet-stream"
}
}
encoded := base64.StdEncoding.EncodeToString(data)
return template.URL("data:" + mimeType + ";base64," + encoded)
}
// PageTemplateData 企业报告 HTML 模板数据（海报内嵌 data URL，PDF 生成不依赖外链）
func PageTemplateData(reportJSON template.JS) gin.H {
loadOnce.Do(loadPosterAssets)
return gin.H{
"ReportJSON": reportJSON,
"BannerPosterURL": bannerURL,
"FooterPosterURL": footerURL,
}
}