Răsfoiți Sursa

feat: 新增限制

zhangyuhan 3 ani în urmă
părinte
comite
3a5c5e42ef
1 a modificat fișierele cu 14 adăugiri și 9 ștergeri
  1. 14 9
      bin/create-pdf/pdf.js

+ 14 - 9
bin/create-pdf/pdf.js

@@ -62,17 +62,22 @@ async function createPDFOfURL (url, config = {}) {
     headless: true,
     args: ['--no-sandbox', '--disable-setuid-sandbox']
   });
-  const page = await browser.newPage();
-  await page.setDefaultNavigationTimeout(0)
-  await page.goto(url, {waitUntil: 'networkidle2'});
-  await wait(config.delay || 1000)
-  logger.log.debug('等待生成', new Date().toLocaleString())
+  try {
+    const page = await browser.newPage();
+    await page.setDefaultNavigationTimeout(0)
+    await page.goto(url, {waitUntil: 'networkidle2'});
+    await wait(config.delay || 1000)
+    logger.log.debug('等待生成', new Date().toLocaleString())
 
-  await page.pdf(outConfig);
-  await browser.close();
+    await page.pdf(outConfig);
+    await browser.close();
 
-  logger.log.debug('当前时间', new Date().toLocaleString())
-  return outConfig.path
+    logger.log.debug('当前时间', new Date().toLocaleString())
+    return outConfig.path
+  } catch (e) {
+    logger.log.debug(e)
+    await browser.close();
+  }
 }
 
 module.exports = createPDFOfURL