@autofleet/rabbit 2.0.4-beta → 2.0.4-beta1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autofleet/rabbit",
-  "version": "2.0.4-beta",
+  "version": "2.0.4-beta1",
   "description": "",
   "main": "dist/index.js",
   "scripts": {

package/src/index.ts

@@ -37,16 +37,14 @@ const defaultOptions = {
 
 const assertExchangeFanout = (c: any, exchangeName: string) => c.assertExchange(exchangeName, 'fanout');
 
-const callbackWithTimeout = async (callback: any, args: any[], timeout: number, timeoutMessage = 'Timeout while ack message') => {
-  await new bluebird.Promise(async (resolve: any, reject: any) => {
-    try {
-      await callback(...args);
-      return resolve();
-    } catch (e) {
-      return reject(e);
-    }
-  }).timeout(timeout, timeoutMessage);
-};
+const callbackWithTimeout = async (callback: any, args: any[], timeout: number, timeoutMessage = 'Timeout while ack message') => Promise.race([
+  new Promise(callback(...args)),
+  new Promise((resolve, reject) => {
+    setTimeout(() => {
+      reject(timeoutMessage);
+    }, timeout);
+  }),
+]);
 
 const connectionCreatedConst = 'connectionCreated';
 class RabbitMq implements IAfRabbitMq {
@@ -226,7 +224,7 @@ class RabbitMq implements IAfRabbitMq {
       return Promise.all([
         c.consume(
           queue,
-          (msg: any) => {
+          async (msg: any) => {
             newTrace(traceTypes.RABBIT);
             if (optionsWithDefaults.timeout) {
               const args = [
@@ -234,7 +232,11 @@ class RabbitMq implements IAfRabbitMq {
                 this.ack(channel),
                 this.nack(channel, queue, optionsWithDefaults, { messageTtl: deadMessageTtl }),
               ];
-              callbackWithTimeout(callback, args, optionsWithDefaults.timeout, optionsWithDefaults.timeoutMessage);
+              try {
+                await callbackWithTimeout(callback, args, optionsWithDefaults.timeout, optionsWithDefaults.timeoutMessage);
+              } catch (e) {
+                this.nack(channel, queue, optionsWithDefaults, { messageTtl: deadMessageTtl });
+              }
             } else {
               callback(RabbitMq.parseMsg(msg), this.ack(channel), this.nack(channel, queue, optionsWithDefaults, { messageTtl: deadMessageTtl }));
             }