redis-smq
Version:
A high-performance, reliable, and scalable message queue for Node.js.
76 lines • 3.35 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports._subscribeConsumer = _subscribeConsumer;
const index_js_1 = require("../../../queue-manager/index.js");
const os_1 = __importDefault(require("os"));
const scripts_js_1 = require("../../../common/redis/scripts.js");
const index_js_2 = require("../../../errors/index.js");
const redis_keys_js_1 = require("../../../common/redis/redis-keys/redis-keys.js");
const with_shared_pool_connection_js_1 = require("../../../common/redis/redis-connection-pool/with-shared-pool-connection.js");
const IPAddresses = (() => {
var _a;
const nets = os_1.default.networkInterfaces();
const addresses = [];
for (const netInterface in nets) {
const addr = (_a = nets[netInterface]) !== null && _a !== void 0 ? _a : [];
for (const netAddr of addr) {
if (netAddr.family === 'IPv4' && !netAddr.internal) {
addresses.push(netAddr.address);
}
}
}
return addresses;
})();
function _subscribeConsumer(consumerId, queueParams, cb) {
(0, with_shared_pool_connection_js_1.withSharedPoolConnection)((redisClient, done) => {
const { queueParams: queue, groupId } = queueParams;
const consumerInfo = {
ipAddress: IPAddresses,
hostname: os_1.default.hostname(),
pid: process.pid,
createdAt: Date.now(),
};
const { keyQueueProcessingQueues, keyQueueConsumers, keyQueueProperties } = redis_keys_js_1.redisKeys.getQueueKeys(queue.ns, queue.name, groupId);
const { keyQueueProcessing } = redis_keys_js_1.redisKeys.getQueueConsumerKeys(queue, consumerId);
const { keyConsumerQueues } = redis_keys_js_1.redisKeys.getConsumerKeys(consumerId);
const keys = [
keyQueueProperties,
keyQueueConsumers,
keyConsumerQueues,
keyQueueProcessingQueues,
keyQueueProcessing,
];
if (groupId) {
const { keyQueueConsumerGroupConsumers } = redis_keys_js_1.redisKeys.getQueueConsumerGroupKeys(queue, groupId);
keys.push(keyQueueConsumerGroupConsumers);
}
const args = [
consumerId,
JSON.stringify(consumerInfo),
JSON.stringify(queue),
index_js_1.EQueueProperty.OPERATIONAL_STATE,
index_js_1.EQueueOperationalState.ACTIVE,
];
redisClient.runScript(scripts_js_1.ERedisScriptName.SUBSCRIBE_CONSUMER, keys, args, (err, reply) => {
if (err)
return done(err);
if (reply === 'QUEUE_NOT_FOUND')
return done(new index_js_2.QueueNotFoundError({
metadata: {
queue: queue,
},
}));
if (reply === 'QUEUE_NOT_ACTIVE')
return done(new index_js_2.QueueNotActiveError({
metadata: { queue },
}));
if (reply === 'OK')
return done();
done(new index_js_2.UnexpectedScriptReplyError({ metadata: { reply } }));
});
}, cb);
}
//# sourceMappingURL=_subscribe-consumer.js.map