Refactor connect (#37)

Signed-off-by: Piotr Piotrowski <piotr@synadia.com>
nats-io · Feb 21, 2024 · 9aa1350 · 9aa1350
1 parent c6f6cef
commit 9aa1350
Show file tree

Hide file tree

Showing 5 changed files with 247 additions and 275 deletions.
diff --git a/Sources/NatsSwift/NatsClient/NatsClient.swift b/Sources/NatsSwift/NatsClient/NatsClient.swift
@@ -58,13 +58,15 @@ public class Client {
 
 extension Client {
     public func connect() async throws {
-        //TODO(jrm): reafactor for reconnection and review error handling.
-        //TODO(jrm): handle response
         logger.debug("connect")
         guard let connectionHandler = self.connectionHandler else {
             throw NatsClientError("internal error: empty connection handler")
         }
-        try await connectionHandler.connect()
+        if !connectionHandler.retryOnFailedConnect {
+            try await connectionHandler.connect()
+        } else {
+            connectionHandler.handleReconnect()
+        }
     }
 
     public func close() async throws {