Rocket/lib/benches/ranked-routing.rs

74 lines
2.0 KiB
Rust
Raw Normal View History

#![feature(test, plugin)]
#![plugin(rocket_codegen)]
extern crate rocket;
use rocket::config::{Environment, Config};
use rocket::http::RawStr;
#[get("/", format = "application/json")]
fn get() -> &'static str { "json" }
#[get("/", format = "text/html")]
fn get2() -> &'static str { "html" }
#[get("/", format = "text/plain")]
fn get3() -> &'static str { "plain" }
#[post("/", format = "application/json")]
fn post() -> &'static str { "json" }
#[post("/", format = "text/html")]
fn post2() -> &'static str { "html" }
#[post("/", format = "text/plain")]
fn post3() -> &'static str { "plain" }
fn rocket() -> rocket::Rocket {
let config = Config::new(Environment::Production).unwrap();
rocket::custom(config, false)
.mount("/", routes![get, get2, get3])
.mount("/", routes![post, post2, post3])
}
#[cfg(feature = "testing")]
mod benches {
extern crate test;
use super::rocket;
use self::test::Bencher;
use rocket::testing::MockRequest;
use rocket::http::Method::*;
use rocket::http::{Accept, ContentType};
#[bench]
fn accept_format(b: &mut Bencher) {
let rocket = rocket();
let mut requests = vec![];
requests.push(MockRequest::new(Get, "/").header(Accept::JSON));
requests.push(MockRequest::new(Get, "/").header(Accept::HTML));
requests.push(MockRequest::new(Get, "/").header(Accept::Plain));
b.iter(|| {
for request in requests.iter_mut() {
request.dispatch_with(&rocket);
}
});
}
#[bench]
fn content_type_format(b: &mut Bencher) {
let rocket = rocket();
let mut requests = vec![];
requests.push(MockRequest::new(Post, "/").header(ContentType::JSON));
requests.push(MockRequest::new(Post, "/").header(ContentType::HTML));
requests.push(MockRequest::new(Post, "/").header(ContentType::Plain));
b.iter(|| {
for request in requests.iter_mut() {
request.dispatch_with(&rocket);
}
});
}
}