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

ws load test: send offer answers

Browse source
This commit is contained in:
Joakim Frostegård 2021-11-02 11:51:51 +01:00
parent 839f516dcb
commit 6d4e090061
1 changed files with 23 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

26
aquatic_ws_load_test/src/network.rs
View file

@ -6,7 +6,7 @@ use std::{
time::Duration, time::Duration,
}; };
use aquatic_ws_protocol::{OfferId, OutMessage, PeerId}; use aquatic_ws_protocol::{InMessage, JsonValue, OfferId, OutMessage, PeerId};
use async_tungstenite::{WebSocketStream, client_async}; use async_tungstenite::{WebSocketStream, client_async};
use futures::{StreamExt, SinkExt}; use futures::{StreamExt, SinkExt};
use futures_rustls::{TlsConnector, client::TlsStream}; use futures_rustls::{TlsConnector, client::TlsStream};
@ -115,9 +115,29 @@ impl Connection {
loop { loop {
if self.can_send { if self.can_send {
let request = let request =
create_random_request(&self.config, &self.load_test_state, &mut self.rng, self.peer_id).to_ws_message(); create_random_request(&self.config, &self.load_test_state, &mut self.rng, self.peer_id);
self.stream.send(request).await?; // 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::json!(
{"sdp": "abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-abcdefg-"}
)));
r.event = None;
r.offers = None;
}
self.send_answer = None;
InMessage::AnnounceRequest(r)
} else {
request
};
self.stream.send(request.to_ws_message()).await?;
self.stream.flush().await?; self.stream.flush().await?;
self.load_test_state self.load_test_state