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

Enable biased mode for tokio::select!

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

4
src/main.rs
View file

@ -213,14 +213,16 @@ impl<'a> ConnDuplex<'a> {
pub async fn run_to_completion(mut self) -> io::Result<()> {
loop {
tokio::select! {
biased;
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?,
res = self.downstream_write.dequeue_write() => res?,
}
}