add send timeout to the relay actor

n0-computer · Jan 13, 2025 · 75c578a · 75c578a
1 parent 0d7a933
commit 75c578a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/iroh/src/magicsock/relay_actor.rs b/iroh/src/magicsock/relay_actor.rs
@@ -641,15 +641,19 @@ impl ActiveRelayActor {
         state: &mut ConnectedRelayState,
         client_stream: &mut iroh_relay::client::ClientStream,
     ) -> Result<()> {
-        // TODO
-        // let mut timeout = pin!(tokio::time::sleep())
+        const SEND_TIMEOUT: Duration = Duration::from_secs(10); // TODO: what should this be?
+
+        let mut timeout = pin!(tokio::time::sleep(SEND_TIMEOUT));
         let mut sending_fut = pin!(sending_fut);
         loop {
             tokio::select! {
                 biased;
                 _ = self.stop_token.cancelled() => {
                     break Ok(());
                 }
+                _ = &mut timeout => {
+                    break Err(anyhow!("Send timeout"));
+                }
                 msg = self.prio_inbox.recv() => {
                     let Some(msg) = msg else {
                         warn!("Priority inbox closed, shutdown.");