YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic ws load test: improve answer sending

Browse source
This commit is contained in:
Joakim Frostegård 2020-08-01 22:29:58 +02:00
parent 3e4e145248
commit f7eae860d3
1 changed files with 7 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aquatic_ws_load_test/src/network.rs
View file

@ -129,13 +129,9 @@ impl Connection {
if text.contains("answer"){
state.statistics.responses_answer
.fetch_add(1, Ordering::SeqCst);
send_random_request = true;
} else if text.contains("offer"){
// If message is an offer, don't send random
// request in return, since that would cause
// exponential growth of number of requests.
// However, add an answer to next request.
// If message is an offer, send an answer but
// no new offers in next announce request.
let res_offer: Result<MiddlemanOfferToPeer, _> = ::serde_json::from_str(&text);
@ -156,14 +152,12 @@ impl Connection {
} else if text.contains("interval"){
state.statistics.responses_announce
.fetch_add(1, Ordering::SeqCst);
send_random_request = true;
} else if text.contains("scrape"){
state.statistics.responses_scrape
.fetch_add(1, Ordering::SeqCst);
send_random_request = true;
}
send_random_request = true;
}
},
Err(tungstenite::Error::Io(err)) if err.kind() == ErrorKind::WouldBlock => {
@ -201,12 +195,14 @@ impl Connection {
rng
);
// Add offer answer data if applicable
// If self.send_answer is set and request is announce request, make
// the request an offer answer
let request = if let InMessage::AnnounceRequest(mut r) = request {
if let Some((peer_id, offer_id)) = self.send_answer {
r.to_peer_id = Some(peer_id);
r.offer_id = Some(offer_id);
r.answer = Some(JsonValue(::serde_json::Value::from("{}")));
r.offers = None;
}
self.send_answer = None;