压力测试

server.js

@@ -25,6 +25,14 @@ app.post('/api/generate_pdf', async (req, res) => {
     }
 
     const brandText = showWatermark !== false ? 'Powered by XWX AI Chat Exporter' : '';
+    
+    // Log HTML size for debugging
+    const htmlSizeMB = (html.length / 1024 / 1024).toFixed(2);
+    console.log(`[PDF-GEN] Received HTML: ${htmlSizeMB} MB`);
+    
+    // Count base64 images in HTML
+    const base64Matches = html.match(/data:image\/[^;]+;base64,/g);
+    console.log(`[PDF-GEN] Found ${base64Matches ? base64Matches.length : 0} base64 images in HTML`);
 
     // Launch Chrome
     // We use the installed 'chromium' from apt-get
@@ -34,15 +42,62 @@ app.post('/api/generate_pdf', async (req, res) => {
         '--no-sandbox',
         '--disable-setuid-sandbox',
         '--disable-dev-shm-usage', // Critical for Docker
-        '--font-render-hinting=none' // Better font rendering
+        '--font-render-hinting=none', // Better font rendering
+        '--disable-gpu', // Disable GPU for stability in Docker
+        '--disable-software-rasterizer',
+        '--memory-pressure-off'
       ],
       headless: 'new'
     });
 
     const page = await browser.newPage();
     
-    // Set content
-    await page.setContent(html, { waitUntil: 'networkidle0' });
+    // Set viewport for consistent rendering
+    await page.setViewport({ width: 1200, height: 800 });
+    
+    // Set content with longer timeout for large HTML
+    // Use 'load' instead of 'networkidle0' because base64 images don't trigger network requests
+    console.log('[PDF-GEN] Setting page content...');
+    await page.setContent(html, { 
+      waitUntil: 'load',
+      timeout: 120000 // 2 minutes for large content
+    });
+    console.log('[PDF-GEN] Page content loaded');
+    
+    // Wait for images to fully render (additional 3 seconds for base64 decoding)
+    console.log('[PDF-GEN] Waiting for images to render...');
+    await page.waitForTimeout(3000);
+    console.log('[PDF-GEN] Image render wait complete');
+    
+    // Count images in the rendered page
+    const pageImageCount = await page.evaluate(() => {
+      return document.querySelectorAll('img').length;
+    });
+    console.log(`[PDF-GEN] Rendered page contains ${pageImageCount} img elements`);
+    
+    // Scroll to bottom to ensure all lazy-loaded images are visible
+    console.log('[PDF-GEN] Scrolling to ensure all images are loaded...');
+    await page.evaluate(async () => {
+      await new Promise((resolve) => {
+        let totalHeight = 0;
+        const distance = 100;
+        const timer = setInterval(() => {
+          const scrollHeight = document.body.scrollHeight;
+          window.scrollBy(0, distance);
+          totalHeight += distance;
+          
+          if (totalHeight >= scrollHeight) {
+            clearInterval(timer);
+            window.scrollTo(0, 0); // Scroll back to top
+            resolve();
+          }
+        }, 50);
+      });
+    });
+    console.log('[PDF-GEN] Scroll complete');
+    
+    // Wait additional time for any lazy-loaded images
+    await page.waitForTimeout(2000);
 
     // Inject styles to ensure 100% width and print media simulation
     await page.addStyleTag({
@@ -74,6 +129,9 @@ app.post('/api/generate_pdf', async (req, res) => {
 
     await browser.close();
     browser = null;
+    
+    const pdfSizeMB = (pdfBuffer.length / 1024 / 1024).toFixed(2);
+    console.log(`[PDF-GEN] PDF generated successfully: ${pdfSizeMB} MB`);
 
     // Send response
     res.setHeader('Content-Type', 'application/pdf');