fbc449e49c
This dedicated `Entity` trait replaces the previous `Stream<Item=impl AsRef<[u8]>, Error=hyper::Error>`. This allows for several improvements immediately, and prepares for HTTP2 support. - The `Entity::is_end_stream` makes up for change away from `Option<Body>`, which was previously used to know if the body should be empty. Since `Request` and `Response` now require a body to be set, this method can be used to tell hyper that the body is actually empty. It also provides the possibility of slight optimizations when polling for data, by allowing to check `is_end_stream` before polling again. This can allow a consumer to know that a body stream has ended without polling for `None` afterwards. - The `Entity::content_length` method allows a body to automatically declare a size, in case a user doesn't set a `Content-Length` or `Transfer-Encoding` header. - It's now possible to send and receive trailers, though this will be for HTTP2 connections only. By being a trait owned by hyper, new methods can be added later as new features are wanted (with default implementations). The `hyper::Body` type now implements `Entity` instead of `Stream`, provides a better channel option, and is easier to use with custom streams via `Body::wrap_stream`. BREAKING CHANGE: All code that was assuming the body was a `Stream` must be adjusted to use an `Entity` instead. Using `hyper::Body` as a `Stream` can call `Body::into_stream` to get a stream wrapper. Passing a custom `impl Stream` will need to either implement `Entity`, or as an easier option, switch to `Body::wrap_stream`. `Body::pair` has been replaced with `Body::channel`, which returns a `hyper::body::Sender` instead of a `futures::sync::mpsc::Sender`. Closes #1438
146 lines
5.5 KiB
Rust
146 lines
5.5 KiB
Rust
#![deny(warnings)]
|
|
extern crate futures;
|
|
extern crate hyper;
|
|
extern crate pretty_env_logger;
|
|
|
|
use futures::{Future/*, Sink*/};
|
|
use futures::sync::oneshot;
|
|
|
|
use hyper::{Body, /*Chunk,*/ Method, Request, Response, StatusCode};
|
|
use hyper::error::Error;
|
|
use hyper::server::{Http, Service};
|
|
|
|
use std::fs::File;
|
|
use std::io::{self, copy/*, Read*/};
|
|
use std::thread;
|
|
|
|
static NOTFOUND: &[u8] = b"Not Found";
|
|
static INDEX: &str = "examples/send_file_index.html";
|
|
|
|
fn simple_file_send(f: &str) -> Box<Future<Item = Response<Body>, Error = hyper::Error>> {
|
|
// Serve a file by reading it entirely into memory. As a result
|
|
// this is limited to serving small files, but it is somewhat
|
|
// simpler with a little less overhead.
|
|
//
|
|
// On channel errors, we panic with the expect method. The thread
|
|
// ends at that point in any case.
|
|
let filename = f.to_string(); // we need to copy for lifetime issues
|
|
let (tx, rx) = oneshot::channel();
|
|
thread::spawn(move || {
|
|
let mut file = match File::open(filename) {
|
|
Ok(f) => f,
|
|
Err(_) => {
|
|
tx.send(Response::builder()
|
|
.status(StatusCode::NOT_FOUND)
|
|
.body(NOTFOUND.into())
|
|
.unwrap())
|
|
.expect("Send error on open");
|
|
return;
|
|
},
|
|
};
|
|
let mut buf: Vec<u8> = Vec::new();
|
|
match copy(&mut file, &mut buf) {
|
|
Ok(_) => {
|
|
let res = Response::new(buf.into());
|
|
tx.send(res).expect("Send error on successful file read");
|
|
},
|
|
Err(_) => {
|
|
tx.send(Response::builder()
|
|
.status(StatusCode::INTERNAL_SERVER_ERROR)
|
|
.body(Body::empty())
|
|
.unwrap())
|
|
.expect("Send error on error reading file");
|
|
},
|
|
};
|
|
});
|
|
|
|
Box::new(rx.map_err(|e| Error::from(io::Error::new(io::ErrorKind::Other, e))))
|
|
}
|
|
|
|
struct ResponseExamples;
|
|
|
|
impl Service for ResponseExamples {
|
|
type Request = Request<Body>;
|
|
type Response = Response<Body>;
|
|
type Error = hyper::Error;
|
|
type Future = Box<Future<Item = Self::Response, Error = Self::Error>>;
|
|
|
|
fn call(&self, req: Request<Body>) -> Self::Future {
|
|
match (req.method(), req.uri().path()) {
|
|
(&Method::GET, "/") | (&Method::GET, "/index.html") => {
|
|
simple_file_send(INDEX)
|
|
},
|
|
(&Method::GET, "/big_file.html") => {
|
|
// Stream a large file in chunks. This requires a
|
|
// little more overhead with two channels, (one for
|
|
// the response future, and a second for the response
|
|
// body), but can handle arbitrarily large files.
|
|
//
|
|
// We use an artificially small buffer, since we have
|
|
// a small test file.
|
|
let (tx, rx) = oneshot::channel();
|
|
thread::spawn(move || {
|
|
let _file = match File::open(INDEX) {
|
|
Ok(f) => f,
|
|
Err(_) => {
|
|
tx.send(Response::builder()
|
|
.status(StatusCode::NOT_FOUND)
|
|
.body(NOTFOUND.into())
|
|
.unwrap())
|
|
.expect("Send error on open");
|
|
return;
|
|
},
|
|
};
|
|
let (_tx_body, rx_body) = Body::channel();
|
|
let res = Response::new(rx_body.into());
|
|
tx.send(res).expect("Send error on successful file read");
|
|
/* TODO: fix once we have futures 0.2 Sink working
|
|
let mut buf = [0u8; 16];
|
|
loop {
|
|
match file.read(&mut buf) {
|
|
Ok(n) => {
|
|
if n == 0 {
|
|
// eof
|
|
tx_body.close().expect("panic closing");
|
|
break;
|
|
} else {
|
|
let chunk: Chunk = buf[0..n].to_vec().into();
|
|
match tx_body.send_data(chunk).wait() {
|
|
Ok(t) => { tx_body = t; },
|
|
Err(_) => { break; }
|
|
};
|
|
}
|
|
},
|
|
Err(_) => { break; }
|
|
}
|
|
}
|
|
*/
|
|
});
|
|
|
|
Box::new(rx.map_err(|e| Error::from(io::Error::new(io::ErrorKind::Other, e))))
|
|
},
|
|
(&Method::GET, "/no_file.html") => {
|
|
// Test what happens when file cannot be be found
|
|
simple_file_send("this_file_should_not_exist.html")
|
|
},
|
|
_ => {
|
|
Box::new(futures::future::ok(Response::builder()
|
|
.status(StatusCode::NOT_FOUND)
|
|
.body(Body::empty())
|
|
.unwrap()))
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
fn main() {
|
|
pretty_env_logger::init();
|
|
let addr = "127.0.0.1:1337".parse().unwrap();
|
|
|
|
let server = Http::new().bind(&addr, || Ok(ResponseExamples)).unwrap();
|
|
println!("Listening on http://{} with 1 thread.", server.local_addr().unwrap());
|
|
server.run().unwrap();
|
|
}
|