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, aquatic_ws_load_test: cargo fmt

Browse source
This commit is contained in:
Joakim Frostegård 2021-11-02 20:33:14 +01:00
parent d922e5e680
commit 78d29770f3
3 changed files with 16 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

1
aquatic_ws_load_test/src/main.rs
View file

@ -97,7 +97,6 @@ fn create_tls_config() -> anyhow::Result<Arc<rustls::ClientConfig>> {
Ok(Arc::new(config))
}
fn monitor_statistics(state: LoadTestState, config: &Config) {
let start_time = Instant::now();
let mut report_avg_response_vec: Vec<f64> = Vec::new();

20
aquatic_ws_load_test/src/network.rs
View file

@ -7,12 +7,12 @@ use std::{
};
use aquatic_ws_protocol::{InMessage, JsonValue, OfferId, OutMessage, PeerId};
use async_tungstenite::{WebSocketStream, client_async};
use futures::{StreamExt, SinkExt};
use futures_rustls::{TlsConnector, client::TlsStream};
use async_tungstenite::{client_async, WebSocketStream};
use futures::{SinkExt, StreamExt};
use futures_rustls::{client::TlsStream, TlsConnector};
use glommio::net::TcpStream;
use glommio::{prelude::*, timer::TimerActionRepeat};
use rand::{Rng, SeedableRng, prelude::SmallRng};
use rand::{prelude::SmallRng, Rng, SeedableRng};
use crate::{common::LoadTestState, config::Config, utils::create_random_request};
@ -80,7 +80,9 @@ impl Connection {
let stream = TcpStream::connect(config.server_address)
.await
.map_err(|err| anyhow::anyhow!("connect: {:?}", err))?;
let stream = TlsConnector::from(tls_config).connect("example.com".try_into().unwrap(), stream).await?;
let stream = TlsConnector::from(tls_config)
.connect("example.com".try_into().unwrap(), stream)
.await?;
let request = format!(
"ws://{}:{}",
config.server_address.ip(),
@ -114,8 +116,12 @@ impl Connection {
async fn run_connection_loop(&mut self) -> anyhow::Result<()> {
loop {
if self.can_send {
let request =
create_random_request(&self.config, &self.load_test_state, &mut self.rng, self.peer_id);
let request = create_random_request(
&self.config,
&self.load_test_state,
&mut self.rng,
self.peer_id,
);
// If self.send_answer is set and request is announce request, make
// the request an offer answer