@autofleet/rabbit 1.4.1 → 1.5.0-1

package/.eslintrc.json

@@ -0,0 +1,66 @@
+{
+  "root": true,
+  "env": {
+    "node": true,
+    "jest": true
+  },
+  "extends": [
+    "airbnb-base",
+    "plugin:@typescript-eslint/recommended"
+  ],
+  "rules": {
+    "import/extensions": [
+      "error",
+      "ignorePackages",
+      {
+        "js": "never",
+        "jsx": "never",
+        "ts": "never",
+        "tsx": "never"
+      }
+    ],
+    "@typescript-eslint/no-unused-vars": ["error"],
+    "@typescript-eslint/no-var-requires": ["off"],
+    "@typescript-eslint/camelcase": ["off"],
+    "max-len": ["off"],
+    "@typescript-eslint/explicit-function-return-type": ["off"]
+  },
+  "overrides": [
+    {
+      // enable the rule specifically for TypeScript files
+      "files": ["src/**.js", "src/**.ts"],
+      "rules": {
+        "@typescript-eslint/explicit-function-return-type": ["off"],
+        "@typescript-eslint/no-var-requires": ["off"],
+        "@typescript-eslint/camelcase": ["off"]
+      }
+    }
+  ],
+  "parser": "@typescript-eslint/parser",
+  "parserOptions": {
+    "project": "./tsconfig.json",
+    "sourceType": "module"
+  },
+  "ignorePatterns": ["dist/*", "node_modules/", "jest.config.js", "migrations/*", "newrelic.js"],
+  "plugins": ["@typescript-eslint"],
+  "settings": {
+    "import/extensions": [
+      "error",
+      "ignorePackages",
+      {
+        "js": "never",
+        "jsx": "never",
+        "ts": "never",
+        "tsx": "never"
+      }
+    ],
+    "import/parsers": {
+      "@typescript-eslint/parser": [".ts",".tsx"]
+    },
+    "import/resolver": {
+      "node": {
+        "extensions": [".js",".jsx",".ts",".tsx", ".json"]
+      }
+    }
+  }
+}

package/dist/example.js

@@ -1,22 +1,33 @@
 "use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 const index_1 = require("./index");
-const init = async () => {
+// process.env.RABBITMQ_SERVICE_HOST="34.76.223.229:5672";
+const init = () => __awaiter(void 0, void 0, void 0, function* () {
     const rabbit = new index_1.default();
-    await rabbit.consumeFromExchange('test-q2', 'test-e', (msg, ack) => {
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    yield rabbit.consumeFromExchange('test-q2', 'test-e', (msg, ack) => {
         console.log('msg2', msg.content);
         // ack(msg);
     });
-    await rabbit.consumeFromExchange('test-q', 'test-e', (msg, ack) => {
+    yield rabbit.consumeFromExchange('test-q', 'test-e', (msg, ack) => {
         console.log('msg', msg.content);
         ack(msg);
     });
-    await rabbit.publish('test-e', 'hey');
-    await rabbit.publish('test-e', 'hey1');
-    await rabbit.consume('test-q-2', (msg, ack) => {
+    yield rabbit.publish('test-e', 'hey');
+    yield rabbit.publish('test-e', 'hey1');
+    yield rabbit.consume('test-q-2', (msg, ack) => {
         console.log('msg-q', msg.content);
         ack(msg);
     });
-    await rabbit.sendToQueue('test-q-2', 'hey-q');
-};
+    yield rabbit.sendToQueue('test-q-2', 'hey-q');
+});
 init();

package/dist/index.d.ts

@@ -6,10 +6,14 @@ interface ExchangesCache {
 }
 declare class RabbitMq {
     static parseMsg(msg: any): any;
+    static validateName(type: string, name: string): void;
+    PUBLISH_TIMEOUT: number;
+    PUBLISH_ERROR_MSG: string;
+    DISCONNECT_MSG: string;
     channel: ChannelWrapper | null;
     connection: AmqpConnectionManager | null;
     em: EventEmitter;
-    creatingConnection: Boolean;
+    creatingConnection: boolean;
     exchanges: ExchangesCache;
     constructor();
     ack: (msg: any) => Promise<void>;
@@ -20,7 +24,7 @@ declare class RabbitMq {
     assertQueue(queue: string, options?: any): Promise<void>;
     consume(queue: string, callback: (msg: any, ack: Function, nack: Function) => any, options?: any): Promise<void>;
     consumeFromExchange(queue: string, exchange: string, callback: (msg: any, ack: Function, nack: Function) => any, options?: any): Promise<void>;
-    publish(exchange: string, content: any): Promise<void>;
+    publish(exchange: string, content: any): Promise<unknown>;
     sendToQueue(queue: string, content: any): Promise<void>;
 }
 export default RabbitMq;

package/dist/index.js

@@ -1,30 +1,46 @@
 "use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
 Object.defineProperty(exports, "__esModule", { value: true });
+// eslint-disable-next-line max-classes-per-file
+const bluebird = require("bluebird");
 const amqp_connection_manager_1 = require("amqp-connection-manager");
 const events_1 = require("events");
+const rabbitError_1 = require("./rabbitError");
 const connectionCreatedConst = 'connectionCreated';
 class RabbitMq {
     constructor() {
-        this.ack = async (msg) => {
+        this.PUBLISH_TIMEOUT = Number(process.env.RABBITMQ_PUBLISH_TIMEOUT) || 60000;
+        this.PUBLISH_ERROR_MSG = `rabbit: publish timeout(${this.PUBLISH_TIMEOUT}ms) has pass, exchange: `;
+        this.DISCONNECT_MSG = 'rabbit: connection disconnect';
+        this.ack = (msg) => __awaiter(this, void 0, void 0, function* () {
             if (this.channel) {
-                await this.channel.ack(msg);
+                yield this.channel.ack(msg);
             }
-        };
-        this.nack = (queue) => async (msg, retries) => {
+        });
+        this.nack = (queue) => (msg, retries) => __awaiter(this, void 0, void 0, function* () {
             if (this.channel) {
                 if (!msg.content['x-retry-count'] || msg.content['x-retry-count'] <= retries) {
+                    // eslint-disable-next-line no-param-reassign
                     msg.content['x-retry-count'] = msg.content['x-retry-count'] ? msg.content['x-retry-count'] + 1 : 1;
-                    await this.sendToQueue(queue, msg.content);
+                    yield this.sendToQueue(queue, msg.content);
                 }
                 else {
                     const deadQueue = `${queue}-dead`;
-                    await this.assertQueue(deadQueue);
-                    await this.sendToQueue(deadQueue, msg.content);
+                    yield this.assertQueue(deadQueue);
+                    yield this.sendToQueue(deadQueue, msg.content);
                 }
-                await this.channel.ack(msg);
+                yield this.channel.ack(msg);
             }
-        };
-        this.em = new events_1.EventEmitter;
+        });
+        this.em = new events_1.EventEmitter();
         this.channel = null;
         this.connection = null;
         this.creatingConnection = false;
@@ -36,94 +52,132 @@ class RabbitMq {
         try {
             content = JSON.parse(content);
         }
-        catch (e) { }
-        return Object.assign({}, msg, { content });
+        catch (e) { } // eslint-disable-line no-empty
+        return Object.assign(Object.assign({}, msg), { content });
     }
-    async getConnection() {
-        return new Promise(async (resolve) => {
-            if (this.creatingConnection) {
-                this.em.on(connectionCreatedConst, resolve);
-                return;
-            }
-            if (this.connection !== null) {
-                return this.connection;
-            }
-            this.creatingConnection = true;
-            let host = process.env.RABBITMQ_SERVICE_HOST || '35.240.13.28';
-            let connection = await amqp_connection_manager_1.connect([`amqp://${host}`]);
-            this.connection = connection;
-            this.creatingConnection = false;
-            this.em.emit(connectionCreatedConst, connection);
-            resolve(connection);
+    static validateName(type, name) {
+        if (!name || name === '') {
+            throw new rabbitError_1.default(`error while using ${type} with no name`);
+        }
+    }
+    getConnection() {
+        return __awaiter(this, void 0, void 0, function* () {
+            // eslint-disable-next-line no-async-promise-executor
+            return new Promise((resolve) => __awaiter(this, void 0, void 0, function* () {
+                if (this.creatingConnection) {
+                    this.em.on(connectionCreatedConst, resolve);
+                    return;
+                }
+                if (this.connection !== null) {
+                    // eslint-disable-next-line consistent-return
+                    return this.connection;
+                }
+                this.creatingConnection = true;
+                const host = process.env.RABBITMQ_SERVICE_HOST || '';
+                const connection = yield amqp_connection_manager_1.connect([`amqp://${host}`]);
+                this.connection = connection;
+                this.connection.on('disconnect', ({ err }) => console.error(`${this.DISCONNECT_MSG}${err && ` - ${err}`}`));
+                this.creatingConnection = false;
+                this.em.emit(connectionCreatedConst, connection);
+                resolve(connection);
+            }));
         });
     }
-    async assertChannel() {
-        return new Promise(async (resolve, reject) => {
-            if (this.channel) {
-                return resolve(this.channel);
-            }
-            let connection = await this.getConnection();
-            try {
-                if (this.channel === null) {
-                    this.channel = connection.createChannel({});
+    assertChannel() {
+        return __awaiter(this, void 0, void 0, function* () {
+            // eslint-disable-next-line no-async-promise-executor,consistent-return
+            return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
+                if (this.channel) {
+                    return resolve(this.channel);
                 }
-                resolve(this.channel);
-            }
-            catch (e) {
-                reject(e);
-            }
+                const connection = yield this.getConnection();
+                try {
+                    if (this.channel === null) {
+                        this.channel = connection.createChannel({});
+                    }
+                    resolve(this.channel);
+                }
+                catch (e) {
+                    reject(e);
+                }
+            }));
         });
     }
-    async assertExchange(exchangeName, options) {
-        const channel = await this.assertChannel();
-        if (this.exchanges[exchangeName]) {
-            return this.exchanges[exchangeName];
-        }
-        return channel.addSetup(async (c) => {
-            const exchange = await c.assertExchange(exchangeName, 'fanout');
-            this.exchanges[exchangeName] = exchange;
-            return exchange;
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    assertExchange(exchangeName, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const channel = yield this.assertChannel();
+            if (this.exchanges[exchangeName]) {
+                return this.exchanges[exchangeName];
+            }
+            return channel.addSetup((c) => __awaiter(this, void 0, void 0, function* () {
+                const exchange = yield c.assertExchange(exchangeName, 'fanout');
+                this.exchanges[exchangeName] = exchange;
+                return exchange;
+            }));
         });
     }
-    async assertQueue(queue, options) {
-        const channel = await this.assertChannel();
-        return channel.addSetup((c) => c.assertQueue(queue));
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    assertQueue(queue, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            RabbitMq.validateName('queue', queue);
+            const channel = yield this.assertChannel();
+            return channel.addSetup((c) => c.assertQueue(queue));
+        });
     }
-    async consume(queue, callback, options) {
-        const { limit } = (options && options.limit) ? options : { limit: 1 };
-        const channel = await this.assertChannel();
-        await this.assertQueue(queue);
-        return channel.addSetup(async (c) => {
-            await c.prefetch(limit, true);
-            return Promise.all([
-                c.consume(queue, (msg) => callback(RabbitMq.parseMsg(msg), this.ack, this.nack(queue))),
-            ]);
+    // eslint-disable-next-line @typescript-eslint/ban-types
+    consume(queue, callback, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            RabbitMq.validateName('queue', queue);
+            const { limit } = (options && options.limit) ? options : { limit: 1 };
+            const channel = yield this.assertChannel();
+            yield this.assertQueue(queue);
+            return channel.addSetup((c) => __awaiter(this, void 0, void 0, function* () {
+                yield c.prefetch(limit, true);
+                return Promise.all([
+                    c.consume(queue, (msg) => callback(RabbitMq.parseMsg(msg), this.ack, this.nack(queue))),
+                ]);
+            }));
         });
     }
-    async consumeFromExchange(queue, exchange, callback, options) {
-        const { limit } = (options && options.limit) ? options : { limit: 1 };
-        const channel = await this.assertChannel();
-        await Promise.all([
-            this.assertExchange(exchange),
-            this.assertQueue(queue),
-        ]);
-        return channel.addSetup(async (c) => {
-            await c.prefetch(limit, true);
-            return Promise.all([
-                c.bindQueue(queue, exchange, ''),
-                c.consume(queue, (msg) => callback(RabbitMq.parseMsg(msg), this.ack, this.nack(queue))),
+    // eslint-disable-next-line @typescript-eslint/ban-types
+    consumeFromExchange(queue, exchange, callback, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            RabbitMq.validateName('queue', queue);
+            RabbitMq.validateName('exchange', exchange);
+            const { limit } = (options && options.limit) ? options : { limit: 1 };
+            const channel = yield this.assertChannel();
+            yield Promise.all([
+                this.assertExchange(exchange),
+                this.assertQueue(queue),
             ]);
+            return channel.addSetup((c) => __awaiter(this, void 0, void 0, function* () {
+                yield c.prefetch(limit, true);
+                return Promise.all([
+                    c.bindQueue(queue, exchange, ''),
+                    c.consume(queue, (msg) => callback(RabbitMq.parseMsg(msg), this.ack, this.nack(queue))),
+                ]);
+            }));
         });
     }
-    async publish(exchange, content) {
-        const channel = await this.assertChannel();
-        await this.assertExchange(exchange);
-        return channel.publish(exchange, '', Buffer.from(JSON.stringify(content)));
+    publish(exchange, content) {
+        return __awaiter(this, void 0, void 0, function* () {
+            RabbitMq.validateName('exchange', exchange);
+            const channel = yield this.assertChannel();
+            yield this.assertExchange(exchange);
+            return new bluebird.Promise((resolve) => __awaiter(this, void 0, void 0, function* () {
+                yield channel.publish(exchange, '', Buffer.from(JSON.stringify(content)));
+                return resolve();
+            })).timeout(this.PUBLISH_TIMEOUT, `${this.PUBLISH_ERROR_MSG}${exchange}`);
+        });
     }
-    async sendToQueue(queue, content) {
-        const channel = await this.assertChannel();
-        await this.assertQueue(queue);
-        return channel.sendToQueue(queue, Buffer.from(JSON.stringify(content)));
+    sendToQueue(queue, content) {
+        return __awaiter(this, void 0, void 0, function* () {
+            RabbitMq.validateName('queue', queue);
+            const channel = yield this.assertChannel();
+            yield this.assertQueue(queue);
+            return channel.sendToQueue(queue, Buffer.from(JSON.stringify(content)));
+        });
     }
 }
 exports.default = RabbitMq;

package/dist/index.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/index.test.js

@@ -0,0 +1,64 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const index_1 = require("./index");
+// import RabbitError from './rabbitError';
+jest.setTimeout(120000);
+const rabbit = new index_1.default();
+const delay = () => new Promise((resolve) => setTimeout(() => resolve(), 500));
+describe('RabbitMQ', () => {
+    it('check exchange', () => __awaiter(void 0, void 0, void 0, function* () {
+        const callback = (msg, ack) => {
+            expect(msg.content).toEqual('hey');
+            ack(msg);
+        };
+        const mockFn = jest.fn(callback);
+        yield rabbit.consumeFromExchange('test-q2', 'test-e', mockFn);
+        yield rabbit.publish('test-e', 'hey');
+        yield delay();
+        expect(mockFn).toBeCalled();
+    }));
+    it('check queue', () => __awaiter(void 0, void 0, void 0, function* () {
+        const callback = (msg, ack) => {
+            expect(msg.content).toEqual('hey-q');
+            ack(msg);
+        };
+        const mockFn = jest.fn(callback);
+        yield rabbit.consume('test-q-2', mockFn);
+        yield rabbit.sendToQueue('test-q-2', 'hey-q');
+        yield delay();
+        expect(mockFn).toBeCalled();
+    }));
+    it('cant publish to unknown exchange', () => __awaiter(void 0, void 0, void 0, function* () {
+        let err;
+        try {
+            yield rabbit.publish('', 'hey');
+        }
+        catch (e) {
+            err = e;
+        }
+        expect(err).toBeDefined();
+    }));
+    it('cant publish due timeout', () => __awaiter(void 0, void 0, void 0, function* () {
+        // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+        // @ts-ignore
+        process.env.RABBITMQ_PUBLISH_TIMEOUT = 1;
+        const rabbit2 = new index_1.default();
+        let err;
+        try {
+            yield rabbit2.publish('test-e', 'hey');
+        }
+        catch (e) {
+            err = e;
+        }
+        expect(err).toBeDefined();
+    }));
+});

package/dist/mock.d.ts

@@ -1,13 +1,13 @@
 import 'jest';
 declare class RabbitMq {
-    ack: jest.Mock<{}>;
-    nack: jest.Mock<{}>;
-    assertChannel: jest.Mock<{}>;
-    assertExchange: jest.Mock<{}>;
-    assertQueue: jest.Mock<{}>;
-    consume: jest.Mock<{}>;
-    consumeFromExchange: jest.Mock<{}>;
-    publish: jest.Mock<{}>;
-    sendToQueue: jest.Mock<{}>;
+    ack: jest.Mock<any, any>;
+    nack: jest.Mock<any, any>;
+    assertChannel: jest.Mock<any, any>;
+    assertExchange: jest.Mock<any, any>;
+    assertQueue: jest.Mock<any, any>;
+    consume: jest.Mock<any, any>;
+    consumeFromExchange: jest.Mock<any, any>;
+    publish: jest.Mock<any, any>;
+    sendToQueue: jest.Mock<any, any>;
 }
 export default RabbitMq;

package/dist/rabbitError.d.ts

@@ -0,0 +1,3 @@
+export default class RabbitError extends Error {
+    constructor(message: string);
+}

package/dist/rabbitError.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+class RabbitError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'RabbitError';
+    }
+}
+exports.default = RabbitError;

package/jest.config.js

@@ -0,0 +1,14 @@
+module.exports = {
+  testEnvironment: 'node',
+  roots: ['<rootDir>/src'],
+  transform: {
+    '^.+\\.tsx?$': 'ts-jest',
+  },
+  testRegex: '(/__tests__/.*|(\\.|/)(test|spec))\\.(tsx?$|jsx?$)',
+  moduleFileExtensions: ['ts', 'tsx', 'js', 'jsx', 'json', 'node'],
+  coverageThreshold: {
+    global: {
+      lines: 75,
+    },
+  },
+}

package/package.json

@@ -1,25 +1,35 @@
 {
   "name": "@autofleet/rabbit",
-  "version": "1.4.1",
+  "version": "1.5.0-1",
   "description": "",
   "main": "dist/index.js",
   "scripts": {
     "start": "ts-node  src/index.ts",
     "example": "ts-node  src/example.ts",
     "prepublish": "tsc",
+    "linter": "./node_modules/.bin/eslint .",
+    "test": "jest --forceExit",
+    "test-local": "RABBITMQ_SERVICE_HOST=34.76.223.229:5672 jest --forceExit",
+    "coverage": "jest --coverage --forceExit --runInBand && rm -rf ./coverage",
     "dev": "nodemon"
   },
   "dependencies": {
+    "@types/jest": "^23.0.0",
     "@types/amqp-connection-manager": "^2.0.2",
     "@types/amqplib": "^0.5.9",
-    "@types/jest": "^23.3.10",
     "amqp-connection-manager": "^3.2.1",
     "amqplib": "^0.6.0",
-    "jest": "^23.6.0"
+    "bluebird": "^3.7.2",
+    "jest": "^25.3.0"
   },
   "devDependencies": {
-    "ts-node": "^7.0.1",
-    "typescript": "^3.2.1"
+    "@typescript-eslint/eslint-plugin": "^4.8.1",
+    "eslint": "^7.13.0",
+    "eslint-config-airbnb-typescript": "^12.0.0",
+    "eslint-plugin-import": "^2.22.1",
+    "ts-jest": "^25.4.0",
+    "ts-node": "^8.6.2",
+    "typescript": "^3.8.3"
   },
   "author": "",
   "license": "ISC"

package/src/example.ts

@@ -1,7 +1,10 @@
 import RabbitMq from './index';
 
+// process.env.RABBITMQ_SERVICE_HOST="34.76.223.229:5672";
+
 const init = async () => {
   const rabbit = new RabbitMq();
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   await rabbit.consumeFromExchange('test-q2', 'test-e', (msg, ack) => {
     console.log('msg2', msg.content);
     // ack(msg);
@@ -11,15 +14,14 @@ const init = async () => {
     console.log('msg', msg.content);
     ack(msg);
   });
-  await rabbit.publish('test-e', 'hey')
-  await rabbit.publish('test-e', 'hey1')
-
+  await rabbit.publish('test-e', 'hey');
+  await rabbit.publish('test-e', 'hey1');
 
   await rabbit.consume('test-q-2', (msg, ack) => {
     console.log('msg-q', msg.content);
     ack(msg);
   });
-  await rabbit.sendToQueue('test-q-2', 'hey-q')
-}
+  await rabbit.sendToQueue('test-q-2', 'hey-q');
+};
 
-init();
+init();

package/src/index.test.ts

@@ -0,0 +1,59 @@
+import RabbitMq from './index';
+// import RabbitError from './rabbitError';
+
+jest.setTimeout(120000);
+
+const rabbit = new RabbitMq();
+
+const delay = () => new Promise((resolve) => setTimeout(() => resolve(), 500));
+
+describe('RabbitMQ', () => {
+  it('check exchange', async () => {
+    const callback = (msg: any, ack: any) => {
+      expect(msg.content).toEqual('hey');
+      ack(msg);
+    };
+    const mockFn = jest.fn(callback);
+    await rabbit.consumeFromExchange('test-q2', 'test-e', mockFn);
+    await rabbit.publish('test-e', 'hey');
+    await delay();
+    expect(mockFn).toBeCalled();
+  });
+
+  it('check queue', async () => {
+    const callback = (msg: any, ack: any) => {
+      expect(msg.content).toEqual('hey-q');
+      ack(msg);
+    };
+    const mockFn = jest.fn(callback);
+    await rabbit.consume('test-q-2', mockFn);
+    await rabbit.sendToQueue('test-q-2', 'hey-q');
+    await delay();
+    expect(mockFn).toBeCalled();
+  });
+
+  it('cant publish to unknown exchange', async () => {
+    let err;
+    try {
+      await rabbit.publish('', 'hey');
+    } catch (e) {
+      err = e;
+    }
+    expect(err).toBeDefined();
+  });
+
+  it('cant publish due timeout', async () => {
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-ignore
+    process.env.RABBITMQ_PUBLISH_TIMEOUT = 1;
+    const rabbit2 = new RabbitMq();
+
+    let err;
+    try {
+      await rabbit2.publish('test-e', 'hey');
+    } catch (e) {
+      err = e;
+    }
+    expect(err).toBeDefined();
+  });
+});

package/src/index.ts

@@ -1,6 +1,9 @@
+// eslint-disable-next-line max-classes-per-file
+import * as bluebird from 'bluebird';
 import { ConfirmChannel } from 'amqplib';
-import { connect, AmqpConnectionManager, ChannelWrapper } from 'amqp-connection-manager';
+import { AmqpConnectionManager, ChannelWrapper, connect } from 'amqp-connection-manager';
 import { EventEmitter } from 'events';
+import RabbitError from './rabbitError';
 
 interface ExchangesCache {
   [key: string]: any
@@ -14,22 +17,38 @@ class RabbitMq {
 
     try {
       content = JSON.parse(content);
-    } catch (e) { }
+    } catch (e) { } // eslint-disable-line no-empty
 
     return {
       ...msg,
       content,
+    };
+  }
+
+  static validateName(type: string, name: string) {
+    if (!name || name === '') {
+      throw new RabbitError(`error while using ${type} with no name`);
     }
   }
-  
+
+  PUBLISH_TIMEOUT = Number(process.env.RABBITMQ_PUBLISH_TIMEOUT) || 60000;
+
+  PUBLISH_ERROR_MSG = `rabbit: publish timeout(${this.PUBLISH_TIMEOUT}ms) has pass, exchange: `;
+
+  DISCONNECT_MSG = 'rabbit: connection disconnect';
+
   channel: ChannelWrapper | null;
+
   connection: AmqpConnectionManager | null;
+
   em: EventEmitter;
-  creatingConnection: Boolean;
+
+  creatingConnection: boolean;
+
   exchanges: ExchangesCache;
 
   constructor() {
-    this.em = new EventEmitter;
+    this.em = new EventEmitter();
     this.channel = null;
     this.connection = null;
     this.creatingConnection = false;
@@ -45,6 +64,7 @@ class RabbitMq {
   public nack = (queue: string) => async (msg: any, retries: number) => {
     if (this.channel) {
       if (!msg.content['x-retry-count'] || msg.content['x-retry-count'] <= retries) {
+        // eslint-disable-next-line no-param-reassign
         msg.content['x-retry-count'] = msg.content['x-retry-count'] ? msg.content['x-retry-count'] + 1 : 1;
         await this.sendToQueue(queue, msg.content);
       } else {
@@ -57,31 +77,36 @@ class RabbitMq {
   }
 
   async getConnection() {
+    // eslint-disable-next-line no-async-promise-executor
     return new Promise<AmqpConnectionManager>(async (resolve) => {
       if (this.creatingConnection) {
         this.em.on(connectionCreatedConst, resolve);
         return;
       }
       if (this.connection !== null) {
+        // eslint-disable-next-line consistent-return
         return this.connection;
       }
       this.creatingConnection = true;
-      let host: string = process.env.RABBITMQ_SERVICE_HOST || '35.240.13.28';
-      let connection: AmqpConnectionManager = await connect([`amqp://${host}`]);
+      const host: string = process.env.RABBITMQ_SERVICE_HOST || '';
+      const connection: AmqpConnectionManager = await connect([`amqp://${host}`]);
       this.connection = connection;
+      this.connection.on('disconnect',
+        ({ err }) => console.error(`${this.DISCONNECT_MSG}${err && ` - ${err}`}`));
       this.creatingConnection = false;
       this.em.emit(connectionCreatedConst, connection);
       resolve(connection);
-    })
+    });
   }
 
   async assertChannel() {
+    // eslint-disable-next-line no-async-promise-executor,consistent-return
     return new Promise<ChannelWrapper>(async (resolve, reject) => {
-      if(this.channel) {
+      if (this.channel) {
         return resolve(this.channel);
       }
 
-      let connection: AmqpConnectionManager = await this.getConnection();
+      const connection: AmqpConnectionManager = await this.getConnection();
 
       try {
         if (this.channel === null) {
@@ -89,29 +114,34 @@ class RabbitMq {
         }
         resolve(this.channel);
       } catch (e) {
-        reject(e)
+        reject(e);
       }
-    })
+    });
   }
 
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async assertExchange(exchangeName: string, options?: any) {
     const channel: ChannelWrapper = await this.assertChannel();
-    if(this.exchanges[exchangeName]) {
-      return this.exchanges[exchangeName]
+    if (this.exchanges[exchangeName]) {
+      return this.exchanges[exchangeName];
     }
     return channel.addSetup(async (c: ConfirmChannel) => {
-      const exchange = await c.assertExchange(exchangeName, 'fanout')
+      const exchange = await c.assertExchange(exchangeName, 'fanout');
       this.exchanges[exchangeName] = exchange;
       return exchange;
     });
   }
 
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async assertQueue(queue: string, options?: any) {
+    RabbitMq.validateName('queue', queue);
     const channel: ChannelWrapper = await this.assertChannel();
     return channel.addSetup((c: ConfirmChannel) => c.assertQueue(queue));
   }
 
+  // eslint-disable-next-line @typescript-eslint/ban-types
   async consume(queue: string, callback: (msg: any, ack: Function, nack: Function) => any, options?: any) {
+    RabbitMq.validateName('queue', queue);
     const { limit } = (options && options.limit) ? options : { limit: 1 };
     const channel: ChannelWrapper = await this.assertChannel();
     await this.assertQueue(queue);
@@ -123,34 +153,42 @@ class RabbitMq {
     });
   }
 
+  // eslint-disable-next-line @typescript-eslint/ban-types
   async consumeFromExchange(queue: string, exchange: string, callback: (msg: any, ack: Function, nack: Function) => any, options?: any) {
+    RabbitMq.validateName('queue', queue);
+    RabbitMq.validateName('exchange', exchange);
     const { limit } = (options && options.limit) ? options : { limit: 1 };
     const channel: ChannelWrapper = await this.assertChannel();
     await Promise.all([
       this.assertExchange(exchange),
       this.assertQueue(queue),
     ]);
-    
+
     return channel.addSetup(async (c: ConfirmChannel) => {
       await c.prefetch(limit, true);
       return Promise.all([
         c.bindQueue(queue, exchange, ''),
         c.consume(queue, (msg: any) => callback(RabbitMq.parseMsg(msg), this.ack, this.nack(queue))),
-      ])
-    })
+      ]);
+    });
   }
 
   async publish(exchange: string, content: any) {
+    RabbitMq.validateName('exchange', exchange);
     const channel: ChannelWrapper = await this.assertChannel();
     await this.assertExchange(exchange);
-    return channel.publish(exchange, '', Buffer.from(JSON.stringify(content)));
+    return new bluebird.Promise(async (resolve) => {
+      await channel.publish(exchange, '', Buffer.from(JSON.stringify(content)));
+      return resolve();
+    }).timeout(this.PUBLISH_TIMEOUT, `${this.PUBLISH_ERROR_MSG}${exchange}`);
   }
 
   async sendToQueue(queue: string, content: any) {
+    RabbitMq.validateName('queue', queue);
     const channel: ChannelWrapper = await this.assertChannel();
     await this.assertQueue(queue);
     return channel.sendToQueue(queue, Buffer.from(JSON.stringify(content)));
   }
 }
 
-export default RabbitMq;
+export default RabbitMq;

package/src/mock.ts

@@ -2,14 +2,22 @@ import 'jest';
 
 class RabbitMq {
   public ack = jest.fn();
+
   public nack = jest.fn();
+
   public assertChannel = jest.fn();
+
   public assertExchange = jest.fn();
+
   public assertQueue = jest.fn();
+
   public consume = jest.fn();
+
   public consumeFromExchange = jest.fn();
+
   public publish = jest.fn();
+
   public sendToQueue = jest.fn();
 }
 
-export default RabbitMq;
+export default RabbitMq;

package/src/rabbitError.ts

@@ -0,0 +1,6 @@
+export default class RabbitError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = 'RabbitError';
+  }
+}

package/tsconfig.json

@@ -1,12 +1,9 @@
 {
   "compilerOptions": {
-    "target": "es2017",
+    "target": "es6",
     "module": "commonjs",
-    "lib": [
-      "es2017"
-    ],
     "declaration": true,
     "outDir": "./dist",
     "strict": true
   }
-}
+}