thunderstrike/puppeteer
thunderstrike/puppeteer
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

perf: amortize handle iterator (#10002)

Browse Source
This commit is contained in:
jrandolf 2023-04-11 15:36:03 +02:00 committed by GitHub
parent dba1b2d9cc
commit ab27f738c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
packages/puppeteer-core/src/common/HandleIterator.ts
View File

@ -29,7 +29,7 @@ const DEFAULT_BATCH_SIZE = 20;
*/
async function* fastTransposeIteratorHandle<T>(
iterator: JSHandle<AwaitableIterator<T>>,
size = DEFAULT_BATCH_SIZE
size: number
) {
const array = await iterator.evaluateHandle(async (iterator, size) => {
const results = [];
@ -56,8 +56,11 @@ async function* fastTransposeIteratorHandle<T>(
async function* transposeIteratorHandle<T>(
iterator: JSHandle<AwaitableIterator<T>>
) {
let size = DEFAULT_BATCH_SIZE;
try {
while (!(yield* fastTransposeIteratorHandle(iterator))) {}
while (!(yield* fastTransposeIteratorHandle(iterator, size))) {
size <<= 1;
}
} finally {
await iterator.dispose();
}