bug fix add connection when created first time

blockworks-foundation · Sep 14, 2023 · f75c665 · f75c665
1 parent b8437c0
commit f75c665
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/core/src/quic_connection.rs b/core/src/quic_connection.rs
@@ -79,16 +79,7 @@ impl QuicConnection {
                     if connection.stable_id() != current_stable_id {
                         Some(connection)
                     } else {
-                        let new_conn = QuicConnectionUtils::connect(
-                            self.identity,
-                            true,
-                            self.endpoint.clone(),
-                            self.socket_address,
-                            self.connection_params.connection_timeout,
-                            self.connection_params.connection_retry_count,
-                            self.exit_signal.clone(),
-                        )
-                        .await;
+                        let new_conn = self.connect().await;
                         if let Some(new_conn) = new_conn {
                             *conn = Some(new_conn);
                             conn.clone()
@@ -101,7 +92,11 @@ impl QuicConnection {
                     Some(connection.clone())
                 }
             }
-            None => self.connect().await,
+            None => {
+                let connection = self.connect().await;
+                *self.connection.write().await = connection.clone();
+                connection
+            }
         }
     }