aquatic_ws: add ErrorResponse, send it when info hash is not allowed

This commit is contained in:
Joakim Frostegård 2021-10-16 01:18:23 +02:00
parent 28cc6c261a
commit 1d5eb0dff9
5 changed files with 43 additions and 4 deletions

View file

@ -12,6 +12,7 @@ pub struct Statistics {
pub responses_offer: AtomicUsize,
pub responses_answer: AtomicUsize,
pub responses_scrape: AtomicUsize,
pub responses_error: AtomicUsize,
}
#[derive(Clone)]

View file

@ -85,13 +85,16 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
statistics.responses_answer.fetch_and(0, Ordering::SeqCst) as f64 / interval_f64;
let responses_scrape_per_second =
statistics.responses_scrape.fetch_and(0, Ordering::SeqCst) as f64 / interval_f64;
let responses_error_per_second =
statistics.responses_error.fetch_and(0, Ordering::SeqCst) as f64 / interval_f64;
let responses_announce_per_second = responses_announce / interval_f64;
let responses_per_second = responses_announce_per_second
+ responses_offer_per_second
+ responses_answer_per_second
+ responses_scrape_per_second;
+ responses_scrape_per_second
+ responses_error_per_second;
report_avg_response_vec.push(responses_per_second);
@ -105,6 +108,7 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
println!(" - Offer responses: {:.2}", responses_offer_per_second);
println!(" - Answer responses: {:.2}", responses_answer_per_second);
println!(" - Scrape responses: {:.2}", responses_scrape_per_second);
println!(" - Error responses: {:.2}", responses_error_per_second);
let time_elapsed = start_time.elapsed();
let duration = Duration::from_secs(config.duration as u64);

View file

@ -151,6 +151,16 @@ impl Connection {
self.can_send = true;
}
Ok(OutMessage::ErrorResponse(response)) => {
state
.statistics
.responses_error
.fetch_add(1, Ordering::SeqCst);
eprintln!("received error response: {:?}", response.failure_reason);
self.can_send = true;
},
Err(err) => {
eprintln!("error deserializing offer: {:?}", err);
}