From ab27f738c9abb56f6083d02f7f45d2b8da9fc3f3 Mon Sep 17 00:00:00 2001
From: jrandolf <101637635+jrandolf@users.noreply.github.com>
Date: Tue, 11 Apr 2023 15:36:03 +0200
Subject: [PATCH] perf: amortize handle iterator (#10002)

---
 packages/puppeteer-core/src/common/HandleIterator.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/puppeteer-core/src/common/HandleIterator.ts b/packages/puppeteer-core/src/common/HandleIterator.ts
index a7a3f0f3cdf..d5df382f881 100644
--- a/packages/puppeteer-core/src/common/HandleIterator.ts
+++ b/packages/puppeteer-core/src/common/HandleIterator.ts
@@ -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();
   }