redis-smq
Version:
A high-performance, reliable, and scalable message queue for Node.js.
69 lines • 2.9 kB
JavaScript
import { EQueueOperationalState, EQueueProperty, } from '../../../queue-manager/index.js';
import os from 'os';
import { ERedisScriptName } from '../../../common/redis/scripts.js';
import { QueueNotActiveError, QueueNotFoundError, UnexpectedScriptReplyError, } from '../../../errors/index.js';
import { redisKeys } from '../../../common/redis/redis-keys/redis-keys.js';
import { withSharedPoolConnection } from '../../../common/redis/redis-connection-pool/with-shared-pool-connection.js';
const IPAddresses = (() => {
const nets = os.networkInterfaces();
const addresses = [];
for (const netInterface in nets) {
const addr = nets[netInterface] ?? [];
for (const netAddr of addr) {
if (netAddr.family === 'IPv4' && !netAddr.internal) {
addresses.push(netAddr.address);
}
}
}
return addresses;
})();
export function _subscribeConsumer(consumerId, queueParams, cb) {
withSharedPoolConnection((redisClient, done) => {
const { queueParams: queue, groupId } = queueParams;
const consumerInfo = {
ipAddress: IPAddresses,
hostname: os.hostname(),
pid: process.pid,
createdAt: Date.now(),
};
const { keyQueueProcessingQueues, keyQueueConsumers, keyQueueProperties } = redisKeys.getQueueKeys(queue.ns, queue.name, groupId);
const { keyQueueProcessing } = redisKeys.getQueueConsumerKeys(queue, consumerId);
const { keyConsumerQueues } = redisKeys.getConsumerKeys(consumerId);
const keys = [
keyQueueProperties,
keyQueueConsumers,
keyConsumerQueues,
keyQueueProcessingQueues,
keyQueueProcessing,
];
if (groupId) {
const { keyQueueConsumerGroupConsumers } = redisKeys.getQueueConsumerGroupKeys(queue, groupId);
keys.push(keyQueueConsumerGroupConsumers);
}
const args = [
consumerId,
JSON.stringify(consumerInfo),
JSON.stringify(queue),
EQueueProperty.OPERATIONAL_STATE,
EQueueOperationalState.ACTIVE,
];
redisClient.runScript(ERedisScriptName.SUBSCRIBE_CONSUMER, keys, args, (err, reply) => {
if (err)
return done(err);
if (reply === 'QUEUE_NOT_FOUND')
return done(new QueueNotFoundError({
metadata: {
queue: queue,
},
}));
if (reply === 'QUEUE_NOT_ACTIVE')
return done(new QueueNotActiveError({
metadata: { queue },
}));
if (reply === 'OK')
return done();
done(new UnexpectedScriptReplyError({ metadata: { reply } }));
});
}, cb);
}
//# sourceMappingURL=_subscribe-consumer.js.map