refine types

repeaterjs · Oct 13, 2019 · 6c91902 · 6c91902
1 parent b085f6d
commit 6c91902
Show file tree

Hide file tree

Showing 2 changed files with 358 additions and 340 deletions.
diff --git a/packages/repeater/src/buffers.ts b/packages/repeater/src/buffers.ts
@@ -1,12 +1,12 @@
 export interface RepeaterBuffer<T> {
   full: boolean;
   empty: boolean;
-  add(value: T): void;
-  remove(): T;
+  add(value: PromiseLike<T> | T): void;
+  remove(): PromiseLike<T> | T;
 }
 
 export class FixedBuffer<T> implements RepeaterBuffer<T> {
-  private arr: T[] = [];
+  private arr: (PromiseLike<T> | T)[] = [];
 
   get empty(): boolean {
     return this.arr.length === 0;
@@ -22,15 +22,15 @@ export class FixedBuffer<T> implements RepeaterBuffer<T> {
     }
   }
 
-  add(value: T): void {
+  add(value: PromiseLike<T> | T): void {
     if (this.full) {
       throw new Error("Buffer full");
     } else {
       this.arr.push(value);
     }
   }
 
-  remove(): T {
+  remove(): PromiseLike<T> | T {
     if (this.empty) {
       throw new Error("Buffer empty");
     }
@@ -41,7 +41,7 @@ export class FixedBuffer<T> implements RepeaterBuffer<T> {
 
 // TODO: use a circular buffer here
 export class SlidingBuffer<T> implements RepeaterBuffer<T> {
-  private arr: T[] = [];
+  private arr: (PromiseLike<T> | T)[] = [];
 
   get empty(): boolean {
     return this.arr.length === 0;
@@ -57,15 +57,15 @@ export class SlidingBuffer<T> implements RepeaterBuffer<T> {
     }
   }
 
-  add(value: T): void {
+  add(value: PromiseLike<T> | T): void {
     while (this.arr.length >= this.capacity) {
       this.arr.shift();
     }
 
     this.arr.push(value);
   }
 
-  remove(): T {
+  remove(): PromiseLike<T> | T {
     if (this.empty) {
       throw new Error("Buffer empty");
     }
@@ -75,7 +75,7 @@ export class SlidingBuffer<T> implements RepeaterBuffer<T> {
 }
 
 export class DroppingBuffer<T> implements RepeaterBuffer<T> {
-  private arr: T[] = [];
+  private arr: (PromiseLike<T> | T)[] = [];
 
   get empty(): boolean {
     return this.arr.length === 0;
@@ -91,13 +91,13 @@ export class DroppingBuffer<T> implements RepeaterBuffer<T> {
     }
   }
 
-  add(value: T): void {
+  add(value: PromiseLike<T> | T): void {
     if (this.arr.length < this.capacity) {
       this.arr.push(value);
     }
   }
 
-  remove(): T {
+  remove(): PromiseLike<T> | T {
     if (this.empty) {
       throw new Error("Buffer empty");
     }