Add heartbeat handling to key iteration

Introduce a heartbeat case to handle key iteration status updates. This ensures proper key fetching and stops the iteration appropriately upon receiving a heartbeat.

Signed-off-by: Alberto Ricart <[email protected]>

kv/src/kv.ts

@@ -899,6 +899,19 @@ export class Bucket implements KV {
       },
     });
 
+    (async () => {
+      for await (const s of iter.status()) {
+        switch (s.type) {
+          // if we get a heartbeat we got all the keys
+          case "heartbeat":
+            keys.push(() => {
+              keys.stop();
+            });
+            break;
+        }
+      }
+    })().then();
+
     iter.closed().then(() => {
       keys.push(() => {
         keys.stop();