优化 puppeteer

This commit is contained in:
🌌 2023-09-06 12:32:10 +08:00
parent 388ced1288
commit 4949406ef5
2 changed files with 30 additions and 44 deletions

View File

@ -21,7 +21,7 @@
"chokidar": "^3.5.3", "chokidar": "^3.5.3",
"express": "^4.18.2", "express": "^4.18.2",
"file-type": "^18.5.0", "file-type": "^18.5.0",
"https-proxy-agent": "7.0.1", "https-proxy-agent": "7.0.2",
"image-size": "^1.0.2", "image-size": "^1.0.2",
"lodash": "^4.17.21", "lodash": "^4.17.21",
"log4js": "^6.9.1", "log4js": "^6.9.1",

View File

@ -103,12 +103,12 @@ export default class PuppeteerRenderer {
} else { } else {
logger.error(err.toString()) logger.error(err.toString())
if (errMsg.includes('Could not find Chromium')) { if (errMsg.includes('Could not find Chromium')) {
logger.error('没有正确安装Chromium可以尝试执行安装命令node ./node_modules/puppeteer/install.js') logger.error('没有正确安装 Chromium可以尝试执行安装命令node node_modules/puppeteer/install.js')
} else if (errMsg.includes('libatk-bridge')) { } else if (errMsg.includes('cannot open shared object file')) {
logger.error('没有正确安装Chromium可尝试执行 sudo yum install -y chromium') logger.error('没有正确安装 Chromium 运行库')
} }
} }
console.log(err, trace) logger.error(err, trace)
}) })
} }
@ -121,7 +121,7 @@ export default class PuppeteerRenderer {
if (connectFlag) { if (connectFlag) {
logger.info('puppeteer Chromium 已连接启动的实例') logger.info('puppeteer Chromium 已连接启动的实例')
} else { } else {
console.log('chromium', this.browser.wsEndpoint()) logger.info(`[Chromium] ${this.browser.wsEndpoint()}`)
if (process.env.pm_id && this.browserMacKey) { if (process.env.pm_id && this.browserMacKey) {
//缓存一下实例30天 //缓存一下实例30天
const expireTime = 60 * 60 * 24 * 30 const expireTime = 60 * 60 * 24 * 30
@ -140,29 +140,15 @@ export default class PuppeteerRenderer {
} }
// 获取Mac地址 // 获取Mac地址
async getMac () { getMac() {
// 获取Mac地址 const mac = '00:00:00:00:00:00'
let mac = '00:00:00:00:00:00'
try { try {
const network = os.networkInterfaces() const network = os.networkInterfaces()
let osMac for (const a in network)
// 判断系统 for (const i of network[a])
if (os.platform() === 'win32') { if (i.mac && i.mac != mac)
// windows下获取mac地址 return i.mac
let osMacList = Object.keys(network).map(key => network[key]).flat() } catch (e) {}
osMacList = osMacList.filter(item => item.family === 'IPv4' && item.mac !== mac)
osMac = osMacList[0].mac
} else if (os.platform() === 'linux') {
// linux下获取mac地址
osMac = network.eth0.filter(item => item.family === 'IPv4' && item.mac !== mac)[0].mac
}
if (osMac) {
mac = String(osMac)
}
} catch (e) {
console.log('获取Mac地址失败', e.toString())
}
mac = mac.replace(/:/g, '')
return mac return mac
} }
@ -225,8 +211,8 @@ export default class PuppeteerRenderer {
if (!data.multiPage) { if (!data.multiPage) {
buff = await body.screenshot(randData) buff = await body.screenshot(randData)
/** 计算图片大小 */ /** 计算图片大小 */
const kb = (buff.length / 1024).toFixed(2) + 'kb' const kb = (buff.length / 1024).toFixed(2) + 'KB'
logger.info(`[图片生成][${name}][${this.renderNum}次] ${kb} ${logger.green(`${Date.now() - start}ms`)}`) logger.mark(`[图片生成][${name}][${this.renderNum}次] ${kb} ${logger.green(`${Date.now() - start}ms`)}`)
this.renderNum++ this.renderNum++
ret.push(buff) ret.push(buff)
} else { } else {
@ -256,18 +242,18 @@ export default class PuppeteerRenderer {
this.renderNum++ this.renderNum++
/** 计算图片大小 */ /** 计算图片大小 */
const kb = (buff.length / 1024).toFixed(2) + 'kb' const kb = (buff.length / 1024).toFixed(2) + 'KB'
logger.info(`[图片生成][${name}][${i}/${num}] ${kb}`) logger.mark(`[图片生成][${name}][${i}/${num}] ${kb}`)
ret.push(buff) ret.push(buff)
} }
if (num > 1) { if (num > 1) {
logger.info(`[图片生成][${name}] 处理完成`) logger.mark(`[图片生成][${name}] 处理完成`)
} }
} }
page.close().catch((err) => logger.error(err)) page.close().catch((err) => logger.error(err))
} catch (error) { } catch (error) {
logger.error(`图片生成失败:${name}:${error}`) logger.error(`[图片生成][${name}] 图片生成失败:${error}`)
/** 关闭浏览器 */ /** 关闭浏览器 */
if (this.browser) { if (this.browser) {
await this.browser.close().catch((err) => logger.error(err)) await this.browser.close().catch((err) => logger.error(err))
@ -280,7 +266,7 @@ export default class PuppeteerRenderer {
this.shoting.pop() this.shoting.pop()
if (ret.length === 0 || !ret[0]) { if (ret.length === 0 || !ret[0]) {
logger.error(`图片生成为空:${name}`) logger.error(`[图片生成][${name}] 图片生成为空`)
return false return false
} }
@ -328,7 +314,7 @@ export default class PuppeteerRenderer {
const watcher = chokidar.watch(tplFile) const watcher = chokidar.watch(tplFile)
watcher.on('change', path => { watcher.on('change', path => {
delete this.html[tplFile] delete this.html[tplFile]
logger.info(`[修改html模板] ${tplFile}`) logger.mark(`[修改html模板] ${tplFile}`)
}) })
this.watcher[tplFile] = watcher this.watcher[tplFile] = watcher
@ -344,7 +330,7 @@ export default class PuppeteerRenderer {
await this.browser.close().catch((err) => logger.error(err)) await this.browser.close().catch((err) => logger.error(err))
} }
this.browser = false this.browser = false
logger.info('puppeteer 关闭重启...') logger.info('puppeteer Chromium 关闭重启...')
}, 100) }, 100)
} }
} }