PeterCxy/wl-mitm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Prioritize writing over reading in tokio::select!

Browse source
This commit is contained in:
Peter Cai 2025-03-09 16:30:23 -04:00
parent cf723891cb
commit 4259730938
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/main.rs
View file

@ -214,15 +214,16 @@ impl<'a> ConnDuplex<'a> {
loop {
tokio::select! {
biased;
res = self.downstream_write.dequeue_write() => res?,
res = self.upstream_write.dequeue_write() => res?,
msg = self.upstream_read.read() => {
control_flow!(self.handle_s2c_event(msg?).await?);
}
res = self.downstream_write.dequeue_write() => res?,
msg = self.downstream_read.read() => {
control_flow!(self.handle_c2s_request(msg?).await?);
}
res = self.upstream_write.dequeue_write() => res?,
}
}