fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-06-05 22:21:20 +00:00
Code Issues Projects Releases Wiki Activity

muxers/mplex/benches: Use combinators to reduce nesting (#2688)

Browse Source
This commit is contained in:
Thomas Eizinger 2022-06-03 17:38:02 +02:00 committed by GitHub
parent bd9834aab4
commit fcc987e0f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
muxers/mplex/benches/split_send_size.rs
View File

@ -104,16 +104,18 @@ fn run(transport: &mut BenchTransport, payload: &Vec<u8>, listen_addr: &Multiadd
} }
transport::ListenerEvent::Upgrade { upgrade, .. } => { transport::ListenerEvent::Upgrade { upgrade, .. } => {
let (_peer, conn) = upgrade.await.unwrap(); let (_peer, conn) = upgrade.await.unwrap();
match poll_fn(|cx| conn.poll_event(cx)).await { let mut s = poll_fn(|cx| conn.poll_event(cx))
Ok(muxing::StreamMuxerEvent::InboundSubstream(mut s)) => { .await
.expect("unexpected error")
.into_inbound_substream()
.expect("Unexpected muxer event");
let mut buf = vec![0u8; payload_len]; let mut buf = vec![0u8; payload_len];
let mut off = 0; let mut off = 0;
loop { loop {
// Read in typical chunk sizes of up to 8KiB. // Read in typical chunk sizes of up to 8KiB.
let end = off + std::cmp::min(buf.len() - off, 8 * 1024); let end = off + std::cmp::min(buf.len() - off, 8 * 1024);
let n = poll_fn(|cx| { let n = poll_fn(|cx| conn.read_substream(cx, &mut s, &mut buf[off..end]))
conn.read_substream(cx, &mut s, &mut buf[off..end])
})
.await .await
.unwrap(); .unwrap();
off += n; off += n;
@ -122,10 +124,6 @@ fn run(transport: &mut BenchTransport, payload: &Vec<u8>, listen_addr: &Multiadd
} }
} }
} }
Ok(_) => panic!("Unexpected muxer event"),
Err(e) => panic!("Unexpected error: {:?}", e),
}
}
_ => panic!("Unexpected listener event"), _ => panic!("Unexpected listener event"),
} }
} }