fix(iroh-gossip): do not drop existing peer connection when we get in…

…coming one
n0-computer · May 22, 2024 · e12b909 · e12b909
1 parent d813089
commit e12b909
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/iroh-gossip/src/net.rs b/iroh-gossip/src/net.rs
@@ -615,11 +615,13 @@ async fn connection_loop(
     loop {
         tokio::select! {
             biased;
-            msg = send_rx.recv() => {
-                match msg {
-                    None => break,
-                    Some(msg) =>  write_message(&mut send, &mut send_buf, &msg).await?,
-                }
+            // If `send_rx` is closed,
+            // stop selecting it but don't quit.
+            // We are not going to use connection for sending anymore,
+            // but the other side may still want to use it to
+            // send data to us.
+            Some(msg) = send_rx.recv(), if !send_rx.is_closed() => {
+                write_message(&mut send, &mut send_buf, &msg).await?
             }
 
             msg = read_message(&mut recv, &mut recv_buf) => {