eb38a11b9a
In 53bba6e , ssl usage was improved. The existing benchmark
implementation wasn't adjusted though.
This commit fixes the client benchmark to work with the latest nightly
rust and instructs travis to try compile the benchmarks
when rustc nightly is used. This should prevent such kind of breakage
in future.
103 lines
2.2 KiB
Rust
103 lines
2.2 KiB
Rust
#![deny(warnings)]
|
|
#![feature(vec_push_all, test)]
|
|
extern crate hyper;
|
|
|
|
extern crate test;
|
|
|
|
use std::fmt;
|
|
use std::io::{self, Read, Write, Cursor};
|
|
use std::net::SocketAddr;
|
|
|
|
use hyper::net;
|
|
|
|
static README: &'static [u8] = include_bytes!("../README.md");
|
|
|
|
struct MockStream {
|
|
read: Cursor<Vec<u8>>
|
|
}
|
|
|
|
impl MockStream {
|
|
fn new() -> MockStream {
|
|
let head = b"HTTP/1.1 200 OK\r\nServer: Mock\r\n\r\n";
|
|
let mut res = head.to_vec();
|
|
res.push_all(README);
|
|
MockStream {
|
|
read: Cursor::new(res)
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Clone for MockStream {
|
|
fn clone(&self) -> MockStream {
|
|
MockStream {
|
|
read: Cursor::new(self.read.get_ref().clone())
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Read for MockStream {
|
|
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
|
self.read.read(buf)
|
|
}
|
|
}
|
|
|
|
impl Write for MockStream {
|
|
fn write(&mut self, msg: &[u8]) -> io::Result<usize> {
|
|
// we're mocking, what do we care.
|
|
Ok(msg.len())
|
|
}
|
|
fn flush(&mut self) -> io::Result<()> {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
struct Foo;
|
|
|
|
impl hyper::header::Header for Foo {
|
|
fn header_name() -> &'static str {
|
|
"x-foo"
|
|
}
|
|
fn parse_header(_: &[Vec<u8>]) -> hyper::Result<Foo> {
|
|
Err(hyper::Error::Header)
|
|
}
|
|
}
|
|
|
|
impl hyper::header::HeaderFormat for Foo {
|
|
fn fmt_header(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
|
|
fmt.write_str("Bar")
|
|
}
|
|
}
|
|
|
|
impl net::NetworkStream for MockStream {
|
|
fn peer_addr(&mut self) -> io::Result<SocketAddr> {
|
|
Ok("127.0.0.1:1337".parse().unwrap())
|
|
}
|
|
}
|
|
|
|
struct MockConnector;
|
|
|
|
impl net::NetworkConnector for MockConnector {
|
|
type Stream = MockStream;
|
|
fn connect(&self, _: &str, _: u16, _: &str) -> hyper::Result<MockStream> {
|
|
Ok(MockStream::new())
|
|
}
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_mock_hyper(b: &mut test::Bencher) {
|
|
let url = "http://127.0.0.1:1337/";
|
|
b.iter(|| {
|
|
let mut req = hyper::client::Request::with_connector(
|
|
hyper::Get, hyper::Url::parse(url).unwrap(), &MockConnector
|
|
).unwrap();
|
|
req.headers_mut().set(Foo);
|
|
|
|
let mut s = String::new();
|
|
req
|
|
.start().unwrap()
|
|
.send().unwrap()
|
|
.read_to_string(&mut s).unwrap()
|
|
});
|
|
}
|