27 lines
615 B
TypeScript
27 lines
615 B
TypeScript
// Copyright 2021 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import PQueue from 'p-queue';
|
|
|
|
export class InMemoryQueues {
|
|
private readonly queues = new Map<string, PQueue>();
|
|
|
|
get(key: string): PQueue {
|
|
const existingQueue = this.queues.get(key);
|
|
if (existingQueue) {
|
|
return existingQueue;
|
|
}
|
|
|
|
const newQueue = new PQueue({ concurrency: 1 });
|
|
newQueue.once('idle', () => {
|
|
this.queues.delete(key);
|
|
});
|
|
|
|
this.queues.set(key, newQueue);
|
|
return newQueue;
|
|
}
|
|
|
|
get allQueues(): ReadonlySet<PQueue> {
|
|
return new Set(this.queues.values());
|
|
}
|
|
}
|