forked/hyper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

docs(examples): Update web_api example to use async await

Browse Source
This commit is contained in:
messense
2019-07-12 14:21:34 +08:00
committed by Sean McArthur
parent 7cdfd3d974
commit 524a453b6f
1 changed files with 61 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
examples_disabled/web_api.rs
View File

@@ -1,24 +1,24 @@
#![feature(async_await)]
#![deny(warnings)] #![deny(warnings)]
extern crate futures;
extern crate hyper; extern crate hyper;
extern crate pretty_env_logger; extern crate pretty_env_logger;
extern crate serde_json; extern crate serde_json;
use futures::{future, Future, Stream};
use hyper::{Body, Chunk, Client, Method, Request, Response, Server, StatusCode, header}; use hyper::{Body, Chunk, Client, Method, Request, Response, Server, StatusCode, header};
use hyper::client::HttpConnector; use hyper::client::HttpConnector;
use hyper::service::service_fn; use hyper::service::{service_fn, make_service_fn};
use futures_util::{TryStreamExt};
type GenericError = Box<dyn std::error::Error + Send + Sync>;
static NOTFOUND: &[u8] = b"Not Found"; static NOTFOUND: &[u8] = b"Not Found";
static URL: &str = "http://127.0.0.1:1337/json_api"; static URL: &str = "http://127.0.0.1:1337/json_api";
static INDEX: &[u8] = b"<a href=\"test.html\">test.html</a>"; static INDEX: &[u8] = b"<a href=\"test.html\">test.html</a>";
static POST_DATA: &str = r#"{"original": "data"}"#; static POST_DATA: &str = r#"{"original": "data"}"#;
type GenericError = Box<dyn std::error::Error + Send + Sync>; async fn client_request_response(client: &Client<HttpConnector>)
type ResponseFuture = Box<dyn Future<Item=Response<Body>, Error=GenericError> + Send>; -> Result<Response<Body>, GenericError>
{
fn client_request_response(client: &Client<HttpConnector>) -> ResponseFuture {
let req = Request::builder() let req = Request::builder()
.method(Method::POST) .method(Method::POST)
.uri(URL) .uri(URL)
@@ -26,24 +26,22 @@ fn client_request_response(client: &Client<HttpConnector>) -> ResponseFuture {
.body(POST_DATA.into()) .body(POST_DATA.into())
.unwrap(); .unwrap();
Box::new(client.request(req).from_err().map(|web_res| { let web_res = client.request(req).await?;
// Compare the JSON we sent (before) with what we received (after): // Compare the JSON we sent (before) with what we received (after):
let body = Body::wrap_stream(web_res.into_body().map(|b| { let body = Body::wrap_stream(web_res.into_body().map_ok(|b| {
Chunk::from(format!("<b>POST request body</b>: {}<br><b>Response</b>: {}", Chunk::from(format!("<b>POST request body</b>: {}<br><b>Response</b>: {}",
POST_DATA, POST_DATA,
std::str::from_utf8(&b).unwrap())) std::str::from_utf8(&b).unwrap()))
})); }));
Response::new(body) Ok(Response::new(body))
}))
} }
fn api_post_response(req: Request<Body>) -> ResponseFuture { async fn api_post_response(req: Request<Body>)
-> Result<Response<Body>, GenericError>
{
// A web api to run against // A web api to run against
Box::new(req.into_body() let entire_body = req.into_body().try_concat().await?;
.concat2() // Concatenate all chunks in the body
.from_err()
.and_then(|entire_body| {
// TODO: Replace all unwraps with proper error handling // TODO: Replace all unwraps with proper error handling
let str = String::from_utf8(entire_body.to_vec())?; let str = String::from_utf8(entire_body.to_vec())?;
let mut data : serde_json::Value = serde_json::from_str(&str)?; let mut data : serde_json::Value = serde_json::from_str(&str)?;
@@ -54,11 +52,9 @@ fn api_post_response(req: Request<Body>) -> ResponseFuture {
.header(header::CONTENT_TYPE, "application/json") .header(header::CONTENT_TYPE, "application/json")
.body(Body::from(json))?; .body(Body::from(json))?;
Ok(response) Ok(response)
})
)
} }
fn api_get_response() -> ResponseFuture { async fn api_get_response() -> Result<Response<Body>, GenericError> {
let data = vec!["foo", "bar"]; let data = vec!["foo", "bar"];
let res = match serde_json::to_string(&data) { let res = match serde_json::to_string(&data) {
Ok(json) => { Ok(json) => {
@@ -74,59 +70,62 @@ fn api_get_response() -> ResponseFuture {
.unwrap() .unwrap()
} }
}; };
Ok(res)
Box::new(future::ok(res))
} }
fn response_examples(req: Request<Body>, client: &Client<HttpConnector>) -> ResponseFuture { async fn response_examples(req: Request<Body>, client: &Client<HttpConnector>)
-> Result<Response<Body>, GenericError>
{
match (req.method(), req.uri().path()) { match (req.method(), req.uri().path()) {
(&Method::GET, "/") | (&Method::GET, "/index.html") => { (&Method::GET, "/") | (&Method::GET, "/index.html") => {
let body = Body::from(INDEX); let body = Body::from(INDEX);
Box::new(future::ok(Response::new(body))) Ok(Response::new(body))
}, },
(&Method::GET, "/test.html") => { (&Method::GET, "/test.html") => {
client_request_response(client) client_request_response(client).await
}, },
(&Method::POST, "/json_api") => { (&Method::POST, "/json_api") => {
api_post_response(req) api_post_response(req).await
}, },
(&Method::GET, "/json_api") => { (&Method::GET, "/json_api") => {
api_get_response() api_get_response().await
} }
_ => { _ => {
// Return 404 not found response. // Return 404 not found response.
let body = Body::from(NOTFOUND); let body = Body::from(NOTFOUND);
Box::new(future::ok(Response::builder() Ok(Response::builder()
.status(StatusCode::NOT_FOUND) .status(StatusCode::NOT_FOUND)
.body(body) .body(body)
.unwrap())) .unwrap())
} }
} }
} }
fn main() { #[hyper::rt::main]
async fn main() -> Result<(), GenericError> {
pretty_env_logger::init(); pretty_env_logger::init();
let addr = "127.0.0.1:1337".parse().unwrap(); let addr = "127.0.0.1:1337".parse().unwrap();
hyper::rt::run(future::lazy(move || {
// Share a `Client` with all `Service`s // Share a `Client` with all `Service`s
let client = Client::new(); let client = Client::new();
let new_service = move || { let new_service = make_service_fn(move |_| {
// Move a clone of `client` into the `service_fn`. // Move a clone of `client` into the `service_fn`.
let client = client.clone(); let client = client.clone();
service_fn(move |req| { async {
Ok::<_, GenericError>(service_fn(move |req| {
response_examples(req, &client) response_examples(req, &client)
}) }))
}; }
});
let server = Server::bind(&addr) let server = Server::bind(&addr)
.serve(new_service) .serve(new_service);
.map_err(|e| eprintln!("server error: {}", e));
println!("Listening on http://{}", addr); println!("Listening on http://{}", addr);
server server.await?;
}));
Ok(())
} }