3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-16 06:11:49 +00:00
zeppelin/backend/src/Queue.ts

50 lines
1.1 KiB
TypeScript
Raw Normal View History

2019-08-18 16:40:15 +03:00
import { SECONDS } from "./utils";
2021-05-22 13:34:25 +03:00
type InternalQueueFn = () => Promise<void>;
type AnyFn = (...args: any[]) => any;
2019-08-18 16:40:15 +03:00
const DEFAULT_TIMEOUT = 10 * SECONDS;
2019-02-23 21:19:46 +02:00
2021-05-22 13:34:25 +03:00
export class Queue<TQueueFunction extends AnyFn = AnyFn> {
protected running = false;
protected queue: InternalQueueFn[] = [];
2019-02-23 21:19:46 +02:00
protected timeout: number;
constructor(timeout = DEFAULT_TIMEOUT) {
this.timeout = timeout;
}
2021-05-22 13:34:25 +03:00
public add(fn: TQueueFunction): Promise<void> {
const promise = new Promise<void>(resolve => {
this.queue.push(async () => {
await fn();
2021-05-22 13:34:25 +03:00
resolve();
});
if (!this.running) this.next();
});
return promise;
}
2021-05-22 13:34:25 +03:00
public next(): void {
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
2021-05-22 13:34:25 +03:00
void fn().then(resolve);
setTimeout(resolve, this.timeout);
}).then(() => this.next());
}
2020-07-27 20:42:10 +03:00
public clear() {
this.queue.splice(0, this.queue.length);
}
}