fix chromium分片截图未导入common (#57)

Co-authored-by: Kokomi <102026640+yoimiya-kokomi@users.noreply.github.com>
This commit is contained in:
story-x 2023-04-11 01:51:43 +08:00 committed by GitHub
parent b22b70e9e1
commit 85b9d03877
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 14 deletions

View File

@ -4,6 +4,7 @@ import lodash from 'lodash'
import template from 'art-template'
import chokidar from 'chokidar'
import cfg from '../config/config.js'
import common from '../common/common.js'
const _path = process.cwd()
@ -74,21 +75,21 @@ class Puppeteer {
let connectFlag = false
try {
//如果是pm2启动尝试连接已有实例
// 如果是pm2启动尝试连接已有实例
if (process.env.pm_id) {
//获取Mac地址
// 获取Mac地址
if (!mac) {
mac = await this.getMac()
this.browserMacKey = `Yz:chromium:browserWSEndpoint:${mac}`
}
//是否有browser实例
// 是否有browser实例
const browserUrl = await redis.get(this.browserMacKey)
if (browserUrl) {
const browserWSEndpoint = await puppeteer.connect({ browserWSEndpoint: browserUrl }).catch((err) => {
logger.error('puppeteer Chromium 缓存的实例已关闭')
redis.del(this.browserMacKey)
})
//如果有实例,直接使用
// 如果有实例,直接使用
if (browserWSEndpoint) {
this.browser = browserWSEndpoint
if (this.browser) {
@ -101,9 +102,8 @@ class Puppeteer {
logger.error('puppeteer Chromium 尝试连接已有实例失败')
}
if (!this.browser || !connectFlag) {
//如果没有实例初始化puppeteer
// 如果没有实例初始化puppeteer
this.browser = await puppeteer.launch(this.config).catch((err, trace) => {
let errMsg = err.toString() + (trace ? trace.toString() : '')
if (typeof err == 'object') {
@ -147,21 +147,21 @@ class Puppeteer {
return this.browser
}
//获取Mac地址
// 获取Mac地址
async getMac () {
//获取Mac地址
// 获取Mac地址
let mac = '00:00:00:00:00:00'
try {
const network = os.networkInterfaces()
let osMac
//判断系统
// 判断系统
if (os.platform() === 'win32') {
//windows下获取mac地址
// windows下获取mac地址
let osMacList = Object.keys(network).map(key => network[key]).flat()
osMacList = osMacList.filter(item => item.family === 'IPv4' && item.mac !== mac)
osMac = osMacList[0].mac
} else if (os.platform() === 'linux') {
//linux下获取mac地址
// linux下获取mac地址
osMac = network.eth0.filter(item => item.family === 'IPv4' && item.mac !== mac)[0].mac
}
if (osMac) {
@ -282,19 +282,19 @@ class Puppeteer {
quality: 90
}
if (i != 1 && i == num) {
if (i !== 1 && i === num) {
await page.setViewport({
width: boundingBox.width,
height: parseInt(boundingBox.height) - pageHeight * (num - 1)
})
}
if (i != 1 && i <= num) {
if (i !== 1 && i <= num) {
await page.evaluate(() => window.scrollBy(0, 7000))
}
let buff
if (num == 1) {
if (num === 1) {
buff = await body.screenshot(randData)
} else {
buff = await page.screenshot(randData)