improving repeat operator

vitaly-t · Aug 3, 2024 · 8d89bc9 · 8d89bc9
1 parent ad62a4e
commit 8d89bc9
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/ops/repeat.ts b/src/ops/repeat.ts
@@ -70,12 +70,15 @@ function repeatSync<T>(
               state: IterationState
           ) => boolean)
 ): Iterable<T> {
+    if (typeof count === 'number' && count < 1) {
+        return iterable;
+    }
     return {
         [$S](): Iterator<T> {
             const i = iterable[$S]();
             const state: IterationState = {};
             const cb = typeof count === 'function' && count;
-            const initialCount = !cb && count > 0 ? count : 0;
+            const initialCount = !cb && count;
             let copyCount = initialCount;
             let index = -1,
                 copied = 0,
@@ -120,12 +123,15 @@ function repeatAsync<T>(
               state: IterationState
           ) => boolean | Promise<boolean>)
 ): AsyncIterable<T> {
+    if (typeof count === 'number' && count < 1) {
+        return iterable;
+    }
     return {
         [$A](): AsyncIterator<T> {
             const i = iterable[$A]();
             const state: IterationState = {};
             const cb = typeof count === 'function' && count;
-            const initialCount = !cb && count > 0 ? count : 0;
+            const initialCount = !cb && count;
             let copyCount = initialCount;
             let index = -1,
                 copied = 0,