redis-smq
Version:
A simple high-performance Redis message queue for Node.js.
71 lines • 2.92 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.QueueMessagesStorageSortedSet = void 0;
const redis_smq_common_1 = require("redis-smq-common");
const redis_keys_js_1 = require("../../../common/redis-keys/redis-keys.js");
const queue_messages_storage_js_1 = require("./queue-messages-storage.js");
class QueueMessagesStorageSortedSet extends queue_messages_storage_js_1.QueueMessagesStorage {
count(queue, redisKey, cb) {
this.redisClient.getSetInstance((err, client) => {
if (err)
cb(err);
else if (!client)
cb(new redis_smq_common_1.CallbackEmptyReplyError());
else {
const keys = redis_keys_js_1.redisKeys.getQueueKeys(queue.queueParams, queue.groupId);
client.zcard(keys[redisKey], cb);
}
});
}
fetchItems(queue, redisKey, offset, limit, cb) {
this.redisClient.getSetInstance((err, client) => {
if (err)
cb(err);
else if (!client)
cb(new redis_smq_common_1.CallbackEmptyReplyError());
else {
const keys = redis_keys_js_1.redisKeys.getQueueKeys(queue.queueParams, queue.groupId);
client.zrange(keys[redisKey], offset, offset + limit - 1, (err, items) => {
if (err)
cb(err);
else
cb(null, items || []);
});
}
});
}
fetchAllItems(queue, redisKey, cb) {
this.redisClient.getSetInstance((err, client) => {
if (err)
cb(err);
else if (!client)
cb(new redis_smq_common_1.CallbackEmptyReplyError());
else {
const keys = redis_keys_js_1.redisKeys.getQueueKeys(queue.queueParams, queue.groupId);
const allItems = [];
const scanRecursive = (cursor) => {
client.zscan(keys[redisKey], cursor, { COUNT: 100 }, (err, result) => {
if (err) {
cb(err);
return;
}
if (!result) {
cb(null, allItems);
return;
}
allItems.push(...result.items);
if (result.cursor === '0') {
cb(null, allItems);
}
else {
setImmediate(() => scanRecursive(result.cursor));
}
});
};
scanRecursive('0');
}
});
}
}
exports.QueueMessagesStorageSortedSet = QueueMessagesStorageSortedSet;
//# sourceMappingURL=queue-messages-storage-sorted-set.js.map