v1.17: quic: delay calling set_max_concurrent_uni_streams/set_receive…

…_window (backport of solana-labs#904) (solana-labs#967) * quic: delay calling set_max_concurrent_uni_streams/set_receive_window (solana-labs#904) * quic: don't call connection.set_max_concurrent_uni_streams if we're going to drop a connection Avoids taking a mutex and waking a task. * quic: don't increase the receive window before we've actually accepted a connection (cherry picked from commit 2770424) # Conflicts: # streamer/src/nonblocking/quic.rs * Fix merge conflicts --------- Co-authored-by: Alessandro Decina <[email protected]>
andreisilviudragnea · Apr 30, 2024 · 72d6e89 · 72d6e89
1 parent 4f9c393
commit 72d6e89
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/streamer/src/nonblocking/quic.rs b/streamer/src/nonblocking/quic.rs
@@ -342,18 +342,12 @@ fn handle_and_cache_new_connection(
         params.total_stake,
     ) as u64)
     {
-        connection.set_max_concurrent_uni_streams(max_uni_streams);
         let receive_window = compute_recieve_window(
             params.max_stake,
             params.min_stake,
             connection_table_l.peer_type,
             params.stake,
         );
-
-        if let Ok(receive_window) = receive_window {
-            connection.set_receive_window(receive_window);
-        }
-
         let remote_addr = connection.remote_address();
 
         debug!(
@@ -376,6 +370,12 @@ fn handle_and_cache_new_connection(
         ) {
             let peer_type = connection_table_l.peer_type;
             drop(connection_table_l);
+
+            if let Ok(receive_window) = receive_window {
+                connection.set_receive_window(receive_window);
+            }
+            connection.set_max_concurrent_uni_streams(max_uni_streams);
+
             tokio::spawn(handle_connection(
                 connection,
                 remote_addr,