fix: update chunking methods to send at least a single chunk

Cypherock · Sep 23, 2024 · 86b7acd · 86b7acd
1 parent e0b4185
commit 86b7acd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/packages/app-inheritance/src/utils/operationHelper.ts b/packages/app-inheritance/src/utils/operationHelper.ts
@@ -94,9 +94,10 @@ export class OperationHelper<Q extends QueryKey, R extends ResultKey> {
   >(data: Uint8Array, queryKey: QK, resultKey: RK) {
     const chunks = OperationHelper.splitIntoChunks(data);
     let remainingSize = data.length;
+    let i = 0;
 
-    for (let i = 0; i < chunks.length; i += 1) {
-      const chunk = chunks[i];
+    do {
+      const chunk = chunks[i] ?? [];
       remainingSize -= chunk.length;
 
       const chunkPayload: ChunkPayload = {
@@ -121,7 +122,8 @@ export class OperationHelper<Q extends QueryKey, R extends ResultKey> {
 
       assertOrThrowInvalidResult(chunkAck);
       assertOrThrowInvalidResult(chunkAck.chunkIndex === i);
-    }
+      i += 1;
+    } while (i < chunks.length);
   }
 
   public async receiveInChunks<
@@ -152,7 +154,7 @@ export class OperationHelper<Q extends QueryKey, R extends ResultKey> {
 
       if (
         chunkPayload.remainingSize === 0 &&
-        chunkPayload.chunkIndex + 1 === chunkPayload.totalChunks
+        chunkPayload.chunkIndex + 1 >= chunkPayload.totalChunks
       ) {
         break;
       }