redis-smq
Version:
A simple high-performance Redis message queue for Node.js.
65 lines • 3.34 kB
JavaScript
import { async, CallbackEmptyReplyError } from 'redis-smq-common';
import { ELuaScriptName } from '../../../common/redis-client/scripts/scripts.js';
import { redisKeys } from '../../../common/redis-keys/redis-keys.js';
import { _getMessage } from '../../message/_/_get-message.js';
import { EMessageProperty, EMessagePropertyStatus, } from '../../message/index.js';
import { EQueueProperty, EQueueType } from '../../queue/index.js';
import { Worker } from './worker.js';
class RequeueUnacknowledgedWorker extends Worker {
work = (cb) => {
const { keyQueueRequeued, keyQueueProperties } = redisKeys.getQueueKeys(this.queueParsedParams.queueParams, this.queueParsedParams.groupId);
const redisClient = this.redisClient.getInstance();
if (redisClient instanceof Error) {
cb(redisClient);
return void 0;
}
redisClient.lrange(keyQueueRequeued, 0, 99, (err, reply) => {
if (err)
cb(err);
else {
const messageIds = reply ?? [];
if (messageIds.length) {
const keys = [keyQueueRequeued, keyQueueProperties];
const argv = [
EQueueProperty.QUEUE_TYPE,
EQueueType.PRIORITY_QUEUE,
EQueueType.LIFO_QUEUE,
EQueueType.FIFO_QUEUE,
EMessageProperty.STATUS,
EMessagePropertyStatus.PENDING,
EMessageProperty.STATE,
];
async.each(messageIds, (messageId, _, done) => {
_getMessage(redisClient, messageId, (err, message) => {
if (err)
done(err);
else if (!message)
cb(new CallbackEmptyReplyError());
else {
const messageId = message.getId();
const messageState = message.getMessageState();
const { keyQueuePriorityPending, keyQueuePending } = redisKeys.getQueueKeys(message.getDestinationQueue(), message.getConsumerGroupId());
const { keyMessage } = redisKeys.getMessageKeys(messageId);
keys.push(keyQueuePriorityPending, keyQueuePending, keyMessage);
messageState.incrAttempts();
const messagePriority = message.producibleMessage.getPriority() ?? '';
argv.push(messageId, messagePriority, JSON.stringify(messageState));
done();
}
});
}, (err) => {
if (err)
cb(err);
else {
redisClient.runScript(ELuaScriptName.REQUEUE_MESSAGE, keys, argv, (err) => cb(err));
}
});
}
else
cb();
}
});
};
}
export default (payload) => new RequeueUnacknowledgedWorker(payload);
//# sourceMappingURL=requeue-unacknowledged.worker.js.map