3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-15 13:51:51 +00:00
zeppelin/src/Queue.ts
2019-02-23 21:19:46 +02:00

42 lines
862 B
TypeScript

type QueueFn = (...args: any[]) => Promise<any>;
const DEFAULT_TIMEOUT = 10 * 1000;
export class Queue {
protected running: boolean = false;
protected queue: QueueFn[] = [];
protected timeout: number;
constructor(timeout = DEFAULT_TIMEOUT) {
this.timeout = timeout;
}
public add(fn) {
const promise = new Promise(resolve => {
this.queue.push(async () => {
await fn();
resolve();
});
if (!this.running) this.next();
});
return promise;
}
public next() {
this.running = true;
if (this.queue.length === 0) {
this.running = false;
return;
}
const fn = this.queue.shift();
new Promise(resolve => {
// Either fn() completes or the timeout is reached
fn().then(resolve);
setTimeout(resolve, this.timeout);
}).then(() => this.next());
}
}