redis-smq
Version:
A simple high-performance Redis message queue for Node.js.
67 lines • 2.59 kB
JavaScript
import { CallbackEmptyReplyError } from 'redis-smq-common';
import { redisKeys } from '../../../common/redis-keys/redis-keys.js';
import { QueueMessagesStorage } from './queue-messages-storage.js';
export class QueueMessagesStorageSortedSet extends QueueMessagesStorage {
count(queue, redisKey, cb) {
this.redisClient.getSetInstance((err, client) => {
if (err)
cb(err);
else if (!client)
cb(new CallbackEmptyReplyError());
else {
const keys = 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 CallbackEmptyReplyError());
else {
const keys = 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 CallbackEmptyReplyError());
else {
const keys = 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');
}
});
}
}
//# sourceMappingURL=queue-messages-storage-sorted-set.js.map