update p2p reconnect & test (#214)

* update reconnect test * update reconnect cases * update reconnect * update peer test * update peer test * update peer test * update peer * update connection * update reconnect * update p2p reconnect & test * update peer * update peer test * update test
open-web3-stack · Nov 4, 2024 · b31130a · b31130a
1 parent 875a7f7
commit b31130a
Show file tree

Hide file tree

Showing 3 changed files with 309 additions and 83 deletions.
diff --git a/Networking/Sources/Networking/Connection.swift b/Networking/Sources/Networking/Connection.swift
@@ -20,12 +20,14 @@ enum ConnectionError: Error {
     case invalidLength
     case unexpectedState
     case closed
+    case reconnect
 }
 
 enum ConnectionState {
     case connecting(continuations: [CheckedContinuation<Void, Error>])
     case connected(publicKey: Data)
     case closed
+    case reconnect(publicKey: Data)
 }
 
 public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoProtocol {
@@ -50,6 +52,8 @@ public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoP
                 publicKey
             case .closed:
                 nil
+            case let .reconnect(publicKey):
+                publicKey
             }
         }
     }
@@ -91,6 +95,18 @@ public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoP
         }
     }
 
+    func reconnect(publicKey: Data) {
+        state.write { state in
+            if case let .connecting(continuations) = state {
+                for continuation in continuations {
+                    continuation.resume(throwing: ConnectionError.reconnect)
+                }
+                state = .reconnect(publicKey: publicKey)
+            }
+            state = .reconnect(publicKey: publicKey)
+        }
+    }
+
     public var isClosed: Bool {
         state.read {
             switch $0 {
@@ -100,7 +116,18 @@ public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoP
                 false
             case .closed:
                 true
+            case .reconnect:
+                false
+            }
+        }
+    }
+
+    public var needReconnect: Bool {
+        state.read {
+            if case .reconnect = $0 {
+                return true
             }
+            return false
         }
     }
 
@@ -113,6 +140,8 @@ public final class Connection<Handler: StreamHandler>: Sendable, ConnectionInfoP
                 true
             case .closed:
                 true
+            case .reconnect:
+                true
             }
         }