feat(concat): accept multiple concat sources

tomi · Apr 15, 2019 · 2ee75c7 · 2ee75c7
1 parent fb5036e
commit 2ee75c7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/src/Sequence.ts b/src/Sequence.ts
@@ -72,8 +72,8 @@ export class Sequence<TItem> implements Iterable<TItem> {
    * from([1, 2]).concat([3, 4]);
    * ```
    */
-  concat<TOther>(other: Iterable<TOther>): Sequence<TItem | TOther> {
-    return new Sequence(createConcatIterable(this._iterable, other));
+  concat<TOther>(...others: Iterable<TOther>[]): Sequence<TItem | TOther> {
+    return new Sequence(createConcatIterable(this._iterable, ...others));
   }
 
   /**

diff --git a/src/transforms/concat.ts b/src/transforms/concat.ts
@@ -2,14 +2,18 @@ import { IterableCreatorIterable } from "../IterableCreatorIterable";
 
 export const createConcatIterable = <TItem, TOther>(
   source: Iterable<TItem>,
-  other: Iterable<TOther>
+  ...others: Iterable<TOther>[]
 ): Iterable<TItem | TOther> =>
-  new IterableCreatorIterable(function* concat(): IterableIterator<TItem | TOther> {
+  new IterableCreatorIterable(function* concat(): IterableIterator<
+    TItem | TOther
+  > {
     for (const item of source) {
       yield item;
     }
 
-    for (const item of other) {
-      yield item;
+    for (const other of others) {
+      for (const item of other) {
+        yield item;
+      }
     }
   });