leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add a semaphore around peers

Browse source
This commit is contained in:
Igor Katson 2021-07-04 11:36:16 +01:00
parent 9472d66bf9
commit 64b1e47c77
4 changed files with 96 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

11
crates/librqbit/src/http_api.rs
View file

@ -2,7 +2,6 @@ use std::sync::Arc;
use librqbit_core::speed_estimator::SpeedEstimator;
use std::io::Write;
use std::sync::atomic::Ordering;
use std::time::Instant;
use warp::Filter;
@ -21,22 +20,22 @@ pub async fn make_and_run_http_api(
let dump_stats = warp::path("stats").map({
let state = state.clone();
let start_time = Instant::now();
let initial_downloaded_and_checked =
state.stats().downloaded_and_checked.load(Ordering::Relaxed);
let initial_downloaded_and_checked = state.stats_snapshot().downloaded_and_checked_bytes;
move || {
let snapshot = state.stats_snapshot();
let mut buf = Vec::new();
writeln!(buf, "{:#?}", state.stats_snapshot()).unwrap();
writeln!(
buf,
"Average download time: {:?}",
state.stats().average_piece_download_time()
snapshot.average_piece_download_time()
)
.unwrap();
// Poor mans download speed computation
let elapsed = start_time.elapsed();
let downloaded_bytes = state.stats().downloaded_and_checked.load(Ordering::Relaxed)
- initial_downloaded_and_checked;
let downloaded_bytes =
snapshot.downloaded_and_checked_bytes - initial_downloaded_and_checked;
let downloaded_mb = downloaded_bytes as f64 / 1024f64 / 1024f64;
writeln!(
buf,