sb-rabbitmq-provider 1.0.21 → 1.0.25

package/dist/index.d.ts

@@ -15,13 +15,16 @@ interface rabbitMQConfig {
     vhost?: string | undefined;
     exchangeName?: string;
     exchangeType: string;
+    dropMessageOnChannelFailure?: boolean;
 }
 type MyCallback = (data: any) => any;
 declare class RabbitMqConnection {
     config: rabbitMQConfig;
     exchange: string;
     exchangeType: string;
+    dropMessageOnChannelFailure: boolean;
     connection: any;
+    channel: any;
     /**
      * constructor for initialising url, exchange name, exchange type
      * @param url rabbitmq url
@@ -38,6 +41,7 @@ declare class RabbitMqConnection {
     /**
      * single connection can have multiple channel like publishChannel and consumeChannel
      * @returns channel object
+     * No need to use this method, as channel is created in createConnection method.
      */
     createChannel(): Promise<RabbitMQChannel>;
     /**
@@ -60,7 +64,10 @@ declare class RabbitMqConnection {
      * @param callback function for handling the data
      * @returns
      */
-    consume(channel: RabbitMQChannel, queueName: string, onMessage: MyCallback, options: Options.Consume): Promise<{
+    consume(channel: RabbitMQChannel, queueName: string, onMessage: MyCallback, options: Options.Consume, queueOptions?: {
+        durable: boolean;
+        autoDelete: boolean;
+    }): Promise<{
         status: boolean;
         message: string;
     }>;

package/dist/index.js

@@ -27,12 +27,16 @@ class RabbitMqConnection {
     constructor(config) {
         this.exchange = 'EXCHANGE';
         this.exchangeType = 'direct';
+        this.dropMessageOnChannelFailure = false;
         if (config.exchangeName && typeof config.exchangeName === 'string') {
             this.exchange = config.exchangeName;
         }
         if (config.exchangeType && typeof config.exchangeType === 'string') {
             this.exchangeType = config.exchangeType;
         }
+        if (config.dropMessageOnChannelFailure) {
+            this.dropMessageOnChannelFailure = true;
+        }
         this.config = config;
     }
     /**
@@ -44,6 +48,11 @@ class RabbitMqConnection {
         return __awaiter(this, void 0, void 0, function* () {
             try {
                 const connection = yield amqplib_1.default.connect(this.config);
+                if (this.exchange && this.exchangeType) {
+                    const channel = yield this.connection.createChannel();
+                    yield channel.assertExchange(this.exchange, this.exchangeType);
+                    this.channel = channel;
+                }
                 // console.log('---RabbitMQ Connection Created---')
                 this.connection = connection;
                 return connection;
@@ -56,6 +65,7 @@ class RabbitMqConnection {
     /**
      * single connection can have multiple channel like publishChannel and consumeChannel
      * @returns channel object
+     * No need to use this method, as channel is created in createConnection method.
      */
     createChannel() {
         return __awaiter(this, void 0, void 0, function* () {
@@ -97,14 +107,25 @@ class RabbitMqConnection {
      * @param callback function for handling the data
      * @returns
      */
-    consume(channel, queueName, onMessage, options) {
+    consume(channel, queueName, onMessage, options, queueOptions = { durable: true, autoDelete: false }) {
         return __awaiter(this, void 0, void 0, function* () {
             try {
-                const { queue } = yield channel.assertQueue(queueName, { durable: true, autoDelete: false });
+                const { queue } = yield channel.assertQueue(queueName, queueOptions);
                 yield channel.bindQueue(queue, this.exchange, queueName);
                 channel.consume(queueName, (message) => {
-                    // channel.ack(message);
-                    onMessage(message);
+                    try {
+                        // channel.ack(message);
+                        onMessage(message);
+                    }
+                    catch (e) {
+                        console.log('Rabbitmq consume catch block ', e);
+                        if (message !== null) {
+                            if (this.dropMessageOnChannelFailure) {
+                                return channel.nack(message, false, false);
+                            }
+                            return channel.nack(message);
+                        }
+                    }
                 }, options);
                 return { status: true, message: `Started consuming Queue: ${queueName}` };
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sb-rabbitmq-provider",
-  "version": "1.0.21",
+  "version": "1.0.25",
   "description": "Pub/Sub model for RabbitMq",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",