1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
#![feature(test)]
extern crate fdlimit;
extern crate test;
extern crate tiny_http;
use std::io::Write;
use std::process::Command;
use tiny_http::Method;
#[test]
#[ignore]
// TODO: obtain time
fn curl_bench() {
let server = tiny_http::Server::http("0.0.0.0:0").unwrap();
let port = server.server_addr().to_ip().unwrap().port();
let num_requests = 10usize;
match Command::new("curl")
.arg("-s")
.arg(format!("http://localhost:{}/?[1-{}]", port, num_requests))
.output()
{
Ok(p) => p,
Err(_) => return, // ignoring test
};
drop(server);
}
#[bench]
fn sequential_requests(bencher: &mut test::Bencher) {
let server = tiny_http::Server::http("0.0.0.0:0").unwrap();
let port = server.server_addr().to_ip().unwrap().port();
let mut stream = std::net::TcpStream::connect(("127.0.0.1", port)).unwrap();
bencher.iter(|| {
(write!(stream, "GET / HTTP/1.1\r\nHost: localhost\r\n\r\n")).unwrap();
let request = server.recv().unwrap();
assert_eq!(request.method(), &Method::Get);
request.respond(tiny_http::Response::new_empty(tiny_http::StatusCode(204)));
});
}
#[bench]
fn parallel_requests(bencher: &mut test::Bencher) {
fdlimit::raise_fd_limit();
let server = tiny_http::Server::http("0.0.0.0:0").unwrap();
let port = server.server_addr().to_ip().unwrap().port();
bencher.iter(|| {
let mut streams = Vec::new();
for _ in 0..1000usize {
let mut stream = std::net::TcpStream::connect(("127.0.0.1", port)).unwrap();
(write!(
stream,
"GET / HTTP/1.1\r\nHost: localhost\r\nConnection: close\r\n\r\n"
))
.unwrap();
streams.push(stream);
}
loop {
let request = match server.try_recv().unwrap() {
None => break,
Some(rq) => rq,
};
assert_eq!(request.method(), &Method::Get);
request.respond(tiny_http::Response::new_empty(tiny_http::StatusCode(204)));
}
});
}
|
