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

aquatic_ws: add InMessage::from_ws_message, use in load test

Browse source
This commit is contained in:
Joakim Frostegård 2020-08-01 22:45:19 +02:00
parent f7eae860d3
commit 6f2dc6a510
4 changed files with 56 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

50
aquatic_ws_load_test/src/network.rs
View file

@ -125,39 +125,39 @@ impl Connection {
loop {
match ws.read_message(){
Ok(message) => {
if let ::tungstenite::Message::Text(text) = message {
if text.contains("answer"){
match OutMessage::from_ws_message(message){
Ok(OutMessage::Offer(offer)) => {
state.statistics.responses_offer
.fetch_add(1, Ordering::SeqCst);
self.send_answer = Some((
offer.peer_id,
offer.offer_id
));
send_random_request = true;
},
Ok(OutMessage::Answer(_)) => {
state.statistics.responses_answer
.fetch_add(1, Ordering::SeqCst);
} else if text.contains("offer"){
// 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);
match res_offer {
Ok(offer) => {
state.statistics.responses_offer
.fetch_add(1, Ordering::SeqCst);
self.send_answer = Some((
offer.peer_id,
offer.offer_id
));
},
Err(err) => {
eprintln!("error decoding offer: {:?}", err);
}
}
} else if text.contains("interval"){
send_random_request = true;
},
Ok(OutMessage::AnnounceResponse(_)) => {
state.statistics.responses_announce
.fetch_add(1, Ordering::SeqCst);
} else if text.contains("scrape"){
send_random_request = true;
},
Ok(OutMessage::ScrapeResponse(_)) => {
state.statistics.responses_scrape
.fetch_add(1, Ordering::SeqCst);
}
send_random_request = true;
send_random_request = true;
},
Err(err) => {
eprintln!("error deserializing offer: {:?}", err);
}
}
},
Err(tungstenite::Error::Io(err)) if err.kind() == ErrorKind::WouldBlock => {