const browser = await puppeteer.launch({
    executablePath: 'google-chrome-stable',
    headless: true,
    args: ['--disable-setuid-sandbox', '--no-sandbox']
  })
 // 打開瀏覽器
  const context = await browser.createIncognitoBrowserContext() // 開啟無痕模式
  const page = await context.newPage() // 打開一個空白頁
  await page.goto('url', { timeout: 3000 })
  await page.waitForSelector('.report-pages.load-finished', { timeout: 60000 })// 等待頁面加載完成
  const bufferStr = await page.pdf({
    scale: 1,
    width: ctx.request.body.width,
    height: ctx.request.body.height + 1, // 加1，解決多生成一個空白頁
    // CSS
    preferCSSPageSize: true,
    // 開啟渲染背景色，因為 puppeteer 是基于 chrome 瀏覽器的，瀏覽器為了打印節(jié)省油墨，默認是不導(dǎo)出背景圖及背景色的 
   // 坑點，必須加
    printBackground: true
    // margin:{top:'2cm',right:'2cm',bottom:'2cm',left:'2cm'}
  })

const handlePool = (urls, max, handler) => { 
 let i = 0  const ret = [] // 存儲所有的異步任務(wù)  
 const executing = [] // 存儲正在執(zhí)行的異步任務(wù)  
 const enqueue = function () {    
 if (i === urls.length) {      
   return Promise.resolve()    
 }    
 const item = urls[i++] // 獲取新的任務(wù)項    
 const p = Promise.resolve().then(() => 
  handler(item, urls))    
  ret.push(p)    
  let r = Promise.resolve()    // 當(dāng)poolLimit值小于或等于總?cè)蝿?wù)個數(shù)時，進行并發(fā)控制    
  if (max <= urls.length) {      // 當(dāng)任務(wù)完成后，從正在執(zhí)行的任務(wù)數(shù)組中移除已完成的任務(wù)      
  const e = p.then(() => executing.splice(executing.indexOf(e), 1)).catch((error) => { })      
  executing.push(e)      
   if (executing.length >= max) {       
    r = Promise.race(executing)       
   }    
  }     // 正在執(zhí)行任務(wù)列表 中較快的任務(wù)執(zhí)行完成之后，才會從array數(shù)組中獲取新的待辦任務(wù)    
  return r.then(() => enqueue())  
 }  
return enqueue().then(() => Promise.all(ret)).catch(error => {    
  console.error('handlePool', error)  
})}

class MainWorker extends EventEmitter {  
    constructor(ctx, jobCount) {    
        super()    
        this.ctx = ctx    
        this.pagePools = [] // 記錄每個job的信息    
        this.jobCount = jobCount || 6    this.instance = null    // 生成jobCount個job，用來后面    
        this.createJobs()  
        
    }  createJob(){    
        return new Promise((resolve, reject) => {      
            this.ctx.pool.use(async instance => {        // instance為瀏覽器實例,寫處理邏輯的handler        
            const page = await instance.newPage()        
            const jobId = Util.token() // 隨機數(shù)ID        
            this.pagePools.push({         
                jobId: jobId,          instance,          page,          isIdle: true        })        
                resolve()      })    })      }  
                async createJobs(){   
                    for (let i = 0;i < this.jobCount;i++){     
                        await this.createJob()   
                        } 
                        }  // 調(diào)用開始就是調(diào)用job開始工作  
                        async start() {    
                            this.pagePools.forEach(el => {      
                                this.send({        type: 'jobReady',        jobId: el.jobId      })    })  }}

const puppeteer = require('puppeteer')
const genericPool = require('generic-pool')
/** * 初始化一個 Puppeteer 池 
* @param {Object} [options={}] 創(chuàng)建池的配置配置 
* @param {Number} [options.max=30] 最多產(chǎn)生多少個 puppeteer 實例 。如果你設(shè)置它，請確保 在引用關(guān)閉時調(diào)用清理池。 pool.drain().then(()=>pool.clear()) 
* @param {Number} [options.min=15] 保證池中最少有多少個實例存活 
* @param {Number} [options.maxUses=2048] 每一個 實例 最大可重用次數(shù)，超過后將重啟實例。0表示不檢驗 
* @param {Number} [options.testOnBorrow=2048] 在將 實例 提供給用戶之前，池應(yīng)該驗證這些實例。 
* @param {Boolean} [options.autostart=false] 是不是需要在 池 初始化時 初始化 實例 
* @param {Number} [options.idleTimeoutMillis=3600000] 如果一個實例 60分鐘 都沒訪問就關(guān)掉他 
* @param {Number} [options.evictionRunIntervalMillis=180000] 每 3分鐘 檢查一次 實例的訪問狀態(tài) 
* @param {Object} [options.puppeteerArgs={}] puppeteer.launch 啟動的參數(shù) 
* @param {Function} [options.validator=(instance)=>Promise.resolve(true))] 用戶自定義校驗 參數(shù)是 取到的一個實例 
* @return {Object} pool */
const initPuppeteerPool = (options = { otherConfig: {} }) => {  
const {    max = 20,    min = 10,    maxUses = 2048,    testOnBorrow = true,    autostart = false,    idleTimeoutMillis = 3600000,    evictionRunIntervalMillis = 180000,    
puppeteerArgs = {      
executablePath: 'google-chrome-stable',       // executablePath: 'C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe',      // executablePath: 'C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe', // 寶丹me路徑      // executablePath: 'google-chrome-stable',       
headless: true,       
devtools: false,      
defaultViewport: {        width: 1920,        height: 1080      },
      slowMo: 200,
      args: [
          '--no-sandbox',
        '--unlimited-storage',
        '--full-memory-crash-report',
        '--disable-gpu',
        '--disable-gpu-sandbox',
        '--disable-gl-drawing-for-tests',
        '--ignore-certificate-errors',
        '--no-sandbox',
        '--disable-setuid-sandbox',
        '--disable-dev-shm-usage',
        '--lang=en-US;q=0.9,en;q=0.8',
        '--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.106 Safari/537.36'
      ]
    },
    validator = () => Promise.resolve(true)
  } = options
  const factory = {
    create: () =>
      puppeteer.launch(puppeteerArgs).then(async(browser) => {
        browser.on('targetdestroyed', () => {
          console.log('1111 page closed')
        })
        browser.on('disconnected', () => {
          console.log('disconnected')
        })
        // 創(chuàng)建一個 puppeteer 實例 ，并且初始化使用次數(shù)為 0
        const instance = await browser.createIncognitoBrowserContext() // 開啟無痕模式
        instance.useCount = 0
        return instance
      }),
    destroy:async (instance) => {
          const pages = await instance.pages()
     console.log('實例 close',pages)
     for(let i=0,l=pages.length;i<l;i++){
      await pages[i].close()
     } 
    await instance.close()
     await browser.close()
    },
    validate: instance => {
      // 執(zhí)行一次自定義校驗，并且校驗校驗 實例已使用次數(shù)。 當(dāng) 返回 reject 時 表示實例不可用
      return validator(instance).then(valid => Promise.resolve(valid && (maxUses <= 0 || instance.useCount < maxUses)))
    }
  }
  const config = {    max,    min,    testOnBorrow,    autostart,    idleTimeoutMillis,    evictionRunIntervalMillis  }
  const pool = genericPool.createPool(factory, config)
  const genericAcquire = pool.acquire.bind(pool)
  // pool.drain = pool.acquire.bind(pool)
  // 重寫了原有池的消費實例的方法。添加一個實例使用次數(shù)的增加 
 pool.acquire = () =>    genericAcquire().then(instance => {
      instance.useCount += 1
      return instance
    })
  // pool.drain = () =>{}
  pool.use = fn => {
    let resource
    // let page
    return pool
      .acquire()
      .then(r => {
        resource = r
        // page = resource.newPage()
        return resource
      })
      .then(fn)
      .then(
        result => {
          // 不管業(yè)務(wù)方使用實例成功與后都表示一下實例消費完成
          pool.release(resource)
          return result
        },
        err => {
          pool.release(resource)
          throw err
        }
      ).catch(err => { 
       pool.release(resource)
        throw err
      })
  } 
 return pool
}
module.exports = initPuppeteerPool