itsscb/tracing
1
0
Fork 0
mirror of https://github.com/tokio-rs/tracing.git synced 2025-10-04 16:24:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

tower: make the load generator example more fun (#276)

Browse Source 
This commit tweaks the `tracing-tower` load-generator example a little
to better simulate a real application under load. In particular, there
are now three load generators rather than a single one, and there is a
small amount of random jitter to when each load generator sends a
request. This has the effect of making events more likely to be
interleaved, so filtering becomes much more necessary.

Signed-off-by: Eliza Weisman <eliza@buoyant.io>
This commit is contained in:
Eliza Weisman 2019-08-13 18:33:28 -07:00 committed by GitHub
parent 99e061f795
commit ad558aad6c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tracing-tower/examples/load.rs
View File

@ -51,6 +51,8 @@ fn main() {
let server = serve(MakeSvc, &addr, "serve") let server = serve(MakeSvc, &addr, "serve")
.join(admin) .join(admin)
.join(load_gen(&addr)) .join(load_gen(&addr))
.join(load_gen(&addr))
.join(load_gen(&addr))
.map(|_| ()) .map(|_| ())
.map_err(|_| ()); .map_err(|_| ());
@ -81,12 +83,10 @@ where
let span = tracing::debug_span!( let span = tracing::debug_span!(
"request", "request",
req.method = ?req.method(), req.method = ?req.method(),
req.uri = ?req.uri(),
req.version = ?req.version(),
req.path = ?req.uri().path(), req.path = ?req.uri().path(),
); );
span.in_scope(|| { span.in_scope(|| {
tracing::debug!(message = "received request.", req.headers = ?req.headers()); tracing::debug!(message = "received request.", req.headers = ?req.headers(), req.version = ?req.version());
}); });
span span
}; };
@ -340,7 +340,8 @@ impl std::error::Error for WrongMethod {}
fn load_gen(addr: &SocketAddr) -> Box<dyn Future<Item = (), Error = ()> + Send + 'static> { fn load_gen(addr: &SocketAddr) -> Box<dyn Future<Item = (), Error = ()> + Send + 'static> {
use rand::Rng; use rand::Rng;
use std::time::Duration; use std::time::{Duration, Instant};
use tokio::timer;
use tokio_buf::util::BufStreamExt; use tokio_buf::util::BufStreamExt;
use tower::ServiceExt; use tower::ServiceExt;
use tower_http_util::body::BodyExt; use tower_http_util::body::BodyExt;
@ -355,7 +356,11 @@ fn load_gen(addr: &SocketAddr) -> Box<dyn Future<Item = (), Error = ()> + Send +
.timeout(Duration::from_secs(5)) .timeout(Duration::from_secs(5))
.service(hyper); .service(hyper);
tokio::timer::Interval::new_interval(Duration::from_millis(50)) timer::Interval::new_interval(Duration::from_millis(50))
.and_then(|_| {
let sleep = rand::thread_rng().gen_range(0, 25);
timer::Delay::new(Instant::now() + Duration::from_millis(sleep)).map_err(Into::into)
})
.fold((svc, authority), |(svc, authority), _| { .fold((svc, authority), |(svc, authority), _| {
let mut rng = rand::thread_rng(); let mut rng = rand::thread_rng();
let idx = rng.gen_range(0, ALPHABET.len()+1); let idx = rng.gen_range(0, ALPHABET.len()+1);