From 900ce1c36fd2c6ccf027209f7580eaafb66d6a2a Mon Sep 17 00:00:00 2001
From: Sean McArthur <sean.monstar@gmail.com>
Date: Fri, 12 May 2017 16:04:47 -0700
Subject: [PATCH] test(server): add benchmark of hyper server vs raw tcp

---
 benches/server.rs | 100 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 100 insertions(+)
 create mode 100644 benches/server.rs

diff --git a/benches/server.rs b/benches/server.rs
new file mode 100644
index 00000000..8148b67a
--- /dev/null
+++ b/benches/server.rs
@@ -0,0 +1,100 @@
+#![feature(test)]
+#![deny(warnings)]
+
+extern crate futures;
+extern crate hyper;
+extern crate pretty_env_logger;
+extern crate test;
+
+use std::io::{Read, Write};
+use std::net::{TcpListener, TcpStream, SocketAddr};
+use std::sync::mpsc;
+
+use futures::Future;
+use futures::sync::oneshot;
+
+use hyper::header::{ContentLength, ContentType};
+use hyper::server::{self, Service};
+
+
+
+#[bench]
+fn bench_server_tcp_throughput(b: &mut test::Bencher) {
+    let (_until_tx, until_rx) = oneshot::channel();
+    let addr = spawn_hello(until_rx);
+
+    let mut tcp = TcpStream::connect(addr).unwrap();
+    let mut buf = [0u8; 4096];
+
+    b.bytes = 130 + 35;
+    b.iter(|| {
+        tcp.write_all(b"GET / HTTP/1.1\r\nHost: localhost\r\n\r\n").unwrap();
+        let n = tcp.read(&mut buf).unwrap();
+        assert_eq!(n, 130);
+    })
+}
+
+
+#[bench]
+fn bench_raw_tcp_throughput(b: &mut test::Bencher) {
+    let (tx, rx) = mpsc::channel();
+    let listener = TcpListener::bind("127.0.0.1:0").unwrap();
+    let addr = listener.local_addr().unwrap();
+    ::std::thread::spawn(move || {
+        let mut sock = listener.accept().unwrap().0;
+
+        let mut buf = [0u8; 4096];
+        while rx.try_recv().is_err() {
+            sock.read(&mut buf).unwrap();
+            sock.write_all(b"\
+                HTTP/1.1 200 OK\r\n\
+                Content-Length: 13\r\n\
+                Content-Type: text/plain; charset=utf-8\r\n\
+                Date: Fri, 12 May 2017 18:21:45 GMT\r\n\
+                \r\n\
+                Hello, World!\
+            ").unwrap();
+        }
+    });
+
+    let mut tcp = TcpStream::connect(addr).unwrap();
+    let mut buf = [0u8; 4096];
+
+    b.bytes = 130 + 35;
+    b.iter(|| {
+        tcp.write_all(b"GET / HTTP/1.1\r\nHost: localhost\r\n\r\n").unwrap();
+        let n = tcp.read(&mut buf).unwrap();
+        assert_eq!(n, 130);
+    });
+    tx.send(()).unwrap();
+}
+
+struct Hello;
+
+impl Service for Hello {
+    type Request = server::Request;
+    type Response = server::Response;
+    type Error = hyper::Error;
+    type Future = ::futures::Finished<Self::Response, hyper::Error>;
+    fn call(&self, _req: Self::Request) -> Self::Future {
+        ::futures::finished(
+            server::Response::new()
+                .with_header(ContentLength("Hello, World!".len() as u64))
+                .with_header(ContentType::plaintext())
+                .with_body("Hello, World!")
+        )
+    }
+}
+
+fn spawn_hello(until: oneshot::Receiver<()>) -> SocketAddr {
+    let (addr_tx, addr_rx) = mpsc::channel();
+    ::std::thread::spawn(move || {
+        let addr = "127.0.0.1:0".parse().unwrap();
+        let srv = hyper::server::Http::new().bind(&addr, || Ok(Hello)).unwrap();
+        let addr = srv.local_addr().unwrap();
+        addr_tx.send(addr).unwrap();
+        srv.run_until(until.map_err(|_| ())).unwrap();
+    });
+
+    addr_rx.recv().unwrap()
+}