Skip to content

Commit ee164e4

Browse files
committed
tests/util/chaosproxy: Change proxy_data() to accept Sender instead of &self
1 parent 7ab93b6 commit ee164e4

File tree

1 file changed

+24
-24
lines changed

1 file changed

+24
-24
lines changed

src/tests/util/chaosproxy.rs

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -110,45 +110,45 @@ impl ChaosProxy {
110110
.await?
111111
.into_split();
112112

113-
let self_clone = self.clone();
113+
let break_networking_send = self.break_networking_send.clone();
114114
tokio::spawn(async move {
115-
if let Err(err) = self_clone.proxy_data(client_read, backend_write).await {
115+
if let Err(err) = proxy_data(break_networking_send, client_read, backend_write).await {
116116
eprintln!("ChaosProxy connection error: {err}");
117117
}
118118
});
119119

120-
let self_clone = self.clone();
120+
let break_networking_send = self.break_networking_send.clone();
121121
tokio::spawn(async move {
122-
if let Err(err) = self_clone.proxy_data(backend_read, client_write).await {
122+
if let Err(err) = proxy_data(break_networking_send, backend_read, client_write).await {
123123
eprintln!("ChaosProxy connection error: {err}");
124124
}
125125
});
126126

127127
Ok(())
128128
}
129+
}
129130

130-
async fn proxy_data(
131-
&self,
132-
mut from: OwnedReadHalf,
133-
mut to: OwnedWriteHalf,
134-
) -> Result<(), Error> {
135-
let mut break_connections_recv = self.break_networking_send.subscribe();
136-
let mut buf = [0; 1024];
137-
138-
loop {
139-
tokio::select! {
140-
len = from.read(&mut buf) => {
141-
let len = len?;
142-
if len == 0 {
143-
// EOF, the socket was closed
144-
return Ok(());
145-
}
146-
to.write_all(&buf[0..len]).await?;
147-
}
148-
_ = break_connections_recv.recv() => {
149-
to.shutdown().await?;
131+
async fn proxy_data(
132+
break_networking_send: Sender<()>,
133+
mut from: OwnedReadHalf,
134+
mut to: OwnedWriteHalf,
135+
) -> Result<(), Error> {
136+
let mut break_connections_recv = break_networking_send.subscribe();
137+
let mut buf = [0; 1024];
138+
139+
loop {
140+
tokio::select! {
141+
len = from.read(&mut buf) => {
142+
let len = len?;
143+
if len == 0 {
144+
// EOF, the socket was closed
150145
return Ok(());
151146
}
147+
to.write_all(&buf[0..len]).await?;
148+
}
149+
_ = break_connections_recv.recv() => {
150+
to.shutdown().await?;
151+
return Ok(());
152152
}
153153
}
154154
}

0 commit comments

Comments
 (0)