fix: we should call .close() in async iterable (#50)

trpc · Oct 7, 2023 · b9a82f4 · b9a82f4
1 parent 627437a
commit b9a82f4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/async/handlers/tsonAsyncIterable.ts b/src/async/handlers/tsonAsyncIterable.ts
@@ -36,21 +36,21 @@ export const tsonAsyncIterator: TsonAsyncType<
 
 				switch (value[0]) {
 					case ITERATOR_DONE: {
+						opts.close();
 						return;
 					}
 
 					case ITERATOR_ERROR: {
+						opts.close();
 						throw TsonPromiseRejectionError.from(value[1]);
 					}
 
 					case ITERATOR_VALUE: {
 						yield value[1];
-						break;
+						break; // <-- breaks the switch, not the loop
 					}
 				}
 			}
-
-			opts.close();
 		})();
 	},
 	key: "AsyncIterable",