Search code examples
rustrust-tokio

How do I use TcpStream split across 2 threads with async?


I am trying to use the read and write of a tcp stream in different threads. This is what I currently have:

use tokio::prelude::*;
use tokio::net::TcpStream;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut stream = TcpStream::connect("localhost:8080").await?;
    let (mut read, mut write) = stream.split();

    tokio::spawn(async move {
        loop {
            let mut buf = [0u8; 32];
            read.read(&mut buf).await.unwrap();
            println!("{}", std::str::from_utf8(&buf));
        }
    });

    Ok(())
}

Im going to use another thread for the write. My problem is that I get the error that 'stream' is dropped while still borrowed.


Solution

  • That happens due to the method signature of Tokio::split, as you can see it takes &mut self, so its part cannot be used in a tokio::spawn future argument due to the 'static bound. So, this is exactly what the error says.

    What you are searching is tokio::io::split. Playground

    use tokio::prelude::*;
    use tokio::net::TcpStream;
    
    #[tokio::main]
    async fn main() -> Result<(), Box<dyn std::error::Error>> {
        let mut stream = TcpStream::connect("localhost:8080").await?;
        let (mut read, mut write) = tokio::io::split(stream);
    
        tokio::spawn(async move {
            loop {
                let mut buf = [0u8; 32];
                read.read(&mut buf).await.unwrap();
                println!("{:?}", std::str::from_utf8(&buf));
            }
        });
    
        Ok(())
    }