mirror of
https://github.com/YGGverse/aquatic.git
synced 2026-03-31 17:55:36 +00:00
aquatic: handlers: pass rng's as function arguments
This commit is contained in:
parent
bd16cdd709
commit
2fd732efc6
5 changed files with 25 additions and 11 deletions
|
|
@ -2,7 +2,7 @@ use std::io::Cursor;
|
|||
use std::time::Instant;
|
||||
use std::net::SocketAddr;
|
||||
|
||||
use rand::Rng;
|
||||
use rand::{Rng, SeedableRng, thread_rng, rngs::SmallRng};
|
||||
use rand_distr::Pareto;
|
||||
|
||||
use aquatic::handlers::*;
|
||||
|
|
@ -29,6 +29,8 @@ pub fn bench(
|
|||
let mut num_responses: usize = 0;
|
||||
let mut dummy = 0u8;
|
||||
|
||||
let mut small_rng = SmallRng::from_rng(thread_rng()).unwrap();
|
||||
|
||||
let now = Instant::now();
|
||||
|
||||
let mut requests: Vec<(AnnounceRequest, SocketAddr)> = requests.into_iter()
|
||||
|
|
@ -46,6 +48,7 @@ pub fn bench(
|
|||
handle_announce_requests(
|
||||
&state,
|
||||
config,
|
||||
&mut small_rng,
|
||||
&mut responses,
|
||||
requests,
|
||||
);
|
||||
|
|
|
|||
|
|
@ -2,7 +2,7 @@ use std::io::Cursor;
|
|||
use std::time::Instant;
|
||||
use std::net::SocketAddr;
|
||||
|
||||
use rand::{Rng, thread_rng, rngs::SmallRng, SeedableRng};
|
||||
use rand::{Rng, SeedableRng, thread_rng, rngs::{SmallRng, StdRng}};
|
||||
|
||||
use aquatic::common::*;
|
||||
use aquatic::handlers::handle_connect_requests;
|
||||
|
|
@ -25,6 +25,8 @@ pub fn bench(
|
|||
let mut num_responses: usize = 0;
|
||||
let mut dummy = 0u8;
|
||||
|
||||
let mut rng = StdRng::from_rng(thread_rng()).unwrap();
|
||||
|
||||
let now = Instant::now();
|
||||
|
||||
let mut requests: Vec<(ConnectRequest, SocketAddr)> = requests.into_iter()
|
||||
|
|
@ -39,7 +41,7 @@ pub fn bench(
|
|||
|
||||
let requests = requests.drain(..);
|
||||
|
||||
handle_connect_requests(&state, &mut responses, requests);
|
||||
handle_connect_requests(&state, &mut rng, &mut responses, requests);
|
||||
|
||||
for (response, _) in responses.drain(..){
|
||||
if let Response::Connect(_) = response {
|
||||
|
|
|
|||
|
|
@ -4,9 +4,9 @@
|
|||
//! ```
|
||||
//! ## Average results over 10 rounds
|
||||
//!
|
||||
//! Connect handler: 2 330 506 requests/second, 429.30 ns/request
|
||||
//! Announce handler: 264 596 requests/second, 3782.88 ns/request
|
||||
//! Scrape handler: 461 530 requests/second, 2169.41 ns/request
|
||||
//! Connect handler: 2 280 522 requests/second, 439.29 ns/request
|
||||
//! Announce handler: 267 790 requests/second, 3736.05 ns/request
|
||||
//! Scrape handler: 570 958 requests/second, 1751.89 ns/request
|
||||
//! ```
|
||||
|
||||
use std::time::{Duration, Instant};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue