mirror of https://github.com/rwf2/Rocket.git
74 lines
2.0 KiB
Rust
74 lines
2.0 KiB
Rust
|
#![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);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|