mercuryworkshop-mirror/epoxy-tls
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/epoxy-tls.git synced 2025-05-13 06:20:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

add block-udp and block-non-http, fix simple-wisp-client

Browse source
This commit is contained in:
Toshit Chawda 2024-03-28 19:22:36 -07:00
parent 77e377c814
commit 1ce39f6492
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
2 changed files with 98 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

28
simple-wisp-client/src/main.rs
View file

@ -126,6 +126,8 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
.header("Sec-WebSocket-Protocol", "wisp-v1")
.body(Empty::<Bytes>::new())?;
println!("{:?}", req);
let (ws, _) = handshake::client(&SpawnExecutor, req, socket).await?;
let (rx, tx) = ws.split(tokio::io::split);
@ -136,7 +138,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
threads.push(tokio::spawn(fut));
let payload = Bytes::from_static(&[0; 1024]);
let payload = Bytes::from(vec![0; 1024 * opts.packet_size]);
let cnt = Arc::new(RelaxedCounter::new(0));
@ -173,10 +175,14 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
interval.tick().await;
let now = cnt_avg.get();
let stat = format!(
"sent &[0; 1024] cnt: {:?}, +{:?}, moving average (100): {:?}",
"sent &[0; 1024 * {}] cnt: {:?} ({} KiB), +{:?} ({} KiB / 100ms), moving average (10 s): {:?} ({} KiB / 10 s)",
opts.packet_size,
now,
now * opts.packet_size,
now - last_time,
avg.get_average()
(now - last_time) * opts.packet_size,
avg.get_average(),
avg.get_average() * opts.packet_size,
);
if is_term {
print!("\x1b[2K{}\r", stat);
@ -208,13 +214,23 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
}));
}
let _ = select_all(threads.into_iter()).await;
let out = select_all(threads.into_iter()).await;
if let Err(err) = out.0? {
println!("\n\nerr: {:?}", err);
}
out.2.into_iter().for_each(|x| x.abort());
let duration_since = Instant::now().duration_since(start_time);
println!(
"\n\nresults: {} packets of &[0; 1024 * {}] sent in {}",
"\n\nresults: {} packets of &[0; 1024 * {}] ({} KiB) sent in {} ({} KiB/s)",
cnt.get(),
opts.packet_size,
format_duration(Instant::now().duration_since(start_time))
cnt.get() * opts.packet_size,
format_duration(duration_since),
(cnt.get() * opts.packet_size) as u64 / duration_since.as_secs(),
);
Ok(())