记载一次wkhtmltopdf生成pdf造成oom题目

题目现象
有个生成pdf的接口，到处pdf为空；docker环境必现，但是本地环境无法复现
1、代码增长各处错误判断，发现docker环境调用接口出现
Loading pages (1/6)
[>                                                           ] 0%
[======>                                                     ] 10%
[======================>                                     ] 37%
[========================>                                   ] 41%
[==============================>                             ] 50%
[====================================>                       ] 61%
[======================================================>     ] 90%
signal: killed

2、查看docker系统日记
dmesg|grep "kill"
发现oom

查看资料后显示wkhtmltopdf创建pdf会造成高内存分配
优化pdf参数，题目得到解决

2. // Configure PDF generator with memory-saving options
3.     pdfg, err := wkhtmltopdf.NewPDFGenerator()
4.     if err != nil {
5.         return nil, fmt.Errorf("PDF generator initialization failed: %w", err)
6.     }
8.     pdfg.PageSize.Set(wkhtmltopdf.PageSizeA4)
9.     pdfg.Dpi.Set(96) // 1、设置dpi，降低内存使用
10.     pdfg.NoCollate.Set(true)  // Disable collation to reduce memory
11.     pdfg.Grayscale.Set(false)
12.    
13.     // Configure page with memory optimization settings
14.     page := wkhtmltopdf.NewPageReader(r)
15.     page.LoadErrorHandling.Set("ignore")
16.     page.JavascriptDelay.Set(2000) // Allow time for JS to execute
17.     pdfg.AddPage(page)
19.     // Set environment variables for memory constraints
20.     os.Setenv("QTWEBKIT_SETTINGS", "MaximumPages=1") //2、限制同时处理的页数，避免内存峰值
21.     os.Setenv("QTWEBKIT_MAX_MEMORY", "256M") // Limit WebKit memory usage
23.     // Create PDF with timeout
24.     // 3、增加超时判断
25.     pdfCtx, pdfCancel := context.WithTimeout(context.Background(), 5*time.Minute)
26.     defer pdfCancel()
28.     errChan := make(chan error, 1)
29.     go func() {
30.         defer close(errChan)
31.         errChan <- pdfg.Create()
32.     }()
34.     select {
35.     case err := <-errChan:
36.         if err != nil {
37.             return nil, fmt.Errorf("PDF generation failed: %w", err)
38.         }
39.     case <-pdfCtx.Done():
40.         return nil, fmt.Errorf("PDF generation timed out")
41.     }
43.     // Return the generated PDF
44.     return pdfg.Bytes(), nil

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

继续阅读请点击广告