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

aquatic_http_load_test: remove mio implementation, clean up

Browse source
This commit is contained in:
Joakim Frostegård 2021-10-28 01:13:18 +02:00
parent 13d18bbf03
commit e458cc54db
6 changed files with 242 additions and 586 deletions
Download patch file Download diff file Expand all files Collapse all files

59
aquatic_http_load_test/src/main.rs
View file

@ -1,5 +1,3 @@
use std::fs::File;
use std::io::BufReader;
use std::sync::{atomic::Ordering, Arc};
use std::thread;
use std::time::{Duration, Instant};
@ -10,7 +8,6 @@ use rand_distr::Pareto;
mod common;
mod config;
mod glommio;
mod network;
mod utils;
@ -65,7 +62,7 @@ fn run(config: Config) -> ::anyhow::Result<()> {
let state = state.clone();
LocalExecutorBuilder::default().spawn(|| async move {
glommio::run_socket_thread(config, tls_config, state).await.unwrap();
run_socket_thread(config, tls_config, state).await.unwrap();
}).unwrap();
}
@ -74,33 +71,6 @@ fn run(config: Config) -> ::anyhow::Result<()> {
Ok(())
}
struct FakeCertificateVerifier;
impl rustls::client::ServerCertVerifier for FakeCertificateVerifier {
fn verify_server_cert(
&self,
_end_entity: &rustls::Certificate,
_intermediates: &[rustls::Certificate],
_server_name: &rustls::ServerName,
_scts: &mut dyn Iterator<Item = &[u8]>,
_ocsp_response: &[u8],
_now: std::time::SystemTime,
) -> Result<rustls::client::ServerCertVerified, rustls::Error> {
Ok(rustls::client::ServerCertVerified::assertion())
}
}
fn create_tls_config() -> anyhow::Result<Arc<rustls::ClientConfig>> {
let mut config = rustls::ClientConfig::builder()
.with_safe_defaults()
.with_root_certificates(rustls::RootCertStore::empty())
.with_no_client_auth();
config.dangerous().set_certificate_verifier(Arc::new(FakeCertificateVerifier));
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();
@ -183,3 +153,30 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
}
}
}
struct FakeCertificateVerifier;
impl rustls::client::ServerCertVerifier for FakeCertificateVerifier {
fn verify_server_cert(
&self,
_end_entity: &rustls::Certificate,
_intermediates: &[rustls::Certificate],
_server_name: &rustls::ServerName,
_scts: &mut dyn Iterator<Item = &[u8]>,
_ocsp_response: &[u8],
_now: std::time::SystemTime,
) -> Result<rustls::client::ServerCertVerified, rustls::Error> {
Ok(rustls::client::ServerCertVerified::assertion())
}
}
fn create_tls_config() -> anyhow::Result<Arc<rustls::ClientConfig>> {
let mut config = rustls::ClientConfig::builder()
.with_safe_defaults()
.with_root_certificates(rustls::RootCertStore::empty())
.with_no_client_auth();
config.dangerous().set_certificate_verifier(Arc::new(FakeCertificateVerifier));
Ok(Arc::new(config))
}