Retry connection to server indefinitely if retries is set to Infinity…

…. Refs #94.
isaacgr · Aug 22, 2021 · 897ba53 · 897ba53
1 parent b255842
commit 897ba53
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/src/client/protocol/base.js b/src/client/protocol/base.js
@@ -81,12 +81,21 @@ class JsonRpcClientProtocol {
           resolve(this.server);
         });
         this.connector.on("error", (error) => {
-          if (error.code === "ECONNREFUSED" && this.factory.remainingRetries) {
-            this.factory.remainingRetries -= 1;
-            console.error(
-              `Unable to connect. Retrying. ${this.factory.remainingRetries} attempts left.`
-            );
-            setTimeout(() => {
+          if (
+            error.code === "ECONNREFUSED"
+            && this.factory.remainingRetries > 0
+          ) {
+            if (Number.isFinite(this.factory.remainingRetries)) {
+              this.factory.remainingRetries -= 1;
+              console.error(
+                `Failed to connect. Address [${this.server.host}:${this.server.port}]. Retrying. ${this.factory.remainingRetries} attempts left.`
+              );
+            } else {
+              console.error(
+                `Failed to connect. Address [${this.server.host}:${this.server.port}]. Retrying.`
+              );
+            }
+            this.connectionTimeout = setTimeout(() => {
               retryConnection();
             }, this.factory.connectionTimeout);
           } else {