fix race condition when reading large amounts of data in iroh (#488)

Author: JannikSt

crates/shared/src/p2p/client.rs

@@ -215,13 +215,14 @@ impl P2PClient {
         // Now send the actual request
         let request = P2PRequest::new(message);
         Self::write_message(&mut send, &request).await?;
-        send.finish()?;
 
         // Read response
         let response: P2PResponse = Self::read_message(&mut recv).await?;
 
         tokio::time::sleep(Duration::from_millis(50)).await;
 
+        send.finish()?;
+
         Ok(response.message)
     }
 

crates/worker/src/docker/docker_manager.rs

@@ -742,7 +742,6 @@ impl DockerManager {
         container_id: &str,
         tail: Option<i64>,
     ) -> Result<String, DockerError> {
-        debug!("Fetching logs for container: {}", container_id);
         let tail_value = tail.unwrap_or(Self::DEFAULT_LOG_TAIL).to_string();
         let options = LogsOptions::<String> {
             stdout: true,

crates/worker/src/docker/service.rs

@@ -320,6 +320,7 @@ impl DockerService {
             Some(task) => {
                 let config_hash = task.generate_config_hash();
                 let container_id = format!("{}-{}-{:x}", TASK_PREFIX, task.id, config_hash);
+
                 let logs = self
                     .docker_manager
                     .get_container_logs(&container_id, None)

crates/worker/src/p2p/service.rs

@@ -509,9 +509,6 @@ impl P2PService {
             Self::write_response(&mut send, response).await?;
         }
 
-        // Finish the send stream
-        send.finish()?;
-
         Ok(())
     }
 
@@ -704,7 +701,6 @@ mod tests {
     #[serial]
     async fn test_ping() {
         let (service, client, _, worker_wallet_address) = setup_test_service(true).await;
-        println!("worker_wallet_address: {:?}", worker_wallet_address);
         let node_id = service.node_id().to_string();
         let addresses = service.listening_addresses().to_vec();
         let random_nonce = rand_v8::thread_rng().gen::<u64>();