add support for retry limit

interviewstreet · Jan 21, 2019 · bd8abf5 · bd8abf5
1 parent 65cbe1c
commit bd8abf5
Show file tree

Hide file tree

Showing 3 changed files with 1,742 additions and 4 deletions.
diff --git a/client/index.js b/client/index.js
@@ -6,7 +6,7 @@ var SockJS;
  * @param {String} url
  * @constructor
  */
-function Client(url) {
+function Client(url, retryAttempts) {
   /** @type {String} */
   this._url = url;
   /** @type {SockJS} */
@@ -19,8 +19,18 @@ function Client(url) {
   this._heartbeatTimeout = null;
   /** @type {Number} */
   this._reopenTimeout = null;
+  /** @type {Number} */
+  this._retryAttempts = retryAttempts || 6;
+  /** @type {Number} */
+  this._counter = 0;
+  /** @type {Function} */
+  this._errorCallback = null;
 }
 
+Client.prototype.onCaptureError = function(callback) {
+  this._errorCallback = callback;
+};
+
 Client.prototype.open = function() {
   clearTimeout(this._reopenTimeout);
   this._reopenTimeout = null;
@@ -39,8 +49,15 @@ Client.prototype._reopen = function() {
   this._sockJS = null;
 
   this._reopenTimeout = setTimeout(function() {
-    this.open();
-  }.bind(this), 1000);
+    if (this._counter < this._retryAttempts) {
+      this._counter++;
+      this.open();
+    } else {
+      if (typeof this._errorCallback === 'function') {
+        this._errorCallback(new Error('Reached max retryAttempts'));
+      }
+    }
+  }.bind(this), 1000 * this._counter);
 };
 
 Client.prototype._onopen = function() {
@@ -49,6 +66,9 @@ Client.prototype._onopen = function() {
     self._subscribe(channel, self._closeStamp);
   });
 
+  /** Reset counter if connected. */
+  this._counter = 0;
+
   this._closeStamp = null;
   this._sockJS.onmessage = function(event) {
     var data = JSON.parse(event.data);

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "socket-redis",
-  "version": "3.3.0",
+  "version": "3.3.4",
   "description": "Redis to SockJS relay",
   "main": "socket-redis.js",
   "bin": {