@nestjstools/messaging-rabbitmq-extension
Version:
Extension to handle messages and dispatch them over AMQP protocol
95 lines • 4.76 kB
JavaScript
;
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.RabbitmqMessagingConsumer = void 0;
const amqp_channel_1 = require("../channel/amqp.channel");
const const_1 = require("../const");
const messaging_1 = require("@nestjstools/messaging");
const common_1 = require("@nestjs/common");
const messaging_2 = require("@nestjstools/messaging");
const rabbitmq_migrator_1 = require("../migrator/rabbitmq.migrator");
const message_retrier_visitor_1 = require("./message-retrier.visitor");
const message_dead_letter_visitor_1 = require("./message-dead-letter.visitor");
let RabbitmqMessagingConsumer = class RabbitmqMessagingConsumer {
constructor(rabbitMqMigrator, messageRetrier, messageDeadLetter) {
this.rabbitMqMigrator = rabbitMqMigrator;
this.messageRetrier = messageRetrier;
this.messageDeadLetter = messageDeadLetter;
this.channelWrappers = new WeakMap();
this.channels = new Set();
}
async consume(dispatcher, channel) {
this.channels.add(channel);
await this.rabbitMqMigrator.run(channel);
if (!channel.connection) {
throw new Error('There is no active connection to RabbitMQ. Cannot consume messages.');
}
const channelWrapper = channel.createChannelWrapper();
await channelWrapper.waitForConnect();
this.channelWrappers.set(channel, channelWrapper);
await channelWrapper.addSetup(async (rawChannel) => {
await rawChannel.prefetch(channel.config.qos, false);
return rawChannel.consume(channel.config.queue, async (msg) => {
if (!msg)
return;
let payload = msg.content;
if (Buffer.isBuffer(payload)) {
try {
payload = JSON.parse(payload.toString());
}
catch {
rawChannel.nack(msg, false, false);
return;
}
}
const retryCount = msg.properties.headers?.[const_1.RABBITMQ_HEADER_RETRY_COUNT] ?? 0;
const routingKey = msg.properties.headers?.[const_1.RABBITMQ_HEADER_ROUTING_KEY] ?? msg.fields.routingKey;
await dispatcher.dispatch(new messaging_1.ConsumerMessage(payload, routingKey, {
[const_1.RABBITMQ_HEADER_RETRY_COUNT]: retryCount,
}));
rawChannel.ack(msg);
}, { noAck: false });
});
}
async onError(errored, channel) {
const channelWrapper = this.channelWrappers.get(channel);
if (!channelWrapper) {
return;
}
if (channel.config.retryMessage) {
const limit = channel.config.retryMessage;
const currentRetryCount = errored.dispatchedConsumerMessage.metadata[const_1.RABBITMQ_HEADER_RETRY_COUNT] ?? 0;
if (currentRetryCount < limit) {
return this.messageRetrier.retryMessage(errored, channel, channelWrapper, currentRetryCount);
}
}
if (channel.config.deadLetterQueueFeature) {
return this.messageDeadLetter.sendToDeadLetter(errored, channel, channelWrapper);
}
}
async onModuleDestroy() {
for (const channel of this.channels) {
if (channel.connection) {
await channel.connection.close();
}
}
this.channels.clear();
}
};
exports.RabbitmqMessagingConsumer = RabbitmqMessagingConsumer;
exports.RabbitmqMessagingConsumer = RabbitmqMessagingConsumer = __decorate([
(0, common_1.Injectable)(),
(0, messaging_2.MessageConsumer)(amqp_channel_1.AmqpChannel),
__metadata("design:paramtypes", [rabbitmq_migrator_1.RabbitmqMigrator,
message_retrier_visitor_1.MessageRetrierVisitor,
message_dead_letter_visitor_1.MessageDeadLetterVisitor])
], RabbitmqMessagingConsumer);
//# sourceMappingURL=rabbitmq-messaging.consumer.js.map