imp(orchestrator): p2p relay setup (#474)

* setup p2p relay

Author: JannikSt

crates/orchestrator/src/p2p/client.rs

@@ -61,7 +61,7 @@ impl P2PClient {
                 worker_addresses,
                 worker_wallet_address,
                 P2PMessage::GetTaskLogs,
-                15, // 15 second timeout for getting logs
+                20,
             )
             .await?;
 

crates/shared/src/p2p/client.rs

@@ -1,7 +1,7 @@
 use alloy::primitives::Address;
 use anyhow::Result;
 use iroh::endpoint::{RecvStream, SendStream};
-use iroh::{Endpoint, NodeAddr, NodeId, SecretKey};
+use iroh::{Endpoint, NodeAddr, NodeId, RelayMode, SecretKey};
 use log::{debug, info};
 use std::str::FromStr;
 use std::time::Duration;
@@ -28,6 +28,7 @@ impl P2PClient {
         let endpoint = Endpoint::builder()
             .secret_key(secret_key)
             .alpns(vec![PRIME_P2P_PROTOCOL.to_vec()])
+            .relay_mode(RelayMode::Default)
             .discovery_n0()
             .bind()
             .await?;
@@ -219,8 +220,6 @@ impl P2PClient {
         // Read response
         let response: P2PResponse = Self::read_message(&mut recv).await?;
 
-        // Close connection
-        connection.close(0u32.into(), b"request complete");
         tokio::time::sleep(Duration::from_millis(50)).await;
 
         Ok(response.message)

crates/worker/src/p2p/service.rs

@@ -6,7 +6,7 @@ use alloy::primitives::{Address, FixedBytes, U256};
 use anyhow::Result;
 use dashmap::DashMap;
 use iroh::endpoint::Incoming;
-use iroh::{Endpoint, SecretKey};
+use iroh::{Endpoint, RelayMode, SecretKey};
 use lazy_static::lazy_static;
 use log::{debug, error, info, warn};
 use rand_v8::Rng;
@@ -77,6 +77,7 @@ impl P2PService {
             .secret_key(secret_key)
             .alpns(vec![PRIME_P2P_PROTOCOL.to_vec()])
             .discovery_n0()
+            .relay_mode(RelayMode::Default)
             .bind()
             .await?;
 
@@ -158,9 +159,6 @@ impl P2PService {
                         }
                         // Wait a bit before closing to ensure client has processed response
                         tokio::time::sleep(tokio::time::Duration::from_millis(100)).await;
-
-                        // Properly close the connection after handling the stream
-                        connection.close(0u32.into(), b"stream complete");
                     }
                     Err(e) => {
                         error!("Failed to accept bi-stream: {}", e);