From 83f2ad633bd9727d66dcbed6297db2f592e16260 Mon Sep 17 00:00:00 2001 From: John Doty Date: Sat, 17 Dec 2022 08:26:05 -0800 Subject: [PATCH] Correct my understanding of mutability a little --- src/client/mod.rs | 23 +++++++---------------- 1 file changed, 7 insertions(+), 16 deletions(-) diff --git a/src/client/mod.rs b/src/client/mod.rs index 7b64b1c..0cba8f3 100644 --- a/src/client/mod.rs +++ b/src/client/mod.rs @@ -185,7 +185,7 @@ async fn client_listen(port: u16, socks_port: u16) -> Result<()> { } async fn client_handle_messages( - reader: &mut MessageReader, + mut reader: MessageReader, events: mpsc::Sender, ) -> Result<()> { loop { @@ -226,8 +226,8 @@ async fn client_pipe_stderr( async fn client_main( socks_port: u16, - reader: &mut MessageReader, - writer: &mut MessageWriter, + mut reader: MessageReader, + mut writer: MessageWriter, events: mpsc::Sender, ) -> Result<()> { // Wait for the server's announcement. @@ -354,8 +354,8 @@ async fn client_connect_loop(remote: &str, events: mpsc::Sender) { } let mut stderr = BufReader::new(stderr); - let mut writer = MessageWriter::new(BufWriter::new(writer)); - let mut reader = MessageReader::new(reader); + let writer = MessageWriter::new(BufWriter::new(writer)); + let reader = MessageReader::new(reader); let sec = events.clone(); tokio::spawn(async move { @@ -363,8 +363,7 @@ async fn client_connect_loop(remote: &str, events: mpsc::Sender) { }); if let Err(e) = - client_main(socks_port, &mut reader, &mut writer, events.clone()) - .await + client_main(socks_port, reader, writer, events.clone()).await { error!("Server disconnected with error: {:?}", e); } else { @@ -422,15 +421,7 @@ mod tests { let (event_sender, event_receiver) = mpsc::channel(1024); let client_result = tokio::spawn(async move { - let mut client_read = client_read; - let mut client_write = client_write; - client_main( - 0, - &mut client_read, - &mut client_write, - event_sender, - ) - .await + client_main(0, client_read, client_write, event_sender).await }); Fixture {