forked from thomasdondorf/puppeteer-cluster
-
Notifications
You must be signed in to change notification settings - Fork 0
/
typings.ts
66 lines (50 loc) · 1.9 KB
/
typings.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import { Cluster } from '../dist';
(async () => {
/******************************* STRING -> NUMBER **********************/
// Queued data is a string and task functions return a number
const cluster1: Cluster<string, number> = await Cluster.launch({
concurrency: Cluster.CONCURRENCY_PAGE,
maxConcurrency: 2,
});
await cluster1.task(async ({ page, data }) => {
// ...
const value = Math.random(); // somehow generate data
return value;
});
// TypeScript now knows that data1 is a number
const data1 = await cluster1.execute('https://www.google.com');
await cluster1.idle();
await cluster1.close();
/******************************* STRING DATA **********************/
// Queued data is a string
const cluster2: Cluster<string> = await Cluster.launch({
concurrency: Cluster.CONCURRENCY_PAGE,
maxConcurrency: 2,
});
await cluster2.task(async ({ page, data: url }) => {
await page.goto(url);
});
cluster2.queue('https://www.google.com');
cluster2.queue('https://www.wikipedia.org');
cluster2.queue('https://github.com/');
await cluster2.idle();
await cluster2.close();
/******************************* COMPLEX DATA **********************/
interface SomeData {
url: string;
someValue: number;
}
const cluster3: Cluster<SomeData> = await Cluster.launch({
concurrency: Cluster.CONCURRENCY_PAGE,
maxConcurrency: 2,
});
await cluster3.task(async ({ page, data }) => {
await page.goto(data.url);
console.log(`some value: ${data.someValue}`);
});
cluster3.queue({ url: 'https://www.google.com', someValue: 1 });
cluster3.queue({ url: 'https://www.wikipedia.org', someValue: 2 });
cluster3.queue({ url: 'https://github.com/', someValue: 3 });
await cluster3.idle();
await cluster3.close();
})();