puppeteer-cluster-connect
v0.0.8
Published
Puppeteer Cluster that support puppeteer connect (puppeteer.connect) , Current Only Support Connect method, use puppeteer-cluster package to use Launch method
Downloads
188
Readme
Puppeteer cluster Connect
Use Cluster.connect instead on Cluster.launch
Installation
Install using your favorite package manager:
npm install --save puppeteer # in case you don't already have it installed
npm install --save puppeteer-cluster-connect
Alternatively, use yarn
:
yarn add puppeteer puppeteer-cluster-connect
Current support : one restart function
Array restart function is current in development
full document please go to :
puppeteer-cluster
(use connect instead of launch and add restartFunction
if you want restart endpoint after crash or error)
Usage
The following is a typical example of using puppeteer-cluster-connect. A cluster is created with 2 concurrent workers. Then a task is defined which includes going to the URL and taking a screenshot. We then queue three jobs and wait for the cluster to finish.
const { Cluster } = require("puppeteer-cluster-connect");
const puppeteer = require("puppeteer-core"); // use puppeteer-core instead of puppeteer
const puppeteerRemote = require("puppeteer");
// restart function
const ws = async () => {
console.log("create browser");
const browser = await puppeteerRemote.launch({
headless: false,
defaultViewport: null,
args: [
"--start-maximized",
// add other if you want
],
executablePath:
"C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe",
});
// Store the endpoint to be able to reconnect to Chromium
const browserWSEndpoint = browser.wsEndpoint();
console.log(browserWSEndpoint);
// Disconnect puppeteer from Chromium
browser.disconnect();
return browserWSEndpoint;
};
(async () => {
try {
const cluster = await Cluster.connect({
concurrency: Cluster.CONCURRENCY_BROWSER,
maxConcurrency: 2,
// provide the puppeteer-core library
puppeteer,
puppeteerOptions: {
browserWSEndpoint: await ws(),
// defaultViewport: null,
},
// Put restart function callback here
restartFunction: ws,
// current in development
// perBrowserOptions: [
// {
// browserWSEndpoint: await ws(),
// },
// {
// browserWSEndpoint: await ws(),
// },
// {
// browserWSEndpoint: await ws(),
// },
// ],
});
await cluster.task(async ({ page, data: url }) => {
console.log("run : " + url);
await page.goto(url);
console.log("went to: " + url);
const screen = await page.screenshot();
// Store screenshot, do something else
});
cluster.queue("https://www.google.com");
cluster.queue("https://www.wikipedia.org");
cluster.queue("https://github.com/");
await cluster.idle();
await cluster.close();
} catch (error) {
console.log(error);
}
})();