Waiting error handled properly now

space307 · Oct 12, 2023 · 9c9e3b0 · 9c9e3b0
1 parent d3a0b10
commit 9c9e3b0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/Sources/Engine/WSEngine.swift b/Sources/Engine/WSEngine.swift
@@ -143,9 +143,7 @@ FrameCollectorDelegate, HTTPHandlerDelegate {
             let wsReq = HTTPWSHeader.createUpgrade(request: request, supportsCompression: framer.supportsCompression(), secKeyValue: secKeyValue)
             let data = httpHandler.convert(request: wsReq)
             transport.write(data: data, completion: {_ in })
-        case .waiting:
-            break
-        case .failed(let error):
+        case .waiting(let error), .failed(let error):
             handleError(error)
         case .viability(let isViable):
             broadcast(event: .viabilityChanged(isViable))

diff --git a/Sources/Transport/TCPTransport.swift b/Sources/Transport/TCPTransport.swift
@@ -110,8 +110,8 @@ public class TCPTransport: Transport {
             switch newState {
             case .ready:
                 self?.delegate?.connectionChanged(state: .connected)
-            case .waiting:
-                self?.delegate?.connectionChanged(state: .waiting)
+            case .waiting(let error):
+                self?.delegate?.connectionChanged(state: .waiting(error))
             case .cancelled:
                 self?.delegate?.connectionChanged(state: .cancelled)
             case .failed(let error):

diff --git a/Sources/Transport/Transport.swift b/Sources/Transport/Transport.swift
@@ -27,7 +27,7 @@ public enum ConnectionState {
     case connected
 
     /// Waiting connections have not yet been started, or do not have a viable network
-    case waiting
+    case waiting(Error?)
 
     /// Cancelled connections have been invalidated by the client and will send no more events
     case cancelled