feat(lib): redesign API to use Futures and Tokio
There are many changes involved with this, but let's just talk about
user-facing changes.
- Creating a `Client` and `Server` now needs a Tokio `Core` event loop
to attach to.
- `Request` and `Response` both no longer implement the
`std::io::{Read,Write}` traits, but instead represent their bodies as a
`futures::Stream` of items, where each item is a `Chunk`.
- The `Client.request` method now takes a `Request`, instead of being
used as a builder, and returns a `Future` that resolves to `Response`.
- The `Handler` trait for servers is no more, and instead the Tokio
`Service` trait is used. This allows interoperability with generic
middleware.
BREAKING CHANGE: A big sweeping set of breaking changes.
This commit is contained in:
354
tests/client.rs
354
tests/client.rs
@@ -1,189 +1,27 @@
|
||||
#![deny(warnings)]
|
||||
extern crate hyper;
|
||||
extern crate futures;
|
||||
extern crate tokio_core;
|
||||
|
||||
use std::io::{self, Read, Write};
|
||||
use std::net::TcpListener;
|
||||
use std::sync::mpsc;
|
||||
use std::thread;
|
||||
use std::time::Duration;
|
||||
|
||||
use hyper::client::{Handler, Request, Response, HttpConnector};
|
||||
use hyper::{Method, StatusCode, Next, Encoder, Decoder};
|
||||
use hyper::header::Headers;
|
||||
use hyper::net::HttpStream;
|
||||
use hyper::client::{Client, Request, HttpConnector};
|
||||
use hyper::{Method, StatusCode};
|
||||
|
||||
fn s(bytes: &[u8]) -> &str {
|
||||
::std::str::from_utf8(bytes.as_ref()).unwrap()
|
||||
use futures::Future;
|
||||
use futures::sync::oneshot;
|
||||
|
||||
use tokio_core::reactor::{Core, Handle};
|
||||
|
||||
fn client(handle: &Handle) -> Client<HttpConnector> {
|
||||
Client::new(handle)
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
struct TestHandler {
|
||||
opts: Opts,
|
||||
tx: mpsc::Sender<Msg>
|
||||
}
|
||||
|
||||
impl TestHandler {
|
||||
fn new(opts: Opts) -> (TestHandler, mpsc::Receiver<Msg>) {
|
||||
let (tx, rx) = mpsc::channel();
|
||||
(TestHandler {
|
||||
opts: opts,
|
||||
tx: tx
|
||||
}, rx)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
enum Msg {
|
||||
Head(Response),
|
||||
Chunk(Vec<u8>),
|
||||
Error(hyper::Error),
|
||||
}
|
||||
|
||||
fn read(opts: &Opts) -> Next {
|
||||
if let Some(timeout) = opts.read_timeout {
|
||||
Next::read().timeout(timeout)
|
||||
} else {
|
||||
Next::read()
|
||||
}
|
||||
}
|
||||
|
||||
impl Handler<HttpStream> for TestHandler {
|
||||
fn on_request(&mut self, req: &mut Request) -> Next {
|
||||
req.set_method(self.opts.method.clone());
|
||||
req.headers_mut().extend(self.opts.headers.iter());
|
||||
if self.opts.body.is_some() {
|
||||
Next::write()
|
||||
} else {
|
||||
read(&self.opts)
|
||||
}
|
||||
}
|
||||
|
||||
fn on_request_writable(&mut self, encoder: &mut Encoder<HttpStream>) -> Next {
|
||||
if let Some(ref mut body) = self.opts.body {
|
||||
let n = encoder.write(body).unwrap();
|
||||
*body = &body[n..];
|
||||
|
||||
if !body.is_empty() {
|
||||
return Next::write()
|
||||
}
|
||||
}
|
||||
encoder.close();
|
||||
read(&self.opts)
|
||||
}
|
||||
|
||||
fn on_response(&mut self, res: Response) -> Next {
|
||||
use hyper::header;
|
||||
// server responses can include a body until eof, if not size is specified
|
||||
let mut has_body = true;
|
||||
if let Some(len) = res.headers().get::<header::ContentLength>() {
|
||||
if **len == 0 {
|
||||
has_body = false;
|
||||
}
|
||||
}
|
||||
self.tx.send(Msg::Head(res)).unwrap();
|
||||
if has_body {
|
||||
read(&self.opts)
|
||||
} else {
|
||||
Next::end()
|
||||
}
|
||||
}
|
||||
|
||||
fn on_response_readable(&mut self, decoder: &mut Decoder<HttpStream>) -> Next {
|
||||
let mut v = vec![0; 512];
|
||||
match decoder.read(&mut v) {
|
||||
Ok(n) => {
|
||||
v.truncate(n);
|
||||
self.tx.send(Msg::Chunk(v)).unwrap();
|
||||
if n == 0 {
|
||||
Next::end()
|
||||
} else {
|
||||
read(&self.opts)
|
||||
}
|
||||
},
|
||||
Err(e) => match e.kind() {
|
||||
io::ErrorKind::WouldBlock => read(&self.opts),
|
||||
_ => panic!("io read error: {:?}", e)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn on_error(&mut self, err: hyper::Error) -> Next {
|
||||
self.tx.send(Msg::Error(err)).unwrap();
|
||||
Next::remove()
|
||||
}
|
||||
}
|
||||
|
||||
struct Client {
|
||||
client: Option<hyper::Client<TestHandler>>,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
struct Opts {
|
||||
body: Option<&'static [u8]>,
|
||||
method: Method,
|
||||
headers: Headers,
|
||||
read_timeout: Option<Duration>,
|
||||
}
|
||||
|
||||
impl Default for Opts {
|
||||
fn default() -> Opts {
|
||||
Opts {
|
||||
body: None,
|
||||
method: Method::Get,
|
||||
headers: Headers::new(),
|
||||
read_timeout: None,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn opts() -> Opts {
|
||||
Opts::default()
|
||||
}
|
||||
|
||||
impl Opts {
|
||||
fn method(mut self, method: Method) -> Opts {
|
||||
self.method = method;
|
||||
self
|
||||
}
|
||||
|
||||
fn header<H: ::hyper::header::Header>(mut self, header: H) -> Opts {
|
||||
self.headers.set(header);
|
||||
self
|
||||
}
|
||||
|
||||
fn body(mut self, body: Option<&'static [u8]>) -> Opts {
|
||||
self.body = body;
|
||||
self
|
||||
}
|
||||
|
||||
fn read_timeout(mut self, timeout: Duration) -> Opts {
|
||||
self.read_timeout = Some(timeout);
|
||||
self
|
||||
}
|
||||
}
|
||||
|
||||
impl Client {
|
||||
fn request<U>(&self, url: U, opts: Opts) -> mpsc::Receiver<Msg>
|
||||
where U: AsRef<str> {
|
||||
let (handler, rx) = TestHandler::new(opts);
|
||||
self.client.as_ref().unwrap()
|
||||
.request(url.as_ref().parse().unwrap(), handler).unwrap();
|
||||
rx
|
||||
}
|
||||
}
|
||||
|
||||
impl Drop for Client {
|
||||
fn drop(&mut self) {
|
||||
self.client.take().map(|c| c.close());
|
||||
}
|
||||
}
|
||||
|
||||
fn client() -> Client {
|
||||
let c = hyper::Client::<TestHandler>::configure()
|
||||
.connector(HttpConnector::default())
|
||||
.build().unwrap();
|
||||
Client {
|
||||
client: Some(c),
|
||||
}
|
||||
fn s(buf: &[u8]) -> &str {
|
||||
::std::str::from_utf8(buf).unwrap()
|
||||
}
|
||||
|
||||
macro_rules! test {
|
||||
@@ -206,43 +44,50 @@ macro_rules! test {
|
||||
) => (
|
||||
#[test]
|
||||
fn $name() {
|
||||
#[allow(unused)]
|
||||
#![allow(unused)]
|
||||
use hyper::header::*;
|
||||
let server = TcpListener::bind("127.0.0.1:0").unwrap();
|
||||
let addr = server.local_addr().unwrap();
|
||||
let client = client();
|
||||
let opts = opts()
|
||||
.method(Method::$client_method)
|
||||
.body($request_body);
|
||||
let mut core = Core::new().unwrap();
|
||||
let client = client(&core.handle());
|
||||
let mut req = Request::new(Method::$client_method, format!($client_url, addr=addr).parse().unwrap());
|
||||
$(
|
||||
let opts = opts.header($request_headers);
|
||||
req.headers_mut().set($request_headers);
|
||||
)*
|
||||
let res = client.request(format!($client_url, addr=addr), opts);
|
||||
|
||||
let mut inc = server.accept().unwrap().0;
|
||||
inc.set_read_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
inc.set_write_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
let expected = format!($server_expected, addr=addr);
|
||||
let mut buf = [0; 4096];
|
||||
let mut n = 0;
|
||||
while n < buf.len() && n < expected.len() {
|
||||
n += inc.read(&mut buf[n..]).unwrap();
|
||||
if let Some(body) = $request_body {
|
||||
let body: &'static str = body;
|
||||
req.set_body(body);
|
||||
}
|
||||
assert_eq!(s(&buf[..n]), expected);
|
||||
let res = client.request(req);
|
||||
|
||||
inc.write_all($server_reply.as_ref()).unwrap();
|
||||
let (tx, rx) = oneshot::channel();
|
||||
|
||||
if let Msg::Head(head) = res.recv().unwrap() {
|
||||
assert_eq!(head.status(), &StatusCode::$client_status);
|
||||
$(
|
||||
assert_eq!(head.headers().get(), Some(&$response_headers));
|
||||
)*
|
||||
} else {
|
||||
panic!("we lost the head!");
|
||||
}
|
||||
//drop(inc);
|
||||
thread::spawn(move || {
|
||||
let mut inc = server.accept().unwrap().0;
|
||||
inc.set_read_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
inc.set_write_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
let expected = format!($server_expected, addr=addr);
|
||||
let mut buf = [0; 4096];
|
||||
let mut n = 0;
|
||||
while n < buf.len() && n < expected.len() {
|
||||
n += inc.read(&mut buf[n..]).unwrap();
|
||||
}
|
||||
assert_eq!(s(&buf[..n]), expected);
|
||||
|
||||
assert!(res.recv().is_err());
|
||||
inc.write_all($server_reply.as_ref()).unwrap();
|
||||
tx.complete(());
|
||||
});
|
||||
|
||||
let rx = rx.map_err(|_| hyper::Error::Io(io::Error::new(io::ErrorKind::Other, "thread panicked")));
|
||||
|
||||
let work = res.join(rx).map(|r| r.0);
|
||||
|
||||
let res = core.run(work).unwrap();
|
||||
assert_eq!(res.status(), &StatusCode::$client_status);
|
||||
$(
|
||||
assert_eq!(res.headers().get(), Some(&$response_headers));
|
||||
)*
|
||||
}
|
||||
);
|
||||
}
|
||||
@@ -311,7 +156,7 @@ test! {
|
||||
headers: [
|
||||
ContentLength(7),
|
||||
],
|
||||
body: Some(b"foo bar"),
|
||||
body: Some("foo bar"),
|
||||
response:
|
||||
status: Ok,
|
||||
headers: [],
|
||||
@@ -340,51 +185,86 @@ test! {
|
||||
headers: [
|
||||
TransferEncoding::chunked(),
|
||||
],
|
||||
body: Some(b"foo bar baz"),
|
||||
body: Some("foo bar baz"),
|
||||
response:
|
||||
status: Ok,
|
||||
headers: [],
|
||||
body: None,
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn client_read_timeout() {
|
||||
let server = TcpListener::bind("127.0.0.1:0").unwrap();
|
||||
let addr = server.local_addr().unwrap();
|
||||
let client = client();
|
||||
let res = client.request(format!("http://{}/", addr), opts().read_timeout(Duration::from_secs(3)));
|
||||
|
||||
let mut inc = server.accept().unwrap().0;
|
||||
let mut buf = [0; 4096];
|
||||
inc.read(&mut buf).unwrap();
|
||||
|
||||
match res.recv() {
|
||||
Ok(Msg::Error(hyper::Error::Timeout)) => (),
|
||||
other => panic!("expected timeout, actual: {:?}", other)
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn client_keep_alive() {
|
||||
let server = TcpListener::bind("127.0.0.1:0").unwrap();
|
||||
let addr = server.local_addr().unwrap();
|
||||
let client = client();
|
||||
let res = client.request(format!("http://{}/a", addr), opts());
|
||||
let mut core = Core::new().unwrap();
|
||||
let client = client(&core.handle());
|
||||
|
||||
let mut sock = server.accept().unwrap().0;
|
||||
sock.set_read_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
sock.set_write_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
let mut buf = [0; 4096];
|
||||
sock.read(&mut buf).expect("read 1");
|
||||
sock.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n").expect("write 1");
|
||||
|
||||
while let Ok(_) = res.recv() {}
|
||||
let (tx1, rx1) = oneshot::channel();
|
||||
let (tx2, rx2) = oneshot::channel();
|
||||
thread::spawn(move || {
|
||||
let mut sock = server.accept().unwrap().0;
|
||||
sock.set_read_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
sock.set_write_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
let mut buf = [0; 4096];
|
||||
sock.read(&mut buf).expect("read 1");
|
||||
sock.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n").expect("write 1");
|
||||
tx1.complete(());
|
||||
|
||||
let res = client.request(format!("http://{}/b", addr), opts());
|
||||
sock.read(&mut buf).expect("read 2");
|
||||
let second_get = b"GET /b HTTP/1.1\r\n";
|
||||
assert_eq!(&buf[..second_get.len()], second_get);
|
||||
sock.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n").expect("write 2");
|
||||
sock.read(&mut buf).expect("read 2");
|
||||
let second_get = b"GET /b HTTP/1.1\r\n";
|
||||
assert_eq!(&buf[..second_get.len()], second_get);
|
||||
sock.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n").expect("write 2");
|
||||
tx2.complete(());
|
||||
});
|
||||
|
||||
while let Ok(_) = res.recv() {}
|
||||
|
||||
|
||||
let rx = rx1.map_err(|_| hyper::Error::Io(io::Error::new(io::ErrorKind::Other, "thread panicked")));
|
||||
let res = client.get(format!("http://{}/a", addr).parse().unwrap());
|
||||
core.run(res.join(rx).map(|r| r.0)).unwrap();
|
||||
|
||||
let rx = rx2.map_err(|_| hyper::Error::Io(io::Error::new(io::ErrorKind::Other, "thread panicked")));
|
||||
let res = client.get(format!("http://{}/b", addr).parse().unwrap());
|
||||
core.run(res.join(rx).map(|r| r.0)).unwrap();
|
||||
}
|
||||
|
||||
|
||||
#[test]
|
||||
fn client_pooled_socket_disconnected() {
|
||||
let server = TcpListener::bind("127.0.0.1:0").unwrap();
|
||||
let addr = server.local_addr().unwrap();
|
||||
let mut core = Core::new().unwrap();
|
||||
let client = client(&core.handle());
|
||||
|
||||
|
||||
let (tx1, rx1) = oneshot::channel();
|
||||
let (tx2, rx2) = oneshot::channel();
|
||||
thread::spawn(move || {
|
||||
let mut sock = server.accept().unwrap().0;
|
||||
sock.set_read_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
sock.set_write_timeout(Some(Duration::from_secs(5))).unwrap();
|
||||
let mut buf = [0; 4096];
|
||||
sock.read(&mut buf).expect("read 1");
|
||||
sock.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n").expect("write 1");
|
||||
drop(sock);
|
||||
tx1.complete(());
|
||||
|
||||
let mut sock = server.accept().unwrap().0;
|
||||
sock.read(&mut buf).expect("read 2");
|
||||
let second_get = b"GET /b HTTP/1.1\r\n";
|
||||
assert_eq!(&buf[..second_get.len()], second_get);
|
||||
sock.write_all(b"HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n").expect("write 2");
|
||||
tx2.complete(());
|
||||
});
|
||||
|
||||
|
||||
|
||||
let rx = rx1.map_err(|_| hyper::Error::Io(io::Error::new(io::ErrorKind::Other, "thread panicked")));
|
||||
let res = client.get(format!("http://{}/a", addr).parse().unwrap());
|
||||
core.run(res.join(rx).map(|r| r.0)).unwrap();
|
||||
|
||||
let rx = rx2.map_err(|_| hyper::Error::Io(io::Error::new(io::ErrorKind::Other, "thread panicked")));
|
||||
let res = client.get(format!("http://{}/b", addr).parse().unwrap());
|
||||
core.run(res.join(rx).map(|r| r.0)).unwrap();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user