feat(LinkedQueue): implement iteration protocols

slikts · Jan 24, 2023 · 68799d7 · 68799d7
1 parent 3d75244
commit 68799d7
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/LinkedQueue.spec.ts b/src/LinkedQueue.spec.ts
@@ -64,4 +64,11 @@ describe('LinkedQueue', () => {
     q.forEach(() => void (n += 1));
     expect(n).toBe(2);
   });
+
+  it('iterates', () => {
+    const q = new LinkedQueue();
+    q.enqueue(1);
+    q.enqueue(2);
+    expect([...q]).toEqual([1, 2]);
+  });
 });
diff --git a/src/LinkedQueue.ts b/src/LinkedQueue.ts
@@ -47,4 +47,22 @@ export default class LinkedQueue<A> {
   forEach(f: (value: A) => void): void {
     this.#list.forEach(f);
   }
+
+  [Symbol.iterator]() {
+    return {
+      next: () => {
+        return this.length > 0
+          ? {
+              value: this.dequeue(),
+              done: false,
+            }
+          : {
+              done: true,
+            };
+      },
+      [Symbol.iterator]() {
+        return this;
+      },
+    };
+  }
 }