From 23f8a73ac38a2537c8185a67f32be69729533096 Mon Sep 17 00:00:00 2001
From: Owen Pearson <owen.pearson@ably.com>
Date: Tue, 27 Jun 2023 17:01:11 +0100
Subject: [PATCH] fix(ConnectionManager): pass connectCount through connectImpl

---
 src/common/lib/transport/connectionmanager.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/common/lib/transport/connectionmanager.ts b/src/common/lib/transport/connectionmanager.ts
index 7f12205900..cfd765b2b3 100644
--- a/src/common/lib/transport/connectionmanager.ts
+++ b/src/common/lib/transport/connectionmanager.ts
@@ -1529,19 +1529,19 @@ class ConnectionManager extends EventEmitter {
     } else if (state == this.states.connected.state) {
       this.upgradeIfNeeded(transportParams);
     } else if (this.transports.length > 1 && this.getTransportPreference()) {
-      this.connectPreference(transportParams);
+      this.connectPreference(transportParams, connectCount);
     } else {
       this.connectBase(transportParams, connectCount);
     }
   }
 
-  connectPreference(transportParams: TransportParams): void {
+  connectPreference(transportParams: TransportParams, connectCount?: number): void {
     const preference = this.getTransportPreference();
     let preferenceTimeoutExpired = false;
 
     if (!Utils.arrIn(this.transports, preference)) {
       this.unpersistTransportPreference();
-      this.connectImpl(transportParams);
+      this.connectImpl(transportParams, connectCount);
     }
 
     Logger.logAction(
@@ -1564,7 +1564,7 @@ class ConnectionManager extends EventEmitter {
         /* Be quite agressive about clearing the stored preference if ever it doesn't work */
         this.unpersistTransportPreference();
       }
-      this.connectImpl(transportParams);
+      this.connectImpl(transportParams, connectCount);
     }, this.options.timeouts.preferenceConnectTimeout);
 
     /* For connectPreference, just use the main host. If host fallback is needed, do it in connectBase.
@@ -1582,7 +1582,7 @@ class ConnectionManager extends EventEmitter {
       } else if (!transport && !fatal) {
         /* Preference failed in a transport-specific way. Try more */
         this.unpersistTransportPreference();
-        this.connectImpl(transportParams);
+        this.connectImpl(transportParams, connectCount);
       }
       /* If suceeded, or failed fatally, nothing to do */
     });