@speckle/shared
Version:
Shared code between various Speckle JS packages
65 lines • 2.73 kB
JavaScript
import Bull from 'bull';
import { Redis } from 'ioredis';
import { isRedisReady } from '../redis/isRedisReady.js';
// we're caching this here, so that there is one client for the app lifecycle
const clientCache = {};
// so we can get all active queues for monitoring
const queueCache = {};
export const initializeQueue = async ({ queueName, redisUrl, options }) => {
if (!(redisUrl in clientCache))
clientCache[redisUrl] = {};
const opts = {
...options,
// redisOpts here will contain at least a property of connectionName which will identify the queue based on its name
createClient(type, redisOpts) {
switch (type) {
case 'client':
if (redisUrl in clientCache && clientCache[redisUrl].client !== undefined) {
return clientCache[redisUrl].client;
}
else {
const client = new Redis(redisUrl, redisOpts ?? {});
clientCache[redisUrl].client = client;
return client;
}
case 'subscriber':
if (redisUrl in clientCache &&
clientCache[redisUrl].subscriber !== undefined) {
return clientCache[redisUrl].subscriber;
}
else {
const subscriber = new Redis(redisUrl, {
...redisOpts,
maxRetriesPerRequest: null,
enableReadyCheck: false
});
clientCache[redisUrl].subscriber = subscriber;
return subscriber;
}
case 'bclient':
return new Redis(redisUrl, {
...redisOpts,
maxRetriesPerRequest: null,
enableReadyCheck: false
});
default:
throw new Error(`Unexpected connection type: ${type}`);
}
}
};
const newQueue = new Bull(queueName, opts);
queueCache[queueName] = newQueue;
// When newQueue closed, remove from cache
newQueue.on('close', () => {
delete queueCache[queueName];
});
newQueue.client.on('end', () => {
delete queueCache[queueName];
});
if (!clientCache[redisUrl].client)
throw new Error('Redis client not properly initialized');
await isRedisReady(clientCache[redisUrl].client);
return await newQueue.isReady();
};
export const getActiveQueues = () => ({ ...queueCache });
//# sourceMappingURL=config.js.map