@azteam/rabbitmq-async 1.0.212 → 1.0.213

package/lib/Provider.js

@@ -19,9 +19,11 @@ var Provider = /*#__PURE__*/function () {
   function Provider() {
     var _this = this;
     var configs = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
+    var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
     _classCallCheck(this, Provider);
     this.connections = {};
     this.configs = {};
+    this.options = options;
     if (Array.isArray(configs)) {
       configs.map(function (config) {
         _this.configs[config.name] = config;
@@ -37,7 +39,7 @@ var Provider = /*#__PURE__*/function () {
       var name = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 'main';
       if (!this.connections[name]) {
         if (this.configs[name]) {
-          this.connections[name] = new _RabbitMQAsync["default"](this.configs[name]);
+          this.connections[name] = new _RabbitMQAsync["default"](this.configs[name], this.options.retryKey || 'RETRY');
         } else {
           return this.getConnection('main');
         }

package/lib/RabbitMQAsync.js

@@ -23,6 +23,7 @@ function _toPrimitive(input, hint) { if (_typeof(input) !== "object" || input ==
 var RabbitMQAsync = /*#__PURE__*/function () {
   function RabbitMQAsync(config) {
     var _this = this;
+    var retryKey = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 'RETRY';
     _classCallCheck(this, RabbitMQAsync);
     this.connected = false;
     this.username = config.username;
@@ -31,6 +32,7 @@ var RabbitMQAsync = /*#__PURE__*/function () {
     this.prefix = config.prefix;
     this.serverIp = config.server_ip;
     this.listConsume = [];
+    this.retryKey = retryKey;
     this.client = null;
     this.channel = null;
     setInterval( /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
@@ -352,7 +354,7 @@ var RabbitMQAsync = /*#__PURE__*/function () {
                           break;
                         case 11:
                           _context7.next = 13;
-                          return messageQueue.send(messageQueue.parsePrefix('RETRY'), {
+                          return messageQueue.send(messageQueue.parsePrefix(this.retryKey), {
                             queueName: queueName,
                             data: data,
                             retry_time: new Date(),
@@ -419,7 +421,7 @@ var RabbitMQAsync = /*#__PURE__*/function () {
                         case "end":
                           return _context7.stop();
                       }
-                    }, _callee7, null, [[3, 41], [17, 24]]);
+                    }, _callee7, this, [[3, 41], [17, 24]]);
                   }));
                   return function (_x5) {
                     return _ref2.apply(this, arguments);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@azteam/rabbitmq-async",
-    "version": "1.0.212",
+    "version": "1.0.213",
     "description": "N/A",
     "keywords": [
         "toda",

package/src/Provider.js

@@ -1,9 +1,12 @@
 import RabbitMQAsync from './RabbitMQAsync';
 
 class Provider {
-    constructor(configs = []) {
+    constructor(configs = [], options = {}) {
         this.connections = {};
         this.configs = {};
+
+        this.options = options;
+
         if (Array.isArray(configs)) {
             configs.map((config) => {
                 this.configs[config.name] = config;
@@ -17,7 +20,7 @@ class Provider {
     getConnection(name = 'main') {
         if (!this.connections[name]) {
             if (this.configs[name]) {
-                this.connections[name] = new RabbitMQAsync(this.configs[name]);
+                this.connections[name] = new RabbitMQAsync(this.configs[name], this.options.retryKey || 'RETRY');
             } else {
                 return this.getConnection('main');
             }

package/src/RabbitMQAsync.js

@@ -4,7 +4,7 @@ import _ from 'lodash';
 import {timeout} from '@azteam/util';
 
 class RabbitMQAsync {
-    constructor(config) {
+    constructor(config, retryKey = 'RETRY') {
         this.connected = false;
         this.username = config.username;
         this.password = config.password;
@@ -14,6 +14,8 @@ class RabbitMQAsync {
 
         this.listConsume = [];
 
+        this.retryKey = retryKey;
+
         this.client = null;
         this.channel = null;
 
@@ -140,7 +142,7 @@ class RabbitMQAsync {
                                             if (data.retry > 0) {
                                                 await messageQueue.send(queueName, data);
                                             } else {
-                                                await messageQueue.send(messageQueue.parsePrefix('RETRY'), {
+                                                await messageQueue.send(messageQueue.parsePrefix(this.retryKey), {
                                                     queueName,
                                                     data,
                                                     retry_time: new Date(),